Skip to content

Commit 89d480b

Browse files
committed
Add missing semicolon
1 parent 5490cd3 commit 89d480b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/libtorchaudio/rnnt/gpu/compute_alphas.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@ RAIIATH compute_alphas(
4444
int32_t logits_dtype;
4545
aoti_torch_get_dtype(logits.get(), &logits_dtype);
4646

47-
aoti_torch_get_current_cuda_stream(logits_device_index, &options.stream_);
48-
cudaSetDevice(logits_device)
47+
aoti_torch_get_current_cuda_stream(logits_device_index, (void**)&options.stream_);
48+
cudaSetDevice(logits_device);
4949
options.device_ = GPU;
5050

5151
int64_t param_sizes[3] = {options.batchSize_ * options.nHypos_, options.maxSrcLen_, options.maxTgtLen_};

0 commit comments

Comments
 (0)