LICENSE
MANIFEST.in
README.md
THIRD_PARTY_LICENSES
pyproject.toml
setup.py
csrc/mfa_bridge.mm
csrc/mfa/Metal.cpp
csrc/mfa/ccv_nnc_mfa.cpp
csrc/mfa/ccv_nnc_mfa.hpp
csrc/mfa/ccv_nnc_mfa_attention.cpp
csrc/mfa/ccv_nnc_mfa_attention.hpp
csrc/mfa/ccv_nnc_mfa_cast.cpp
csrc/mfa/ccv_nnc_mfa_cast.hpp
csrc/mfa/ccv_nnc_mfa_defines.hpp
csrc/mfa/ccv_nnc_mfa_error.cpp
csrc/mfa/ccv_nnc_mfa_error.hpp
csrc/mfa/ccv_nnc_mfa_hash.hpp
csrc/mfa/libmfa.inc
csrc/mfa/3rdparty/libmfaios16-v1.0.2-b.metallib
csrc/mfa/3rdparty/libmfamacos13-v1.0.2-b.metallib
csrc/mfa/3rdparty/metal-cpp/Dispatch.cpp
csrc/mfa/3rdparty/metal-cpp/Dispatch.hpp
csrc/mfa/3rdparty/metal-cpp/Metal.hpp
csrc/mfa/v2/AttentionDescriptor.cpp
csrc/mfa/v2/AttentionDescriptor.hpp
csrc/mfa/v2/AttentionKernel+Precompiled.cpp
csrc/mfa/v2/AttentionKernel+Precompiled.inc
csrc/mfa/v2/AttentionKernel.cpp
csrc/mfa/v2/AttentionKernel.hpp
csrc/mfa/v2/AttentionKernelDescriptor.cpp
csrc/mfa/v2/AttentionKernelDescriptor.hpp
csrc/mfa/v2/AttentionKernelType.hpp
csrc/mfa/v2/AttentionOperand.hpp
csrc/mfa/v2/CastDescriptor.cpp
csrc/mfa/v2/CastDescriptor.hpp
csrc/mfa/v2/CastKernel.cpp
csrc/mfa/v2/CastKernel.hpp
csrc/mfa/v2/CodeWriter.cpp
csrc/mfa/v2/CodeWriter.hpp
csrc/mfa/v2/DeviceProperties.hpp
csrc/mfa/v2/GEMMHeaders.cpp
csrc/mfa/v2/GEMMHeaders.hpp
csrc/mfa/v2/GEMMOperandPrecision.hpp
csrc/mfa/v2/NAAttentionDescriptor.cpp
csrc/mfa/v2/NAAttentionDescriptor.hpp
csrc/mfa/v2/NAAttentionKernel.cpp
csrc/mfa/v2/NAAttentionKernel.hpp
csrc/mfa/v2/NAAttentionKernelDescriptor.cpp
csrc/mfa/v2/NAAttentionKernelDescriptor.hpp
csrc/mfa/v2/NAMatMulDescriptor.cpp
csrc/mfa/v2/NAMatMulDescriptor.hpp
csrc/mfa/v2/NAMatMulKernel.cpp
csrc/mfa/v2/NAMatMulKernel.hpp
csrc/mfa/v2/NAMatMulKernelDescriptor.cpp
csrc/mfa/v2/NAMatMulKernelDescriptor.hpp
csrc/mfa/v2/PipelineValue.hpp
csrc/mfa/v2/ShaderCache.hpp
csrc/nnc/ccv_nnc.h
csrc/nnc/mfa/Metal.cpp
csrc/nnc/mfa/ccv_nnc_mfa.cpp
csrc/nnc/mfa/ccv_nnc_mfa.hpp
csrc/nnc/mfa/ccv_nnc_mfa_attention.cpp
csrc/nnc/mfa/ccv_nnc_mfa_attention.hpp
csrc/nnc/mfa/ccv_nnc_mfa_cast.cpp
csrc/nnc/mfa/ccv_nnc_mfa_cast.hpp
csrc/nnc/mfa/ccv_nnc_mfa_defines.hpp
csrc/nnc/mfa/ccv_nnc_mfa_error.cpp
csrc/nnc/mfa/ccv_nnc_mfa_error.hpp
csrc/nnc/mfa/ccv_nnc_mfa_hash.hpp
csrc/nnc/mfa/libmfa.inc
csrc/nnc/mfa/3rdparty/libmfaios16-v1.0.2-b.metallib
csrc/nnc/mfa/3rdparty/libmfamacos13-v1.0.2-b.metallib
csrc/nnc/mfa/3rdparty/metal-cpp/Dispatch.cpp
csrc/nnc/mfa/3rdparty/metal-cpp/Dispatch.hpp
csrc/nnc/mfa/3rdparty/metal-cpp/Metal.hpp
csrc/nnc/mfa/v2/AttentionDescriptor.cpp
csrc/nnc/mfa/v2/AttentionDescriptor.hpp
csrc/nnc/mfa/v2/AttentionKernel+Precompiled.cpp
csrc/nnc/mfa/v2/AttentionKernel+Precompiled.inc
csrc/nnc/mfa/v2/AttentionKernel.cpp
csrc/nnc/mfa/v2/AttentionKernel.hpp
csrc/nnc/mfa/v2/AttentionKernelDescriptor.cpp
csrc/nnc/mfa/v2/AttentionKernelDescriptor.hpp
csrc/nnc/mfa/v2/AttentionKernelType.hpp
csrc/nnc/mfa/v2/AttentionOperand.hpp
csrc/nnc/mfa/v2/CastDescriptor.cpp
csrc/nnc/mfa/v2/CastDescriptor.hpp
csrc/nnc/mfa/v2/CastKernel.cpp
csrc/nnc/mfa/v2/CastKernel.hpp
csrc/nnc/mfa/v2/CodeWriter.cpp
csrc/nnc/mfa/v2/CodeWriter.hpp
csrc/nnc/mfa/v2/DeviceProperties.hpp
csrc/nnc/mfa/v2/GEMMHeaders.cpp
csrc/nnc/mfa/v2/GEMMHeaders.hpp
csrc/nnc/mfa/v2/GEMMOperandPrecision.hpp
csrc/nnc/mfa/v2/NAAttentionDescriptor.cpp
csrc/nnc/mfa/v2/NAAttentionDescriptor.hpp
csrc/nnc/mfa/v2/NAAttentionKernel.cpp
csrc/nnc/mfa/v2/NAAttentionKernel.hpp
csrc/nnc/mfa/v2/NAAttentionKernelDescriptor.cpp
csrc/nnc/mfa/v2/NAAttentionKernelDescriptor.hpp
csrc/nnc/mfa/v2/NAMatMulDescriptor.cpp
csrc/nnc/mfa/v2/NAMatMulDescriptor.hpp
csrc/nnc/mfa/v2/NAMatMulKernel.cpp
csrc/nnc/mfa/v2/NAMatMulKernel.hpp
csrc/nnc/mfa/v2/NAMatMulKernelDescriptor.cpp
csrc/nnc/mfa/v2/NAMatMulKernelDescriptor.hpp
csrc/nnc/mfa/v2/PipelineValue.hpp
csrc/nnc/mfa/v2/ShaderCache.hpp
metal_flash_sdpa/__init__.py
mps_flash_sdpa.egg-info/PKG-INFO
mps_flash_sdpa.egg-info/SOURCES.txt
mps_flash_sdpa.egg-info/dependency_links.txt
mps_flash_sdpa.egg-info/requires.txt
mps_flash_sdpa.egg-info/top_level.txt
tests/test_backward.py
tests/test_causal.py
tests/test_diffusers.py
tests/test_forward.py
tests/test_integration.py