CMakeLists.txt
LICENSE
README.md
envs.py
pyproject.toml
setup.py
attention_kernels/__init__.py
attention_kernels/_custom_ops.py
attention_kernels.egg-info/PKG-INFO
attention_kernels.egg-info/SOURCES.txt
attention_kernels.egg-info/dependency_links.txt
attention_kernels.egg-info/requires.txt
attention_kernels.egg-info/top_level.txt
cmake/cpu_extension.cmake
cmake/hipify.py
cmake/utils.cmake
csrc/cache_kernels.cu
csrc/cuda_compat.h
csrc/dispatch_utils.h
csrc/ops.h
csrc/torch_bindings.cpp
csrc/attention/attention_dtypes.h
csrc/attention/attention_generic.cuh
csrc/attention/attention_kernels.cu
csrc/attention/attention_utils.cuh
csrc/attention/dtype_bfloat16.cuh
csrc/attention/dtype_float16.cuh
csrc/attention/dtype_float32.cuh
csrc/attention/dtype_fp8.cuh
csrc/core/registration.h
csrc/quantization/fp8/amd_detail/hip_float8.h
csrc/quantization/fp8/amd_detail/hip_float8_impl.h
csrc/quantization/fp8/amd_detail/quant_utils.cuh
csrc/quantization/fp8_e5m2_kvcache/quant_utils.cuh