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

26612d0c · Xin Yao · GitHub · 943abba3 · 26612d0c
Unverified Commit 26612d0c authored Dec 02, 2022 by Xin Yao Committed by GitHub Dec 02, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tensoradapter/pytorch/torch.cpp tensoradapter/pytorch/torch.cpp +1 -1

No files found.
--- 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,