[Fix] Fix record_stream for PyTorch 1.14 (dmlc#4975)

XiaoYixin7132 · Dec 2, 2022 · 26612d0 · 26612d0
1 parent 943abba
commit 26612d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensoradapter/pytorch/torch.cpp b/tensoradapter/pytorch/torch.cpp
@@ -41,7 +41,7 @@ TA_EXPORTS cudaStream_t CUDACurrentStream() {
 
 TA_EXPORTS void RecordStream(void* ptr, cudaStream_t stream, int device_id) {
   c10::DataPtr data_ptr{
-      ptr, ptr, &c10::cuda::CUDACachingAllocator::raw_delete,
+      ptr, ptr, c10::cuda::CUDACachingAllocator::get()->raw_deleter(),
       c10::Device(c10::DeviceType::CUDA, device_id)};
   c10::cuda::CUDACachingAllocator::recordStream(
       data_ptr,