diff --git a/torchaudio/csrc/transducer.cpp b/torchaudio/csrc/transducer.cpp index 2d2b7a8b51..3b7f4e29ef 100644 --- a/torchaudio/csrc/transducer.cpp +++ b/torchaudio/csrc/transducer.cpp @@ -43,10 +43,10 @@ int64_t cpu_rnnt_loss(torch::Tensor acts, std::vector cpu_workspace(cpu_size_bytes / sizeof(float), 0); - compute_rnnt_loss(acts.data(), grads.data(), - labels.data(), label_lengths.data(), - input_lengths.data(), alphabet_size, - minibatch_size, costs.data(), + compute_rnnt_loss(acts.data_ptr(), grads.data_ptr(), + labels.data_ptr(), label_lengths.data_ptr(), + input_lengths.data_ptr(), alphabet_size, + minibatch_size, costs.data_ptr(), cpu_workspace.data(), options); return 0; @@ -59,10 +59,10 @@ int64_t cpu_rnnt_loss(torch::Tensor acts, std::vector cpu_workspace(cpu_size_bytes / sizeof(double), 0); - compute_rnnt_loss_fp64(acts.data(), grads.data(), - labels.data(), label_lengths.data(), - input_lengths.data(), alphabet_size, - minibatch_size, costs.data(), + compute_rnnt_loss_fp64(acts.data_ptr(), grads.data_ptr(), + labels.data_ptr(), label_lengths.data_ptr(), + input_lengths.data_ptr(), alphabet_size, + minibatch_size, costs.data_ptr(), cpu_workspace.data(), options); return 0;