Skip to content

Commit d1d8045

Browse files
committed
Disable shim_temporary.h
1 parent a6a3a6d commit d1d8045

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/libtorchaudio/cuda_utils.h

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
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(
2120
inline 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

2827
inline 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

3736
inline 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

0 commit comments

Comments
 (0)