From 5da1c9aadfa69f63f419e56e005ff1d481a2baf4 Mon Sep 17 00:00:00 2001 From: foreverpiano Date: Mon, 24 Mar 2025 05:14:31 +0000 Subject: [PATCH] remove csrc to sta backend kernel --- .gitmodules | 2 +- {csrc => sta_kernel}/sliding_tile_attention/MANIFEST.in | 0 {csrc => sta_kernel}/sliding_tile_attention/README.md | 0 {csrc => sta_kernel}/sliding_tile_attention/config.py | 0 {csrc => sta_kernel}/sliding_tile_attention/setup.py | 0 {csrc => sta_kernel}/sliding_tile_attention/st_attn.cpp | 0 {csrc => sta_kernel}/sliding_tile_attention/st_attn/__init__.py | 0 .../sliding_tile_attention/st_attn/st_attn_h100.cu | 0 {csrc => sta_kernel}/sliding_tile_attention/test/bench.py | 0 .../sliding_tile_attention/test/flex_sta_ref.py | 0 {csrc => sta_kernel}/sliding_tile_attention/test/test_sta.py | 0 {csrc => sta_kernel}/sliding_tile_attention/tk | 0 12 files changed, 1 insertion(+), 1 deletion(-) rename {csrc => sta_kernel}/sliding_tile_attention/MANIFEST.in (100%) rename {csrc => sta_kernel}/sliding_tile_attention/README.md (100%) rename {csrc => sta_kernel}/sliding_tile_attention/config.py (100%) rename {csrc => sta_kernel}/sliding_tile_attention/setup.py (100%) rename {csrc => sta_kernel}/sliding_tile_attention/st_attn.cpp (100%) rename {csrc => sta_kernel}/sliding_tile_attention/st_attn/__init__.py (100%) rename {csrc => sta_kernel}/sliding_tile_attention/st_attn/st_attn_h100.cu (100%) rename {csrc => sta_kernel}/sliding_tile_attention/test/bench.py (100%) rename {csrc => sta_kernel}/sliding_tile_attention/test/flex_sta_ref.py (100%) rename {csrc => sta_kernel}/sliding_tile_attention/test/test_sta.py (100%) rename {csrc => sta_kernel}/sliding_tile_attention/tk (100%) diff --git a/.gitmodules b/.gitmodules index d284208d0..d0d2cb3ef 100644 --- a/.gitmodules +++ b/.gitmodules @@ -1,3 +1,3 @@ [submodule "csrc/sliding_tile_attention/tk"] - path = csrc/sliding_tile_attention/tk + path = sta_kernel/sliding_tile_attention/tk url = https://github.com/HazyResearch/ThunderKittens.git diff --git a/csrc/sliding_tile_attention/MANIFEST.in b/sta_kernel/sliding_tile_attention/MANIFEST.in similarity index 100% rename from csrc/sliding_tile_attention/MANIFEST.in rename to sta_kernel/sliding_tile_attention/MANIFEST.in diff --git a/csrc/sliding_tile_attention/README.md b/sta_kernel/sliding_tile_attention/README.md similarity index 100% rename from csrc/sliding_tile_attention/README.md rename to sta_kernel/sliding_tile_attention/README.md diff --git a/csrc/sliding_tile_attention/config.py b/sta_kernel/sliding_tile_attention/config.py similarity index 100% rename from csrc/sliding_tile_attention/config.py rename to sta_kernel/sliding_tile_attention/config.py diff --git a/csrc/sliding_tile_attention/setup.py b/sta_kernel/sliding_tile_attention/setup.py similarity index 100% rename from csrc/sliding_tile_attention/setup.py rename to sta_kernel/sliding_tile_attention/setup.py diff --git a/csrc/sliding_tile_attention/st_attn.cpp b/sta_kernel/sliding_tile_attention/st_attn.cpp similarity index 100% rename from csrc/sliding_tile_attention/st_attn.cpp rename to sta_kernel/sliding_tile_attention/st_attn.cpp diff --git a/csrc/sliding_tile_attention/st_attn/__init__.py b/sta_kernel/sliding_tile_attention/st_attn/__init__.py similarity index 100% rename from csrc/sliding_tile_attention/st_attn/__init__.py rename to sta_kernel/sliding_tile_attention/st_attn/__init__.py diff --git a/csrc/sliding_tile_attention/st_attn/st_attn_h100.cu b/sta_kernel/sliding_tile_attention/st_attn/st_attn_h100.cu similarity index 100% rename from csrc/sliding_tile_attention/st_attn/st_attn_h100.cu rename to sta_kernel/sliding_tile_attention/st_attn/st_attn_h100.cu diff --git a/csrc/sliding_tile_attention/test/bench.py b/sta_kernel/sliding_tile_attention/test/bench.py similarity index 100% rename from csrc/sliding_tile_attention/test/bench.py rename to sta_kernel/sliding_tile_attention/test/bench.py diff --git a/csrc/sliding_tile_attention/test/flex_sta_ref.py b/sta_kernel/sliding_tile_attention/test/flex_sta_ref.py similarity index 100% rename from csrc/sliding_tile_attention/test/flex_sta_ref.py rename to sta_kernel/sliding_tile_attention/test/flex_sta_ref.py diff --git a/csrc/sliding_tile_attention/test/test_sta.py b/sta_kernel/sliding_tile_attention/test/test_sta.py similarity index 100% rename from csrc/sliding_tile_attention/test/test_sta.py rename to sta_kernel/sliding_tile_attention/test/test_sta.py diff --git a/csrc/sliding_tile_attention/tk b/sta_kernel/sliding_tile_attention/tk similarity index 100% rename from csrc/sliding_tile_attention/tk rename to sta_kernel/sliding_tile_attention/tk