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 89d480b commit 7f11d1dCopy full SHA for 7f11d1d
src/libtorchaudio/rnnt/gpu/compute_betas.cu
@@ -46,8 +46,8 @@ RAIIATH compute_betas(
46
int32_t logits_dtype;
47
aoti_torch_get_dtype(logits.get(), &logits_dtype);
48
49
- aoti_torch_get_current_cuda_stream(logits_device_index, &options.stream_);
50
- cudaSetDevice(logits_device)
+ aoti_torch_get_current_cuda_stream(logits_device_index, (void**)&options.stream_);
+ cudaSetDevice(logits_device);
51
options.device_ = GPU;
52
53
int64_t cost_sizes[1] = {options.batchSize_};
0 commit comments