Merge pull request #73 from fuzic/pytorch_bindings

Fix workspace allocation for cpu-mode to have correct size and deallocation
SeanNaren · Jul 27, 2018 · ac045b6 · ac045b6
2 parents 7fcbf65 + 025cc42
commit ac045b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_binding/src/binding.cpp b/pytorch_binding/src/binding.cpp
@@ -54,15 +54,15 @@ extern "C" int cpu_ctc(THFloatTensor *probs,
                        probs_size, minibatch_size,
                        options, &cpu_size_bytes);
 
-    float* cpu_workspace = (float*) new unsigned char[cpu_size_bytes];
+    float* cpu_workspace = new float[cpu_size_bytes / sizeof(float)];
 
     compute_ctc_loss(probs_ptr, grads_ptr,
                      labels_ptr, label_sizes_ptr,
                      sizes_ptr, probs_size,
                      minibatch_size, costs_ptr,
                      cpu_workspace, options);
 
-    delete cpu_workspace;
+    delete[] cpu_workspace;
     return 1;
 }