warpctc from github

a60ed4d7 · lishen · 949fcc19 · a60ed4d7 · a60ed4d7 · a60ed4d7
Commit a60ed4d7 authored May 16, 2023 by lishen
12 changed files
--- a/pytorch_binding/src/cpu_binding.h
+++ b/pytorch_binding/src/cpu_binding.h
+/*
+int cpu_ctc(THFloatTensor *probs,
+                        THFloatTensor *grads,
+                        THIntTensor *labels_ptr,
+                        THIntTensor *label_sizes_ptr,
+                        THIntTensor *sizes,
+                        int minibatch_size,
+                        THFloatTensor *costs,
+                        int blank_label);
+*/
+
+int cpu_ctc(torch::Tensor probs,
+            torch::Tensor grads,
+            torch::Tensor labels,
+            torch::Tensor label_sizes,
+            torch::Tensor sizes,
+            int minibatch_size,
+            torch::Tensor costs,
+            int blank_label); 
--- a/pytorch_binding/src/gpu_binding.h
+++ b/pytorch_binding/src/gpu_binding.h
--- a/pytorch_binding/tests/test_cpu.py
+++ b/pytorch_binding/tests/test_cpu.py
--- a/pytorch_binding/tests/test_gpu.py
+++ b/pytorch_binding/tests/test_gpu.py
--- a/pytorch_binding/warpctc_pytorch/__init__.py
+++ b/pytorch_binding/warpctc_pytorch/__init__.py
--- a/src/ctc_entrypoint.cpp
+++ b/src/ctc_entrypoint.cpp
--- a/src/ctc_entrypoint.cu
+++ b/src/ctc_entrypoint.cu
+ctc_entrypoint.cpp
\ No newline at end of file
--- a/src/reduce.cu
+++ b/src/reduce.cu
--- a/tests/random.cpp
+++ b/tests/random.cpp
--- a/tests/test.h
+++ b/tests/test.h
--- a/tests/test_cpu.cpp
+++ b/tests/test_cpu.cpp
--- a/tests/test_gpu.cu
+++ b/tests/test_gpu.cu