From cf99efb67eae50a0f99d028a6ee6a55078d36d6f Mon Sep 17 00:00:00 2001 From: skishore Date: Thu, 5 Mar 2026 16:56:55 +0000 Subject: [PATCH 1/2] remove hip namespace shim --- src/libtorchaudio/hip_namespace_shim.h | 31 -------------------------- 1 file changed, 31 deletions(-) delete mode 100644 src/libtorchaudio/hip_namespace_shim.h diff --git a/src/libtorchaudio/hip_namespace_shim.h b/src/libtorchaudio/hip_namespace_shim.h deleted file mode 100644 index 7d03d919a7..0000000000 --- a/src/libtorchaudio/hip_namespace_shim.h +++ /dev/null @@ -1,31 +0,0 @@ -#pragma once - -#include - -namespace libtorchaudio::hip { - -inline hipStream_t getCurrentHIPStreamMasqueradingAsCUDA( - torch::stable::DeviceIndex device_index = -1) { - return cuda::getCurrentHIPStreamMasqueradingAsCUDA(device_index); -} - -inline void setCurrentHIPStreamMasqueradingAsCUDA( - hipStream_t stream, - torch::stable::DeviceIndex device_index = -1) { - cuda::setCurrentHIPStreamMasqueradingAsCUDA(stream, device_index); -} - -inline hipStream_t getStreamFromPoolMasqueradingAsCUDA( - const bool isHighPriority = false, - torch::stable::DeviceIndex device_index = -1) { - return cuda::getStreamFromPoolMasqueradingAsCUDA( - isHighPriority, device_index); -} - -inline void synchronize( - hipStream_t stream, - torch::stable::DeviceIndex device_index = -1) { - cuda::synchronize(stream, device_index); -} - -} // namespace libtorchaudio::hip From faf4a35fec0f9ae0b82f25c5846e8b06287f6861 Mon Sep 17 00:00:00 2001 From: skishore Date: Fri, 6 Mar 2026 21:26:41 +0000 Subject: [PATCH 2/2] remove reference of hip namespace shim --- src/libtorchaudio/forced_align/gpu/compute.cu | 4 ---- src/libtorchaudio/rnnt/gpu/compute.cu | 4 ---- 2 files changed, 8 deletions(-) diff --git a/src/libtorchaudio/forced_align/gpu/compute.cu b/src/libtorchaudio/forced_align/gpu/compute.cu index c3be77a256..444a4f8f6d 100644 --- a/src/libtorchaudio/forced_align/gpu/compute.cu +++ b/src/libtorchaudio/forced_align/gpu/compute.cu @@ -8,10 +8,6 @@ #include #include -#ifdef USE_ROCM - #include -#endif - namespace { constexpr int kNumThreads = 1024; // Number of threads to run CUDA kernel in parallel. diff --git a/src/libtorchaudio/rnnt/gpu/compute.cu b/src/libtorchaudio/rnnt/gpu/compute.cu index 40c5945f81..7e99fec395 100644 --- a/src/libtorchaudio/rnnt/gpu/compute.cu +++ b/src/libtorchaudio/rnnt/gpu/compute.cu @@ -6,10 +6,6 @@ #include #include -#ifdef USE_ROCM - #include -#endif - namespace torchaudio { namespace rnnt { namespace gpu {