README.md
requirements-dev.txt
requirements.txt
setup.py
version.txt
benchmark/__init__.py
benchmark/benchmark_qconv_cuda_kernel.py
benchmark/benchmark_qconv_cutlass_4-bit.py
benchmark/benchmark_qembeddingbag_cuda.py
benchmark/benchmark_qlinear_cuda_kernel.py
benchmark/benchmark_qlinear_cutlass_4-bit.py
benchmark/util.py
bitorch_engine/__init__.py
bitorch_engine.egg-info/PKG-INFO
bitorch_engine.egg-info/SOURCES.txt
bitorch_engine.egg-info/dependency_links.txt
bitorch_engine.egg-info/requires.txt
bitorch_engine.egg-info/top_level.txt
bitorch_engine/extensions/__init__.py
bitorch_engine/functions/__init__.py
bitorch_engine/functions/cuda/__init__.py
bitorch_engine/functions/cuda/extension.py
bitorch_engine/functions/cuda/functions.py
bitorch_engine/layers/__init__.py
bitorch_engine/layers/qconv/__init__.py
bitorch_engine/layers/qconv/layer.py
bitorch_engine/layers/qconv/binary/__init__.py
bitorch_engine/layers/qconv/binary/layer.py
bitorch_engine/layers/qconv/binary/cpp/__init__.py
bitorch_engine/layers/qconv/binary/cpp/binary_conv.cpp
bitorch_engine/layers/qconv/binary/cpp/extension.py
bitorch_engine/layers/qconv/binary/cpp/layer.py
bitorch_engine/layers/qconv/binary/cuda/__init__.py
bitorch_engine/layers/qconv/binary/cuda/bcnv.py
bitorch_engine/layers/qconv/binary/cuda/extension.py
bitorch_engine/layers/qconv/binary/cuda/layer.py
bitorch_engine/layers/qconv/nbit/__init__.py
bitorch_engine/layers/qconv/nbit/layer.py
bitorch_engine/layers/qconv/nbit/cutlass/__init__.py
bitorch_engine/layers/qconv/nbit/cutlass/extension.py
bitorch_engine/layers/qconv/nbit/cutlass/layer.py
bitorch_engine/layers/qembedding/__init__.py
bitorch_engine/layers/qembedding/layer.py
bitorch_engine/layers/qembedding/binary/__init__.py
bitorch_engine/layers/qembedding/binary/layer.py
bitorch_engine/layers/qembedding/binary/cuda/__init__.py
bitorch_engine/layers/qembedding/binary/cuda/extension.py
bitorch_engine/layers/qembedding/binary/cuda/layer.py
bitorch_engine/layers/qlinear/__init__.py
bitorch_engine/layers/qlinear/layer.py
bitorch_engine/layers/qlinear/qlinear_implementation.py
bitorch_engine/layers/qlinear/binary/__init__.py
bitorch_engine/layers/qlinear/binary/binary_implementation.py
bitorch_engine/layers/qlinear/binary/layer.py
bitorch_engine/layers/qlinear/binary/cpp/__init__.py
bitorch_engine/layers/qlinear/binary/cpp/binary_linear.cpp
bitorch_engine/layers/qlinear/binary/cpp/extension.py
bitorch_engine/layers/qlinear/binary/cpp/layer.py
bitorch_engine/layers/qlinear/binary/cuda/__init__.py
bitorch_engine/layers/qlinear/binary/cuda/bmm.py
bitorch_engine/layers/qlinear/binary/cuda/extension.py
bitorch_engine/layers/qlinear/binary/cuda/layer.py
bitorch_engine/layers/qlinear/binary/cutlass/__init__.py
bitorch_engine/layers/qlinear/binary/cutlass/extension.py
bitorch_engine/layers/qlinear/binary/cutlass/layer.py
bitorch_engine/layers/qlinear/nbit/__init__.py
bitorch_engine/layers/qlinear/nbit/layer.py
bitorch_engine/layers/qlinear/nbit/cutlass/__init__.py
bitorch_engine/layers/qlinear/nbit/cutlass/extension.py
bitorch_engine/layers/qlinear/nbit/cutlass/layer.py
bitorch_engine/utils/__init__.py
bitorch_engine/utils/cpp_extension.py
bitorch_engine/utils/cuda_extension.py
bitorch_engine/utils/cutlass_path.py
bitorch_engine/utils/quant_operators.py
bitorch_engine/utils/safe_import.py
tests/__init__.py
tests/functions/__init__.py
tests/functions/test_quant_ops.py
tests/layers/__init__.py
tests/layers/test_binary_conv.py
tests/layers/test_binary_embeddingbag.py
tests/layers/test_binary_linear.py
tests/layers/test_integration.py
tests/layers/test_nbit_conv.py
tests/layers/test_nbit_linear.py
tests/layers/util.py