Skip to content

Commit 7f11d1d

Browse files
committed
Cast to void pointer pointer
1 parent 89d480b commit 7f11d1d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/libtorchaudio/rnnt/gpu/compute_betas.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ RAIIATH compute_betas(
4646
int32_t logits_dtype;
4747
aoti_torch_get_dtype(logits.get(), &logits_dtype);
4848

49-
aoti_torch_get_current_cuda_stream(logits_device_index, &options.stream_);
50-
cudaSetDevice(logits_device)
49+
aoti_torch_get_current_cuda_stream(logits_device_index, (void**)&options.stream_);
50+
cudaSetDevice(logits_device);
5151
options.device_ = GPU;
5252

5353
int64_t cost_sizes[1] = {options.batchSize_};

0 commit comments

Comments
 (0)