init

4b485dd1 · wangkx1 · 5c5de278 · 4b485dd1 · 4b485dd1 · 4b485dd1
Commit 4b485dd1 authored Apr 15, 2026 by wangkx1
5 changed files
--- a/fallback_org.sh
+++ b/fallback_org.sh
+cp /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py.org /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py
\ No newline at end of file
--- a/patch_torch.sh
+++ b/patch_torch.sh
+cp /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py.org
+cp src/flash_mla_interface_torch.py /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py
\ No newline at end of file
--- a/patch_triton.sh
+++ b/patch_triton.sh
+cp /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py.org
+cp src/flash_mla_interface_triton.py /usr/local/lib/python3.10/dist-packages/flash_mla/flash_mla_interface.py
\ No newline at end of file
--- a/src/flash_mla_interface_torch.py
+++ b/src/flash_mla_interface_torch.py
--- a/src/flash_mla_interface_triton.py
+++ b/src/flash_mla_interface_triton.py