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 5490cd3 commit 89d480bCopy full SHA for 89d480b
src/libtorchaudio/rnnt/gpu/compute_alphas.cu
@@ -44,8 +44,8 @@ RAIIATH compute_alphas(
44
int32_t logits_dtype;
45
aoti_torch_get_dtype(logits.get(), &logits_dtype);
46
47
- aoti_torch_get_current_cuda_stream(logits_device_index, &options.stream_);
48
- cudaSetDevice(logits_device)
+ aoti_torch_get_current_cuda_stream(logits_device_index, (void**)&options.stream_);
+ cudaSetDevice(logits_device);
49
options.device_ = GPU;
50
51
int64_t param_sizes[3] = {options.batchSize_ * options.nHypos_, options.maxSrcLen_, options.maxTgtLen_};
0 commit comments