Refactor extension structure

mthrok · Jun 19, 2017 · e669fc1 · e669fc1
1 parent f8b6636
commit e669fc1
Show file tree

Hide file tree

Showing 155 changed files with 30 additions and 17 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,6 +1,5 @@
 build
 _ext
-pytorch_ctc
 *.pyc
 
 # Byte-compiled / optimized / DLL files

diff --git a/ctc_decode/__init__.py b/ctc_decode/__init__.py
diff --git a/pytorch_ctc/__init__.py b/pytorch_ctc/__init__.py
@@ -0,0 +1,26 @@
+import torch
+
+from cffi import FFI
+ffi = FFI()
+from ._ext.ctc_decode import ctc_beam_decode
+
+
+def beam_decode(probs, seq_len=None, top_paths=1, beam_width=10, merge_repeated=True):
+    prob_size = probs.size()
+    max_seq_len = prob_size[0]
+    batch_size = prob_size[1]
+    num_classes = prob_size[2]
+
+    if seq_len is not None and batch_size != seq_len.size():
+        raise ValueError("seq_len shape must be a 1xbatch_size tensor or None")
+    if top_paths < 1 or top_paths > beam_width:
+        raise ValueError("top_paths must be greater than 1 and less than or equal to the beam_width")
+
+    seq_len = torch.IntTensor(batch_size).zero_().add_(max_seq_len) if seq_len is None else seq_len
+    output = torch.IntTensor(top_paths, batch_size, max_seq_len)
+    scores = torch.FloatTensor(top_paths, batch_size)
+
+    merge_int = 1 if merge_repeated else 0
+    result = ctc_beam_decode(probs, seq_len, output, scores, top_paths, beam_width, merge_int)
+
+    return output, scores
diff --git a/src/cpu_binding.cpp → pytorch_ctc/src/cpu_binding.cpp b/src/cpu_binding.cpp → pytorch_ctc/src/cpu_binding.cpp
diff --git a/src/cpu_binding.h → pytorch_ctc/src/cpu_binding.h b/src/cpu_binding.h → pytorch_ctc/src/cpu_binding.h
diff --git a/src/ctc_beam_entry.h → pytorch_ctc/src/ctc_beam_entry.h b/src/ctc_beam_entry.h → pytorch_ctc/src/ctc_beam_entry.h
diff --git a/src/ctc_beam_scorer.h → pytorch_ctc/src/ctc_beam_scorer.h b/src/ctc_beam_scorer.h → pytorch_ctc/src/ctc_beam_scorer.h
diff --git a/src/ctc_beam_search.h → pytorch_ctc/src/ctc_beam_search.h b/src/ctc_beam_search.h → pytorch_ctc/src/ctc_beam_search.h
diff --git a/src/ctc_decoder.h → pytorch_ctc/src/ctc_decoder.h b/src/ctc_decoder.h → pytorch_ctc/src/ctc_decoder.h
diff --git a/src/ctc_loss_util.h → pytorch_ctc/src/ctc_loss_util.h b/src/ctc_loss_util.h → pytorch_ctc/src/ctc_loss_util.h
diff --git a/src/third_party/eigen3/Eigen/Core → ...rch_ctc/src/third_party/eigen3/Eigen/Core b/src/third_party/eigen3/Eigen/Core → ...rch_ctc/src/third_party/eigen3/Eigen/Core
diff --git a/...third_party/eigen3/Eigen/src/Core/Array.h → ...third_party/eigen3/Eigen/src/Core/Array.h b/...third_party/eigen3/Eigen/src/Core/Array.h → ...third_party/eigen3/Eigen/src/Core/Array.h
diff --git a/...d_party/eigen3/Eigen/src/Core/ArrayBase.h → ...d_party/eigen3/Eigen/src/Core/ArrayBase.h b/...d_party/eigen3/Eigen/src/Core/ArrayBase.h → ...d_party/eigen3/Eigen/src/Core/ArrayBase.h
diff --git a/...arty/eigen3/Eigen/src/Core/ArrayWrapper.h → ...arty/eigen3/Eigen/src/Core/ArrayWrapper.h b/...arty/eigen3/Eigen/src/Core/ArrayWrapper.h → ...arty/eigen3/Eigen/src/Core/ArrayWrapper.h
diff --git a/...hird_party/eigen3/Eigen/src/Core/Assign.h → ...hird_party/eigen3/Eigen/src/Core/Assign.h b/...hird_party/eigen3/Eigen/src/Core/Assign.h → ...hird_party/eigen3/Eigen/src/Core/Assign.h
diff --git a/...y/eigen3/Eigen/src/Core/AssignEvaluator.h → ...y/eigen3/Eigen/src/Core/AssignEvaluator.h b/...y/eigen3/Eigen/src/Core/AssignEvaluator.h → ...y/eigen3/Eigen/src/Core/AssignEvaluator.h
diff --git a/..._party/eigen3/Eigen/src/Core/Assign_MKL.h → ..._party/eigen3/Eigen/src/Core/Assign_MKL.h b/..._party/eigen3/Eigen/src/Core/Assign_MKL.h → ..._party/eigen3/Eigen/src/Core/Assign_MKL.h
diff --git a/..._party/eigen3/Eigen/src/Core/BandMatrix.h → ..._party/eigen3/Eigen/src/Core/BandMatrix.h b/..._party/eigen3/Eigen/src/Core/BandMatrix.h → ..._party/eigen3/Eigen/src/Core/BandMatrix.h
diff --git a/...third_party/eigen3/Eigen/src/Core/Block.h → ...third_party/eigen3/Eigen/src/Core/Block.h b/...third_party/eigen3/Eigen/src/Core/Block.h → ...third_party/eigen3/Eigen/src/Core/Block.h
diff --git a/...arty/eigen3/Eigen/src/Core/BooleanRedux.h → ...arty/eigen3/Eigen/src/Core/BooleanRedux.h b/...arty/eigen3/Eigen/src/Core/BooleanRedux.h → ...arty/eigen3/Eigen/src/Core/BooleanRedux.h
diff --git a/.../eigen3/Eigen/src/Core/CommaInitializer.h → .../eigen3/Eigen/src/Core/CommaInitializer.h b/.../eigen3/Eigen/src/Core/CommaInitializer.h → .../eigen3/Eigen/src/Core/CommaInitializer.h
diff --git a/...igen3/Eigen/src/Core/ConditionEstimator.h → ...igen3/Eigen/src/Core/ConditionEstimator.h b/...igen3/Eigen/src/Core/ConditionEstimator.h → ...igen3/Eigen/src/Core/ConditionEstimator.h
diff --git a/...ty/eigen3/Eigen/src/Core/CoreEvaluators.h → ...ty/eigen3/Eigen/src/Core/CoreEvaluators.h b/...ty/eigen3/Eigen/src/Core/CoreEvaluators.h → ...ty/eigen3/Eigen/src/Core/CoreEvaluators.h
diff --git a/...rty/eigen3/Eigen/src/Core/CoreIterators.h → ...rty/eigen3/Eigen/src/Core/CoreIterators.h b/...rty/eigen3/Eigen/src/Core/CoreIterators.h → ...rty/eigen3/Eigen/src/Core/CoreIterators.h
diff --git a/...rty/eigen3/Eigen/src/Core/CwiseBinaryOp.h → ...rty/eigen3/Eigen/src/Core/CwiseBinaryOp.h b/...rty/eigen3/Eigen/src/Core/CwiseBinaryOp.h → ...rty/eigen3/Eigen/src/Core/CwiseBinaryOp.h
diff --git a/...ty/eigen3/Eigen/src/Core/CwiseNullaryOp.h → ...ty/eigen3/Eigen/src/Core/CwiseNullaryOp.h b/...ty/eigen3/Eigen/src/Core/CwiseNullaryOp.h → ...ty/eigen3/Eigen/src/Core/CwiseNullaryOp.h
diff --git a/...ty/eigen3/Eigen/src/Core/CwiseTernaryOp.h → ...ty/eigen3/Eigen/src/Core/CwiseTernaryOp.h b/...ty/eigen3/Eigen/src/Core/CwiseTernaryOp.h → ...ty/eigen3/Eigen/src/Core/CwiseTernaryOp.h
diff --git a/...arty/eigen3/Eigen/src/Core/CwiseUnaryOp.h → ...arty/eigen3/Eigen/src/Core/CwiseUnaryOp.h b/...arty/eigen3/Eigen/src/Core/CwiseUnaryOp.h → ...arty/eigen3/Eigen/src/Core/CwiseUnaryOp.h
diff --git a/...ty/eigen3/Eigen/src/Core/CwiseUnaryView.h → ...ty/eigen3/Eigen/src/Core/CwiseUnaryView.h b/...ty/eigen3/Eigen/src/Core/CwiseUnaryView.h → ...ty/eigen3/Eigen/src/Core/CwiseUnaryView.h
diff --git a/...d_party/eigen3/Eigen/src/Core/DenseBase.h → ...d_party/eigen3/Eigen/src/Core/DenseBase.h b/...d_party/eigen3/Eigen/src/Core/DenseBase.h → ...d_party/eigen3/Eigen/src/Core/DenseBase.h
diff --git a/...y/eigen3/Eigen/src/Core/DenseCoeffsBase.h → ...y/eigen3/Eigen/src/Core/DenseCoeffsBase.h b/...y/eigen3/Eigen/src/Core/DenseCoeffsBase.h → ...y/eigen3/Eigen/src/Core/DenseCoeffsBase.h
diff --git a/...arty/eigen3/Eigen/src/Core/DenseStorage.h → ...arty/eigen3/Eigen/src/Core/DenseStorage.h b/...arty/eigen3/Eigen/src/Core/DenseStorage.h → ...arty/eigen3/Eigen/src/Core/DenseStorage.h
diff --git a/...rd_party/eigen3/Eigen/src/Core/Diagonal.h → ...rd_party/eigen3/Eigen/src/Core/Diagonal.h b/...rd_party/eigen3/Eigen/src/Core/Diagonal.h → ...rd_party/eigen3/Eigen/src/Core/Diagonal.h
diff --git a/...ty/eigen3/Eigen/src/Core/DiagonalMatrix.h → ...ty/eigen3/Eigen/src/Core/DiagonalMatrix.h b/...ty/eigen3/Eigen/src/Core/DiagonalMatrix.h → ...ty/eigen3/Eigen/src/Core/DiagonalMatrix.h
diff --git a/...y/eigen3/Eigen/src/Core/DiagonalProduct.h → ...y/eigen3/Eigen/src/Core/DiagonalProduct.h b/...y/eigen3/Eigen/src/Core/DiagonalProduct.h → ...y/eigen3/Eigen/src/Core/DiagonalProduct.h
diff --git a/src/third_party/eigen3/Eigen/src/Core/Dot.h → ...c/third_party/eigen3/Eigen/src/Core/Dot.h b/src/third_party/eigen3/Eigen/src/Core/Dot.h → ...c/third_party/eigen3/Eigen/src/Core/Dot.h
diff --git a/...d_party/eigen3/Eigen/src/Core/EigenBase.h → ...d_party/eigen3/Eigen/src/Core/EigenBase.h b/...d_party/eigen3/Eigen/src/Core/EigenBase.h → ...d_party/eigen3/Eigen/src/Core/EigenBase.h
diff --git a/...igen3/Eigen/src/Core/ForceAlignedAccess.h → ...igen3/Eigen/src/Core/ForceAlignedAccess.h b/...igen3/Eigen/src/Core/ForceAlignedAccess.h → ...igen3/Eigen/src/Core/ForceAlignedAccess.h
diff --git a/...third_party/eigen3/Eigen/src/Core/Fuzzy.h → ...third_party/eigen3/Eigen/src/Core/Fuzzy.h b/...third_party/eigen3/Eigen/src/Core/Fuzzy.h → ...third_party/eigen3/Eigen/src/Core/Fuzzy.h
diff --git a/...ty/eigen3/Eigen/src/Core/GeneralProduct.h → ...ty/eigen3/Eigen/src/Core/GeneralProduct.h b/...ty/eigen3/Eigen/src/Core/GeneralProduct.h → ...ty/eigen3/Eigen/src/Core/GeneralProduct.h
diff --git a/...eigen3/Eigen/src/Core/GenericPacketMath.h → ...eigen3/Eigen/src/Core/GenericPacketMath.h b/...eigen3/Eigen/src/Core/GenericPacketMath.h → ...eigen3/Eigen/src/Core/GenericPacketMath.h
diff --git a/...y/eigen3/Eigen/src/Core/GlobalFunctions.h → ...y/eigen3/Eigen/src/Core/GlobalFunctions.h b/...y/eigen3/Eigen/src/Core/GlobalFunctions.h → ...y/eigen3/Eigen/src/Core/GlobalFunctions.h
diff --git a/src/third_party/eigen3/Eigen/src/Core/IO.h → ...rc/third_party/eigen3/Eigen/src/Core/IO.h b/src/third_party/eigen3/Eigen/src/Core/IO.h → ...rc/third_party/eigen3/Eigen/src/Core/IO.h
diff --git a/...ird_party/eigen3/Eigen/src/Core/Inverse.h → ...ird_party/eigen3/Eigen/src/Core/Inverse.h b/...ird_party/eigen3/Eigen/src/Core/Inverse.h → ...ird_party/eigen3/Eigen/src/Core/Inverse.h
diff --git a/src/third_party/eigen3/Eigen/src/Core/Map.h → ...c/third_party/eigen3/Eigen/src/Core/Map.h b/src/third_party/eigen3/Eigen/src/Core/Map.h → ...c/third_party/eigen3/Eigen/src/Core/Map.h
diff --git a/...ird_party/eigen3/Eigen/src/Core/MapBase.h → ...ird_party/eigen3/Eigen/src/Core/MapBase.h b/...ird_party/eigen3/Eigen/src/Core/MapBase.h → ...ird_party/eigen3/Eigen/src/Core/MapBase.h
diff --git a/...rty/eigen3/Eigen/src/Core/MathFunctions.h → ...rty/eigen3/Eigen/src/Core/MathFunctions.h b/...rty/eigen3/Eigen/src/Core/MathFunctions.h → ...rty/eigen3/Eigen/src/Core/MathFunctions.h
diff --git a/...eigen3/Eigen/src/Core/MathFunctionsImpl.h → ...eigen3/Eigen/src/Core/MathFunctionsImpl.h b/...eigen3/Eigen/src/Core/MathFunctionsImpl.h → ...eigen3/Eigen/src/Core/MathFunctionsImpl.h
diff --git a/...hird_party/eigen3/Eigen/src/Core/Matrix.h → ...hird_party/eigen3/Eigen/src/Core/Matrix.h b/...hird_party/eigen3/Eigen/src/Core/Matrix.h → ...hird_party/eigen3/Eigen/src/Core/Matrix.h
diff --git a/..._party/eigen3/Eigen/src/Core/MatrixBase.h → ..._party/eigen3/Eigen/src/Core/MatrixBase.h b/..._party/eigen3/Eigen/src/Core/MatrixBase.h → ..._party/eigen3/Eigen/src/Core/MatrixBase.h
diff --git a/...party/eigen3/Eigen/src/Core/NestByValue.h → ...party/eigen3/Eigen/src/Core/NestByValue.h b/...party/eigen3/Eigen/src/Core/NestByValue.h → ...party/eigen3/Eigen/src/Core/NestByValue.h
diff --git a/...ird_party/eigen3/Eigen/src/Core/NoAlias.h → ...ird_party/eigen3/Eigen/src/Core/NoAlias.h b/...ird_party/eigen3/Eigen/src/Core/NoAlias.h → ...ird_party/eigen3/Eigen/src/Core/NoAlias.h
diff --git a/...d_party/eigen3/Eigen/src/Core/NumTraits.h → ...d_party/eigen3/Eigen/src/Core/NumTraits.h b/...d_party/eigen3/Eigen/src/Core/NumTraits.h → ...d_party/eigen3/Eigen/src/Core/NumTraits.h
diff --git a/...eigen3/Eigen/src/Core/PermutationMatrix.h → ...eigen3/Eigen/src/Core/PermutationMatrix.h b/...eigen3/Eigen/src/Core/PermutationMatrix.h → ...eigen3/Eigen/src/Core/PermutationMatrix.h
diff --git a/...y/eigen3/Eigen/src/Core/PlainObjectBase.h → ...y/eigen3/Eigen/src/Core/PlainObjectBase.h b/...y/eigen3/Eigen/src/Core/PlainObjectBase.h → ...y/eigen3/Eigen/src/Core/PlainObjectBase.h
diff --git a/...ird_party/eigen3/Eigen/src/Core/Product.h → ...ird_party/eigen3/Eigen/src/Core/Product.h b/...ird_party/eigen3/Eigen/src/Core/Product.h → ...ird_party/eigen3/Eigen/src/Core/Product.h
diff --git a/...eigen3/Eigen/src/Core/ProductEvaluators.h → ...eigen3/Eigen/src/Core/ProductEvaluators.h b/...eigen3/Eigen/src/Core/ProductEvaluators.h → ...eigen3/Eigen/src/Core/ProductEvaluators.h
diff --git a/...hird_party/eigen3/Eigen/src/Core/Random.h → ...hird_party/eigen3/Eigen/src/Core/Random.h b/...hird_party/eigen3/Eigen/src/Core/Random.h → ...hird_party/eigen3/Eigen/src/Core/Random.h
diff --git a/...third_party/eigen3/Eigen/src/Core/Redux.h → ...third_party/eigen3/Eigen/src/Core/Redux.h b/...third_party/eigen3/Eigen/src/Core/Redux.h → ...third_party/eigen3/Eigen/src/Core/Redux.h
diff --git a/src/third_party/eigen3/Eigen/src/Core/Ref.h → ...c/third_party/eigen3/Eigen/src/Core/Ref.h b/src/third_party/eigen3/Eigen/src/Core/Ref.h → ...c/third_party/eigen3/Eigen/src/Core/Ref.h
diff --git a/...d_party/eigen3/Eigen/src/Core/Replicate.h → ...d_party/eigen3/Eigen/src/Core/Replicate.h b/...d_party/eigen3/Eigen/src/Core/Replicate.h → ...d_party/eigen3/Eigen/src/Core/Replicate.h
diff --git a/...rty/eigen3/Eigen/src/Core/ReturnByValue.h → ...rty/eigen3/Eigen/src/Core/ReturnByValue.h b/...rty/eigen3/Eigen/src/Core/ReturnByValue.h → ...rty/eigen3/Eigen/src/Core/ReturnByValue.h
diff --git a/...ird_party/eigen3/Eigen/src/Core/Reverse.h → ...ird_party/eigen3/Eigen/src/Core/Reverse.h b/...ird_party/eigen3/Eigen/src/Core/Reverse.h → ...ird_party/eigen3/Eigen/src/Core/Reverse.h
diff --git a/...hird_party/eigen3/Eigen/src/Core/Select.h → ...hird_party/eigen3/Eigen/src/Core/Select.h b/...hird_party/eigen3/Eigen/src/Core/Select.h → ...hird_party/eigen3/Eigen/src/Core/Select.h
diff --git a/...y/eigen3/Eigen/src/Core/SelfAdjointView.h → ...y/eigen3/Eigen/src/Core/SelfAdjointView.h b/...y/eigen3/Eigen/src/Core/SelfAdjointView.h → ...y/eigen3/Eigen/src/Core/SelfAdjointView.h
diff --git a/...eigen3/Eigen/src/Core/SelfCwiseBinaryOp.h → ...eigen3/Eigen/src/Core/SelfCwiseBinaryOp.h b/...eigen3/Eigen/src/Core/SelfCwiseBinaryOp.h → ...eigen3/Eigen/src/Core/SelfCwiseBinaryOp.h
diff --git a/...third_party/eigen3/Eigen/src/Core/Solve.h → ...third_party/eigen3/Eigen/src/Core/Solve.h b/...third_party/eigen3/Eigen/src/Core/Solve.h → ...third_party/eigen3/Eigen/src/Core/Solve.h
diff --git a/...y/eigen3/Eigen/src/Core/SolveTriangular.h → ...y/eigen3/Eigen/src/Core/SolveTriangular.h b/...y/eigen3/Eigen/src/Core/SolveTriangular.h → ...y/eigen3/Eigen/src/Core/SolveTriangular.h
diff --git a/..._party/eigen3/Eigen/src/Core/SolverBase.h → ..._party/eigen3/Eigen/src/Core/SolverBase.h b/..._party/eigen3/Eigen/src/Core/SolverBase.h → ..._party/eigen3/Eigen/src/Core/SolverBase.h
diff --git a/..._party/eigen3/Eigen/src/Core/StableNorm.h → ..._party/eigen3/Eigen/src/Core/StableNorm.h b/..._party/eigen3/Eigen/src/Core/StableNorm.h → ..._party/eigen3/Eigen/src/Core/StableNorm.h
diff --git a/...hird_party/eigen3/Eigen/src/Core/Stride.h → ...hird_party/eigen3/Eigen/src/Core/Stride.h b/...hird_party/eigen3/Eigen/src/Core/Stride.h → ...hird_party/eigen3/Eigen/src/Core/Stride.h
diff --git a/src/third_party/eigen3/Eigen/src/Core/Swap.h → .../third_party/eigen3/Eigen/src/Core/Swap.h b/src/third_party/eigen3/Eigen/src/Core/Swap.h → .../third_party/eigen3/Eigen/src/Core/Swap.h
diff --git a/...d_party/eigen3/Eigen/src/Core/Transpose.h → ...d_party/eigen3/Eigen/src/Core/Transpose.h b/...d_party/eigen3/Eigen/src/Core/Transpose.h → ...d_party/eigen3/Eigen/src/Core/Transpose.h
diff --git a/...ty/eigen3/Eigen/src/Core/Transpositions.h → ...ty/eigen3/Eigen/src/Core/Transpositions.h b/...ty/eigen3/Eigen/src/Core/Transpositions.h → ...ty/eigen3/Eigen/src/Core/Transpositions.h
diff --git a/.../eigen3/Eigen/src/Core/TriangularMatrix.h → .../eigen3/Eigen/src/Core/TriangularMatrix.h b/.../eigen3/Eigen/src/Core/TriangularMatrix.h → .../eigen3/Eigen/src/Core/TriangularMatrix.h
diff --git a/...party/eigen3/Eigen/src/Core/VectorBlock.h → ...party/eigen3/Eigen/src/Core/VectorBlock.h b/...party/eigen3/Eigen/src/Core/VectorBlock.h → ...party/eigen3/Eigen/src/Core/VectorBlock.h
diff --git a/...arty/eigen3/Eigen/src/Core/VectorwiseOp.h → ...arty/eigen3/Eigen/src/Core/VectorwiseOp.h b/...arty/eigen3/Eigen/src/Core/VectorwiseOp.h → ...arty/eigen3/Eigen/src/Core/VectorwiseOp.h
diff --git a/...ird_party/eigen3/Eigen/src/Core/Visitor.h → ...ird_party/eigen3/Eigen/src/Core/Visitor.h b/...ird_party/eigen3/Eigen/src/Core/Visitor.h → ...ird_party/eigen3/Eigen/src/Core/Visitor.h
diff --git a/.../eigen3/Eigen/src/Core/arch/AVX/Complex.h → .../eigen3/Eigen/src/Core/arch/AVX/Complex.h b/.../eigen3/Eigen/src/Core/arch/AVX/Complex.h → .../eigen3/Eigen/src/Core/arch/AVX/Complex.h
diff --git a/...3/Eigen/src/Core/arch/AVX/MathFunctions.h → ...3/Eigen/src/Core/arch/AVX/MathFunctions.h b/...3/Eigen/src/Core/arch/AVX/MathFunctions.h → ...3/Eigen/src/Core/arch/AVX/MathFunctions.h
diff --git a/...gen3/Eigen/src/Core/arch/AVX/PacketMath.h → ...gen3/Eigen/src/Core/arch/AVX/PacketMath.h b/...gen3/Eigen/src/Core/arch/AVX/PacketMath.h → ...gen3/Eigen/src/Core/arch/AVX/PacketMath.h
diff --git a/...en3/Eigen/src/Core/arch/AVX/TypeCasting.h → ...en3/Eigen/src/Core/arch/AVX/TypeCasting.h b/...en3/Eigen/src/Core/arch/AVX/TypeCasting.h → ...en3/Eigen/src/Core/arch/AVX/TypeCasting.h
diff --git a/...igen/src/Core/arch/AVX512/MathFunctions.h → ...igen/src/Core/arch/AVX512/MathFunctions.h b/...igen/src/Core/arch/AVX512/MathFunctions.h → ...igen/src/Core/arch/AVX512/MathFunctions.h
diff --git a/...3/Eigen/src/Core/arch/AVX512/PacketMath.h → ...3/Eigen/src/Core/arch/AVX512/PacketMath.h b/...3/Eigen/src/Core/arch/AVX512/PacketMath.h → ...3/Eigen/src/Core/arch/AVX512/PacketMath.h
diff --git a/...en3/Eigen/src/Core/arch/AltiVec/Complex.h → ...en3/Eigen/src/Core/arch/AltiVec/Complex.h b/...en3/Eigen/src/Core/arch/AltiVec/Complex.h → ...en3/Eigen/src/Core/arch/AltiVec/Complex.h
diff --git a/...gen/src/Core/arch/AltiVec/MathFunctions.h → ...gen/src/Core/arch/AltiVec/MathFunctions.h b/...gen/src/Core/arch/AltiVec/MathFunctions.h → ...gen/src/Core/arch/AltiVec/MathFunctions.h
diff --git a/.../Eigen/src/Core/arch/AltiVec/PacketMath.h → .../Eigen/src/Core/arch/AltiVec/PacketMath.h b/.../Eigen/src/Core/arch/AltiVec/PacketMath.h → .../Eigen/src/Core/arch/AltiVec/PacketMath.h
diff --git a/...eigen3/Eigen/src/Core/arch/CUDA/Complex.h → ...eigen3/Eigen/src/Core/arch/CUDA/Complex.h b/...eigen3/Eigen/src/Core/arch/CUDA/Complex.h → ...eigen3/Eigen/src/Core/arch/CUDA/Complex.h
diff --git a/...ty/eigen3/Eigen/src/Core/arch/CUDA/Half.h → ...ty/eigen3/Eigen/src/Core/arch/CUDA/Half.h b/...ty/eigen3/Eigen/src/Core/arch/CUDA/Half.h → ...ty/eigen3/Eigen/src/Core/arch/CUDA/Half.h
diff --git a/.../Eigen/src/Core/arch/CUDA/MathFunctions.h → .../Eigen/src/Core/arch/CUDA/MathFunctions.h b/.../Eigen/src/Core/arch/CUDA/MathFunctions.h → .../Eigen/src/Core/arch/CUDA/MathFunctions.h
diff --git a/...en3/Eigen/src/Core/arch/CUDA/PacketMath.h → ...en3/Eigen/src/Core/arch/CUDA/PacketMath.h b/...en3/Eigen/src/Core/arch/CUDA/PacketMath.h → ...en3/Eigen/src/Core/arch/CUDA/PacketMath.h
diff --git a/...Eigen/src/Core/arch/CUDA/PacketMathHalf.h → ...Eigen/src/Core/arch/CUDA/PacketMathHalf.h b/...Eigen/src/Core/arch/CUDA/PacketMathHalf.h → ...Eigen/src/Core/arch/CUDA/PacketMathHalf.h
diff --git a/...n3/Eigen/src/Core/arch/CUDA/TypeCasting.h → ...n3/Eigen/src/Core/arch/CUDA/TypeCasting.h b/...n3/Eigen/src/Core/arch/CUDA/TypeCasting.h → ...n3/Eigen/src/Core/arch/CUDA/TypeCasting.h
diff --git a/...n3/Eigen/src/Core/arch/Default/Settings.h → ...n3/Eigen/src/Core/arch/Default/Settings.h b/...n3/Eigen/src/Core/arch/Default/Settings.h → ...n3/Eigen/src/Core/arch/Default/Settings.h
diff --git a/...eigen3/Eigen/src/Core/arch/NEON/Complex.h → ...eigen3/Eigen/src/Core/arch/NEON/Complex.h b/...eigen3/Eigen/src/Core/arch/NEON/Complex.h → ...eigen3/Eigen/src/Core/arch/NEON/Complex.h
diff --git a/.../Eigen/src/Core/arch/NEON/MathFunctions.h → .../Eigen/src/Core/arch/NEON/MathFunctions.h b/.../Eigen/src/Core/arch/NEON/MathFunctions.h → .../Eigen/src/Core/arch/NEON/MathFunctions.h
diff --git a/...en3/Eigen/src/Core/arch/NEON/PacketMath.h → ...en3/Eigen/src/Core/arch/NEON/PacketMath.h b/...en3/Eigen/src/Core/arch/NEON/PacketMath.h → ...en3/Eigen/src/Core/arch/NEON/PacketMath.h
diff --git a/.../eigen3/Eigen/src/Core/arch/SSE/Complex.h → .../eigen3/Eigen/src/Core/arch/SSE/Complex.h b/.../eigen3/Eigen/src/Core/arch/SSE/Complex.h → .../eigen3/Eigen/src/Core/arch/SSE/Complex.h
diff --git a/...3/Eigen/src/Core/arch/SSE/MathFunctions.h → ...3/Eigen/src/Core/arch/SSE/MathFunctions.h b/...3/Eigen/src/Core/arch/SSE/MathFunctions.h → ...3/Eigen/src/Core/arch/SSE/MathFunctions.h
diff --git a/...gen3/Eigen/src/Core/arch/SSE/PacketMath.h → ...gen3/Eigen/src/Core/arch/SSE/PacketMath.h b/...gen3/Eigen/src/Core/arch/SSE/PacketMath.h → ...gen3/Eigen/src/Core/arch/SSE/PacketMath.h
diff --git a/...en3/Eigen/src/Core/arch/SSE/TypeCasting.h → ...en3/Eigen/src/Core/arch/SSE/TypeCasting.h b/...en3/Eigen/src/Core/arch/SSE/TypeCasting.h → ...en3/Eigen/src/Core/arch/SSE/TypeCasting.h
diff --git a/...en3/Eigen/src/Core/arch/ZVector/Complex.h → ...en3/Eigen/src/Core/arch/ZVector/Complex.h b/...en3/Eigen/src/Core/arch/ZVector/Complex.h → ...en3/Eigen/src/Core/arch/ZVector/Complex.h
diff --git a/...gen/src/Core/arch/ZVector/MathFunctions.h → ...gen/src/Core/arch/ZVector/MathFunctions.h b/...gen/src/Core/arch/ZVector/MathFunctions.h → ...gen/src/Core/arch/ZVector/MathFunctions.h
diff --git a/.../Eigen/src/Core/arch/ZVector/PacketMath.h → .../Eigen/src/Core/arch/ZVector/PacketMath.h b/.../Eigen/src/Core/arch/ZVector/PacketMath.h → .../Eigen/src/Core/arch/ZVector/PacketMath.h
diff --git a/...en/src/Core/functors/AssignmentFunctors.h → ...en/src/Core/functors/AssignmentFunctors.h b/...en/src/Core/functors/AssignmentFunctors.h → ...en/src/Core/functors/AssignmentFunctors.h
diff --git a/.../Eigen/src/Core/functors/BinaryFunctors.h → .../Eigen/src/Core/functors/BinaryFunctors.h b/.../Eigen/src/Core/functors/BinaryFunctors.h → .../Eigen/src/Core/functors/BinaryFunctors.h
diff --git a/...Eigen/src/Core/functors/NullaryFunctors.h → ...Eigen/src/Core/functors/NullaryFunctors.h b/...Eigen/src/Core/functors/NullaryFunctors.h → ...Eigen/src/Core/functors/NullaryFunctors.h
diff --git a/...en3/Eigen/src/Core/functors/StlFunctors.h → ...en3/Eigen/src/Core/functors/StlFunctors.h b/...en3/Eigen/src/Core/functors/StlFunctors.h → ...en3/Eigen/src/Core/functors/StlFunctors.h
diff --git a/...Eigen/src/Core/functors/TernaryFunctors.h → ...Eigen/src/Core/functors/TernaryFunctors.h b/...Eigen/src/Core/functors/TernaryFunctors.h → ...Eigen/src/Core/functors/TernaryFunctors.h
diff --git a/...3/Eigen/src/Core/functors/UnaryFunctors.h → ...3/Eigen/src/Core/functors/UnaryFunctors.h b/...3/Eigen/src/Core/functors/UnaryFunctors.h → ...3/Eigen/src/Core/functors/UnaryFunctors.h
diff --git a/...c/Core/products/GeneralBlockPanelKernel.h → ...c/Core/products/GeneralBlockPanelKernel.h b/...c/Core/products/GeneralBlockPanelKernel.h → ...c/Core/products/GeneralBlockPanelKernel.h
diff --git a/...n/src/Core/products/GeneralMatrixMatrix.h → ...n/src/Core/products/GeneralMatrixMatrix.h b/...n/src/Core/products/GeneralMatrixMatrix.h → ...n/src/Core/products/GeneralMatrixMatrix.h
diff --git a/.../products/GeneralMatrixMatrixTriangular.h → .../products/GeneralMatrixMatrixTriangular.h b/.../products/GeneralMatrixMatrixTriangular.h → .../products/GeneralMatrixMatrixTriangular.h
diff --git a/...ucts/GeneralMatrixMatrixTriangular_BLAS.h → ...ucts/GeneralMatrixMatrixTriangular_BLAS.h b/...ucts/GeneralMatrixMatrixTriangular_BLAS.h → ...ucts/GeneralMatrixMatrixTriangular_BLAS.h
diff --git a/.../Core/products/GeneralMatrixMatrix_BLAS.h → .../Core/products/GeneralMatrixMatrix_BLAS.h b/.../Core/products/GeneralMatrixMatrix_BLAS.h → .../Core/products/GeneralMatrixMatrix_BLAS.h
diff --git a/...n/src/Core/products/GeneralMatrixVector.h → ...n/src/Core/products/GeneralMatrixVector.h b/...n/src/Core/products/GeneralMatrixVector.h → ...n/src/Core/products/GeneralMatrixVector.h
diff --git a/.../Core/products/GeneralMatrixVector_BLAS.h → .../Core/products/GeneralMatrixVector_BLAS.h b/.../Core/products/GeneralMatrixVector_BLAS.h → .../Core/products/GeneralMatrixVector_BLAS.h
diff --git a/...n3/Eigen/src/Core/products/Parallelizer.h → ...n3/Eigen/src/Core/products/Parallelizer.h b/...n3/Eigen/src/Core/products/Parallelizer.h → ...n3/Eigen/src/Core/products/Parallelizer.h
diff --git a/...c/Core/products/SelfadjointMatrixMatrix.h → ...c/Core/products/SelfadjointMatrixMatrix.h b/...c/Core/products/SelfadjointMatrixMatrix.h → ...c/Core/products/SelfadjointMatrixMatrix.h
diff --git a/...e/products/SelfadjointMatrixMatrix_BLAS.h → ...e/products/SelfadjointMatrixMatrix_BLAS.h b/...e/products/SelfadjointMatrixMatrix_BLAS.h → ...e/products/SelfadjointMatrixMatrix_BLAS.h
diff --git a/...c/Core/products/SelfadjointMatrixVector.h → ...c/Core/products/SelfadjointMatrixVector.h b/...c/Core/products/SelfadjointMatrixVector.h → ...c/Core/products/SelfadjointMatrixVector.h
diff --git a/...e/products/SelfadjointMatrixVector_BLAS.h → ...e/products/SelfadjointMatrixVector_BLAS.h b/...e/products/SelfadjointMatrixVector_BLAS.h → ...e/products/SelfadjointMatrixVector_BLAS.h
diff --git a/...en/src/Core/products/SelfadjointProduct.h → ...en/src/Core/products/SelfadjointProduct.h b/...en/src/Core/products/SelfadjointProduct.h → ...en/src/Core/products/SelfadjointProduct.h
diff --git a/...rc/Core/products/SelfadjointRank2Update.h → ...rc/Core/products/SelfadjointRank2Update.h b/...rc/Core/products/SelfadjointRank2Update.h → ...rc/Core/products/SelfadjointRank2Update.h
diff --git a/...rc/Core/products/TriangularMatrixMatrix.h → ...rc/Core/products/TriangularMatrixMatrix.h b/...rc/Core/products/TriangularMatrixMatrix.h → ...rc/Core/products/TriangularMatrixMatrix.h
diff --git a/...re/products/TriangularMatrixMatrix_BLAS.h → ...re/products/TriangularMatrixMatrix_BLAS.h b/...re/products/TriangularMatrixMatrix_BLAS.h → ...re/products/TriangularMatrixMatrix_BLAS.h
diff --git a/...rc/Core/products/TriangularMatrixVector.h → ...rc/Core/products/TriangularMatrixVector.h b/...rc/Core/products/TriangularMatrixVector.h → ...rc/Core/products/TriangularMatrixVector.h
diff --git a/...re/products/TriangularMatrixVector_BLAS.h → ...re/products/TriangularMatrixVector_BLAS.h b/...re/products/TriangularMatrixVector_BLAS.h → ...re/products/TriangularMatrixVector_BLAS.h
diff --git a/...rc/Core/products/TriangularSolverMatrix.h → ...rc/Core/products/TriangularSolverMatrix.h b/...rc/Core/products/TriangularSolverMatrix.h → ...rc/Core/products/TriangularSolverMatrix.h
diff --git a/...re/products/TriangularSolverMatrix_BLAS.h → ...re/products/TriangularSolverMatrix_BLAS.h b/...re/products/TriangularSolverMatrix_BLAS.h → ...re/products/TriangularSolverMatrix_BLAS.h
diff --git a/...rc/Core/products/TriangularSolverVector.h → ...rc/Core/products/TriangularSolverVector.h b/...rc/Core/products/TriangularSolverVector.h → ...rc/Core/products/TriangularSolverVector.h
diff --git a/...rty/eigen3/Eigen/src/Core/util/BlasUtil.h → ...rty/eigen3/Eigen/src/Core/util/BlasUtil.h b/...rty/eigen3/Eigen/src/Core/util/BlasUtil.h → ...rty/eigen3/Eigen/src/Core/util/BlasUtil.h
diff --git a/...ty/eigen3/Eigen/src/Core/util/Constants.h → ...ty/eigen3/Eigen/src/Core/util/Constants.h b/...ty/eigen3/Eigen/src/Core/util/Constants.h → ...ty/eigen3/Eigen/src/Core/util/Constants.h
diff --git a/...gen/src/Core/util/DisableStupidWarnings.h → ...gen/src/Core/util/DisableStupidWarnings.h b/...gen/src/Core/util/DisableStupidWarnings.h → ...gen/src/Core/util/DisableStupidWarnings.h
diff --git a/...Eigen/src/Core/util/ForwardDeclarations.h → ...Eigen/src/Core/util/ForwardDeclarations.h b/...Eigen/src/Core/util/ForwardDeclarations.h → ...Eigen/src/Core/util/ForwardDeclarations.h
diff --git a/.../eigen3/Eigen/src/Core/util/MKL_support.h → .../eigen3/Eigen/src/Core/util/MKL_support.h b/.../eigen3/Eigen/src/Core/util/MKL_support.h → .../eigen3/Eigen/src/Core/util/MKL_support.h
diff --git a/...party/eigen3/Eigen/src/Core/util/Macros.h → ...party/eigen3/Eigen/src/Core/util/Macros.h b/...party/eigen3/Eigen/src/Core/util/Macros.h → ...party/eigen3/Eigen/src/Core/util/Macros.h
diff --git a/...party/eigen3/Eigen/src/Core/util/Memory.h → ...party/eigen3/Eigen/src/Core/util/Memory.h b/...party/eigen3/Eigen/src/Core/util/Memory.h → ...party/eigen3/Eigen/src/Core/util/Memory.h
diff --git a/...d_party/eigen3/Eigen/src/Core/util/Meta.h → ...d_party/eigen3/Eigen/src/Core/util/Meta.h b/...d_party/eigen3/Eigen/src/Core/util/Meta.h → ...d_party/eigen3/Eigen/src/Core/util/Meta.h
diff --git a/...arty/eigen3/Eigen/src/Core/util/NonMPL2.h → ...arty/eigen3/Eigen/src/Core/util/NonMPL2.h b/...arty/eigen3/Eigen/src/Core/util/NonMPL2.h → ...arty/eigen3/Eigen/src/Core/util/NonMPL2.h
diff --git a/...en/src/Core/util/ReenableStupidWarnings.h → ...en/src/Core/util/ReenableStupidWarnings.h b/...en/src/Core/util/ReenableStupidWarnings.h → ...en/src/Core/util/ReenableStupidWarnings.h
diff --git a/...eigen3/Eigen/src/Core/util/StaticAssert.h → ...eigen3/Eigen/src/Core/util/StaticAssert.h b/...eigen3/Eigen/src/Core/util/StaticAssert.h → ...eigen3/Eigen/src/Core/util/StaticAssert.h
diff --git a/...ty/eigen3/Eigen/src/Core/util/XprHelper.h → ...ty/eigen3/Eigen/src/Core/util/XprHelper.h b/...ty/eigen3/Eigen/src/Core/util/XprHelper.h → ...ty/eigen3/Eigen/src/Core/util/XprHelper.h
diff --git a/...3/Eigen/src/plugins/ArrayCwiseBinaryOps.h → ...3/Eigen/src/plugins/ArrayCwiseBinaryOps.h b/...3/Eigen/src/plugins/ArrayCwiseBinaryOps.h → ...3/Eigen/src/plugins/ArrayCwiseBinaryOps.h
diff --git a/...n3/Eigen/src/plugins/ArrayCwiseUnaryOps.h → ...n3/Eigen/src/plugins/ArrayCwiseUnaryOps.h b/...n3/Eigen/src/plugins/ArrayCwiseUnaryOps.h → ...n3/Eigen/src/plugins/ArrayCwiseUnaryOps.h
diff --git a/...y/eigen3/Eigen/src/plugins/BlockMethods.h → ...y/eigen3/Eigen/src/plugins/BlockMethods.h b/...y/eigen3/Eigen/src/plugins/BlockMethods.h → ...y/eigen3/Eigen/src/plugins/BlockMethods.h
diff --git a/.../Eigen/src/plugins/CommonCwiseBinaryOps.h → .../Eigen/src/plugins/CommonCwiseBinaryOps.h b/.../Eigen/src/plugins/CommonCwiseBinaryOps.h → .../Eigen/src/plugins/CommonCwiseBinaryOps.h
diff --git a/...3/Eigen/src/plugins/CommonCwiseUnaryOps.h → ...3/Eigen/src/plugins/CommonCwiseUnaryOps.h b/...3/Eigen/src/plugins/CommonCwiseUnaryOps.h → ...3/Eigen/src/plugins/CommonCwiseUnaryOps.h
diff --git a/.../Eigen/src/plugins/MatrixCwiseBinaryOps.h → .../Eigen/src/plugins/MatrixCwiseBinaryOps.h b/.../Eigen/src/plugins/MatrixCwiseBinaryOps.h → .../Eigen/src/plugins/MatrixCwiseBinaryOps.h
diff --git a/...3/Eigen/src/plugins/MatrixCwiseUnaryOps.h → ...3/Eigen/src/plugins/MatrixCwiseUnaryOps.h b/...3/Eigen/src/plugins/MatrixCwiseUnaryOps.h → ...3/Eigen/src/plugins/MatrixCwiseUnaryOps.h
diff --git a/src/util/macros.h → pytorch_ctc/src/util/macros.h b/src/util/macros.h → pytorch_ctc/src/util/macros.h
diff --git a/src/util/status.cpp → pytorch_ctc/src/util/status.cpp b/src/util/status.cpp → pytorch_ctc/src/util/status.cpp
diff --git a/src/util/status.h → pytorch_ctc/src/util/status.h b/src/util/status.h → pytorch_ctc/src/util/status.h
diff --git a/src/util/top_n.h → pytorch_ctc/src/util/top_n.h b/src/util/top_n.h → pytorch_ctc/src/util/top_n.h
diff --git a/setup.py b/setup.py
@@ -6,19 +6,19 @@
 from distutils.core import setup
 from torch.utils.ffi import create_extension
 
-sources = ['src/cpu_binding.cpp', 'src/util/status.cpp']
-headers = ['src/cpu_binding.h']
+sources = ['pytorch_ctc/src/cpu_binding.cpp', 'pytorch_ctc/src/util/status.cpp']
+headers = ['pytorch_ctc/src/cpu_binding.h']
 
 ffi = create_extension(
-    name='ctc_decode',
+    name='pytorch_ctc._ext.ctc_decode',
+    package=True,
     language='c++',
     headers=headers,
     sources=sources,
     with_cuda=False,
     extra_compile_args=['-std=c++11', '-fPIC', '-w']
 )
 ffi = ffi.distutils_extension()
-ffi.name = 'pytorch_ctc._ctc_decode'
 
 setup(
     name="pytorch_ctc",