Unverified Commit 422edb18 authored by Vincent QB's avatar Vincent QB Committed by GitHub
Browse files

update data pointer (#1160)

parent f2da5861
...@@ -43,10 +43,10 @@ int64_t cpu_rnnt_loss(torch::Tensor acts, ...@@ -43,10 +43,10 @@ int64_t cpu_rnnt_loss(torch::Tensor acts,
std::vector<float> cpu_workspace(cpu_size_bytes / sizeof(float), 0); std::vector<float> cpu_workspace(cpu_size_bytes / sizeof(float), 0);
compute_rnnt_loss(acts.data<float>(), grads.data<float>(), compute_rnnt_loss(acts.data_ptr<float>(), grads.data_ptr<float>(),
labels.data<int>(), label_lengths.data<int>(), labels.data_ptr<int>(), label_lengths.data_ptr<int>(),
input_lengths.data<int>(), alphabet_size, input_lengths.data_ptr<int>(), alphabet_size,
minibatch_size, costs.data<float>(), minibatch_size, costs.data_ptr<float>(),
cpu_workspace.data(), options); cpu_workspace.data(), options);
return 0; return 0;
...@@ -59,10 +59,10 @@ int64_t cpu_rnnt_loss(torch::Tensor acts, ...@@ -59,10 +59,10 @@ int64_t cpu_rnnt_loss(torch::Tensor acts,
std::vector<double> cpu_workspace(cpu_size_bytes / sizeof(double), 0); std::vector<double> cpu_workspace(cpu_size_bytes / sizeof(double), 0);
compute_rnnt_loss_fp64(acts.data<double>(), grads.data<double>(), compute_rnnt_loss_fp64(acts.data_ptr<double>(), grads.data_ptr<double>(),
labels.data<int>(), label_lengths.data<int>(), labels.data_ptr<int>(), label_lengths.data_ptr<int>(),
input_lengths.data<int>(), alphabet_size, input_lengths.data_ptr<int>(), alphabet_size,
minibatch_size, costs.data<double>(), minibatch_size, costs.data_ptr<double>(),
cpu_workspace.data(), options); cpu_workspace.data(), options);
return 0; return 0;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment