Skip to content

GitLab

Unverified Commit 26612d0c authored by Xin Yao's avatar Xin Yao Committed by GitHub
Browse files

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

parent 943abba3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensoradapter/pytorch/torch.cpp
View file @ 26612d0c
...@@ -41,7 +41,7 @@ TA_EXPORTS cudaStream_t CUDACurrentStream() { ...@@ -41,7 +41,7 @@ TA_EXPORTS cudaStream_t CUDACurrentStream() {
TA_EXPORTS void RecordStream(void* ptr, cudaStream_t stream, int device_id) { TA_EXPORTS void RecordStream(void* ptr, cudaStream_t stream, int device_id) {
c10::DataPtr data_ptr{ 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::Device(c10::DeviceType::CUDA, device_id)};
c10::cuda::CUDACachingAllocator::recordStream( c10::cuda::CUDACachingAllocator::recordStream(
data_ptr, data_ptr,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment