Skip to content

Commit 76a32f6

Browse files
committed
Re-enable device guard
1 parent 644c4a9 commit 76a32f6

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/libtorchaudio/iir_cuda.cu

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,7 @@ Tensor cuda_lfilter_core_loop(
6565

6666
STD_TORCH_CHECK(in.size(2) + a_flipped.size(1) - 1 == padded_out.size(2));
6767

68-
// TODO: enable device guard:
69-
// const at::cuda::OptionalCUDAGuard device_guard(in.device());
68+
const at::cuda::OptionalCUDAGuard device_guard(in.get_device_index());
7069

7170
const dim3 threads(256);
7271
const dim3 blocks((N * C + threads.x - 1) / threads.x);

0 commit comments

Comments
 (0)