From 26612d0cb0d226e447fb42470f2f27a9128c6f51 Mon Sep 17 00:00:00 2001 From: Xin Yao Date: Fri, 2 Dec 2022 13:31:47 +0800 Subject: [PATCH] [Fix] Fix record_stream for PyTorch 1.14 (#4975) --- tensoradapter/pytorch/torch.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensoradapter/pytorch/torch.cpp b/tensoradapter/pytorch/torch.cpp index bc18113e318c..64125a566dce 100644 --- 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,