LICENSE
MANIFEST.in
README.md
pyproject.toml
requirements.txt
setup.cfg
setup.py
version.txt
tests/test_attention_patterns.py
tests/test_checkpoint.py
tests/test_fmha_flop_formula.py
tests/test_fmha_merge_attentions.py
tests/test_fwbw_overlap.py
tests/test_indexing.py
tests/test_mem_eff_attention.py
tests/test_multiprocessing_utils.py
tests/test_profiler.py
tests/test_rmsnorm.py
tests/test_rope_padded.py
tests/test_seqpar.py
tests/test_sequence_parallel_fused_ops.py
tests/test_sparse_tensors.py
tests/test_splitk_reference.py
tests/test_tiled_matmul.py
tests/test_tree_attention.py
tests/test_unbind.py
xformers/__init__.py
xformers/_cpp_lib.py
xformers/_deprecation_warning.py
xformers/attn_bias_utils.py
xformers/checkpoint.py
xformers/fwbw_overlap.py
xformers/info.py
xformers/test.py
xformers/utils.py
xformers.egg-info/PKG-INFO
xformers.egg-info/SOURCES.txt
xformers.egg-info/dependency_links.txt
xformers.egg-info/not-zip-safe
xformers.egg-info/requires.txt
xformers.egg-info/top_level.txt
xformers/benchmarks/__init__.py
xformers/benchmarks/benchmark_attn_decoding.py
xformers/benchmarks/benchmark_indexing.py
xformers/benchmarks/benchmark_mem_eff_attention.py
xformers/benchmarks/benchmark_merge_attentions.py
xformers/benchmarks/benchmark_sequence_parallel_fused.py
xformers/benchmarks/benchmark_tiled_matmul.py
xformers/benchmarks/utils.py
xformers/csrc/nvcc_info.cu
xformers/csrc/pt_stable_utils.cu
xformers/csrc/pt_stable_utils.h
xformers/flash_attn_3/__init__.py
xformers/ops/__init__.py
xformers/ops/common.py
xformers/ops/differentiable_collectives.py
xformers/ops/indexing.py
xformers/ops/modpar_layers.py
xformers/ops/rmsnorm.py
xformers/ops/rope_padded.py
xformers/ops/seqpar.py
xformers/ops/sequence_parallel_fused_ops.py
xformers/ops/swiglu_op.py
xformers/ops/tiled_matmul.py
xformers/ops/tree_attention.py
xformers/ops/unbind.py
xformers/ops/_triton/__init__.py
xformers/ops/_triton/k_index_select_cat.py
xformers/ops/_triton/k_scaled_index_add.py
xformers/ops/_triton/matmul_perf_model.py
xformers/ops/_triton/rmsnorm_kernels.py
xformers/ops/_triton/rope_padded_kernels.py
xformers/ops/_triton/tiled_matmul_kernels.py
xformers/ops/fmha/__init__.py
xformers/ops/fmha/attn_bias.py
xformers/ops/fmha/ck.py
xformers/ops/fmha/ck_splitk.py
xformers/ops/fmha/common.py
xformers/ops/fmha/cutlass.py
xformers/ops/fmha/cutlass_blackwell.py
xformers/ops/fmha/dispatch.py
xformers/ops/fmha/flash.py
xformers/ops/fmha/flash3.py
xformers/ops/fmha/merge_training.py
xformers/ops/fmha/torch_attention_compat.py
xformers/ops/fmha/triton_splitk.py
xformers/ops/fmha/_triton/__init__.py
xformers/ops/fmha/_triton/splitk_kernels.py
xformers/profiler/__init__.py
xformers/profiler/api.py
xformers/profiler/device_limits.py
xformers/profiler/find_slowest.py
xformers/profiler/profile_analyzer.py
xformers/profiler/profiler.py
xformers/profiler/profiler_dcgm.py
xformers/profiler/profiler_dcgm_impl.py
xformers/sparse/__init__.py
xformers/sparse/blocksparse_tensor.py
xformers/sparse/utils.py
xformers/triton/__init__.py
xformers/triton/importing.py