We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 644c4a9 commit 76a32f6Copy full SHA for 76a32f6
src/libtorchaudio/iir_cuda.cu
@@ -65,8 +65,7 @@ Tensor cuda_lfilter_core_loop(
65
66
STD_TORCH_CHECK(in.size(2) + a_flipped.size(1) - 1 == padded_out.size(2));
67
68
- // TODO: enable device guard:
69
- // const at::cuda::OptionalCUDAGuard device_guard(in.device());
+ const at::cuda::OptionalCUDAGuard device_guard(in.get_device_index());
70
71
const dim3 threads(256);
72
const dim3 blocks((N * C + threads.x - 1) / threads.x);
0 commit comments