From 528aae2be7d5700d07d8579a916f4a4980e9d37e Mon Sep 17 00:00:00 2001 From: Ethan Date: Sat, 15 Nov 2025 15:05:28 +0800 Subject: [PATCH 1/4] get submodule from my repositories --- .gitmodules | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.gitmodules b/.gitmodules index e837ed9a..fc9806ff 100644 --- a/.gitmodules +++ b/.gitmodules @@ -1,4 +1,4 @@ [submodule "ThunderKittens"] path = ThunderKittens - url = https://github.com/HazyResearch/ThunderKittens.git + url = https://github.com/HaoZhang-Ethan/ThunderKittens.git branch = bvm-single-ctrl-pre-new-warps From 108dfbc0570fac493406a2d18740531e1af147da Mon Sep 17 00:00:00 2001 From: Ethan Date: Sat, 15 Nov 2025 16:20:36 +0800 Subject: [PATCH 2/4] refactor(scripts): update model path in generate.py config Use local model path instead of default remote path for testing purposes --- megakernels/scripts/generate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/megakernels/scripts/generate.py b/megakernels/scripts/generate.py index 2b43f943..f31b518b 100644 --- a/megakernels/scripts/generate.py +++ b/megakernels/scripts/generate.py @@ -26,7 +26,8 @@ class ScriptConfig(pydra.Config): - model: str = "meta-llama/Llama-3.2-1B-Instruct" + # model: str = "meta-llama/Llama-3.2-1B-Instruct" + model: str = "/home/fit/qianxueh/WORK/alchem/zhanghao/Mega/Models/Meta-Llama-3.2-1B-Instruct" device: str = "cuda:0" prompt: str = "tell me a funny joke about cookies" chat: bool = False From d2f9722b74e2bbb63a88bbca196d0d468dfe0ce4 Mon Sep 17 00:00:00 2001 From: Ethan Date: Sat, 15 Nov 2025 16:52:06 +0800 Subject: [PATCH 3/4] chore: update ThunderKittens submodule to latest commit --- ThunderKittens | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ThunderKittens b/ThunderKittens index 664c108d..0d002359 160000 --- a/ThunderKittens +++ b/ThunderKittens @@ -1 +1 @@ -Subproject commit 664c108d16f12707a73d3072ab525f26fb2b4f62 +Subproject commit 0d002359ff313ff9ee5c5354439d1d7ef42704a5 From c14337db7352dfede357b2eca7df6c2744b10efa Mon Sep 17 00:00:00 2001 From: Ethan Date: Sat, 15 Nov 2025 20:42:44 +0800 Subject: [PATCH 4/4] chore: update ThunderKittens submodule to latest commit --- ThunderKittens | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ThunderKittens b/ThunderKittens index 0d002359..7bd4b5c8 160000 --- a/ThunderKittens +++ b/ThunderKittens @@ -1 +1 @@ -Subproject commit 0d002359ff313ff9ee5c5354439d1d7ef42704a5 +Subproject commit 7bd4b5c889803c8cbc33fc4dc8223f5c74fc78eb