File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed
Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change 11#pragma once
22
3- #include < libtorchaudio/shim_temporary.h>
43#include < torch/csrc/stable/c/shim.h>
54#include < torch/csrc/stable/device.h>
65
@@ -21,23 +20,23 @@ inline cudaStream_t getCurrentCUDAStream(
2120inline void setCurrentCUDAStream (
2221 cudaStream_t stream,
2322 torch::stable::DeviceIndex device_index = -1 ) {
24- TORCH_ERROR_CODE_CHECK (tmp_torch_set_current_cuda_stream (
23+ TORCH_ERROR_CODE_CHECK (torch_set_current_cuda_stream (
2524 static_cast <void *>(stream), device_index));
2625}
2726
2827inline cudaStream_t getStreamFromPool (
2928 const bool isHighPriority = false ,
3029 torch::stable::DeviceIndex device_index = -1 ) {
3130 void * stream_ptr = nullptr ;
32- TORCH_ERROR_CODE_CHECK (tmp_torch_get_cuda_stream_from_pool (
31+ TORCH_ERROR_CODE_CHECK (torch_get_cuda_stream_from_pool (
3332 isHighPriority, device_index, &stream_ptr));
3433 return static_cast <cudaStream_t>(stream_ptr);
3534}
3635
3736inline void synchronize (
3837 cudaStream_t stream,
3938 torch::stable::DeviceIndex device_index = -1 ) {
40- TORCH_ERROR_CODE_CHECK (tmp_torch_cuda_stream_synchronize (
39+ TORCH_ERROR_CODE_CHECK (torch_cuda_stream_synchronize (
4140 static_cast <void *>(stream), device_index));
4241}
4342
You can’t perform that action at this time.
0 commit comments