LICENSE
README.md
setup.py
csrc/python_api.cpp
deep_gemm/__init__.py
deep_gemm.egg-info/PKG-INFO
deep_gemm.egg-info/SOURCES.txt
deep_gemm.egg-info/dependency_links.txt
deep_gemm.egg-info/not-zip-safe
deep_gemm.egg-info/top_level.txt
deep_gemm/include/deep_gemm/common/cute_tie.cuh
deep_gemm/include/deep_gemm/common/epilogue_utils.cuh
deep_gemm/include/deep_gemm/common/reduction.cuh
deep_gemm/include/deep_gemm/common/scheduler.cuh
deep_gemm/include/deep_gemm/common/sm100_utils.cuh
deep_gemm/include/deep_gemm/common/sm90_utils.cuh
deep_gemm/include/deep_gemm/common/types.hpp
deep_gemm/include/deep_gemm/common/utils.cuh
deep_gemm/include/deep_gemm/impls/sm100_bf16_gemm.cuh
deep_gemm/include/deep_gemm/impls/sm100_bmk_bnk_mn.cuh
deep_gemm/include/deep_gemm/impls/sm100_fp8_gemm_1d1d.cuh
deep_gemm/include/deep_gemm/impls/sm100_fp8_gemm_1d2d.cuh
deep_gemm/include/deep_gemm/impls/sm100_fp8_mqa_logits.cuh
deep_gemm/include/deep_gemm/impls/sm100_fp8_paged_mqa_logits.cuh
deep_gemm/include/deep_gemm/impls/sm90_bf16_gemm.cuh
deep_gemm/include/deep_gemm/impls/sm90_bmk_bnk_mn.cuh
deep_gemm/include/deep_gemm/impls/sm90_fp8_gemm_1d1d.cuh
deep_gemm/include/deep_gemm/impls/sm90_fp8_gemm_1d2d.cuh
deep_gemm/include/deep_gemm/impls/sm90_fp8_mqa_logits.cuh
deep_gemm/include/deep_gemm/impls/sm90_fp8_paged_mqa_logits.cuh
deep_gemm/include/deep_gemm/impls/smxx_clean_logits.cuh
deep_gemm/include/deep_gemm/impls/smxx_layout.cuh
deep_gemm/testing/__init__.py
deep_gemm/testing/bench.py
deep_gemm/testing/numeric.py
deep_gemm/utils/__init__.py
deep_gemm/utils/layout.py
deep_gemm/utils/math.py
tests/test_attention.py
tests/test_bf16.py
tests/test_einsum.py
tests/test_fp8.py
tests/test_layout.py
tests/test_lazy_init.py