From 63669f3b1d6365ead5e2f759d9b5468cdc0737d7 Mon Sep 17 00:00:00 2001 From: YangKai0616 Date: Wed, 11 Feb 2026 03:59:12 +0000 Subject: [PATCH 1/3] Change cutlass-sycl name to sycl-tla --- flash-attn2/build.toml | 2 +- megablocks/build.toml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/flash-attn2/build.toml b/flash-attn2/build.toml index 88390ae8..22ea2a19 100644 --- a/flash-attn2/build.toml +++ b/flash-attn2/build.toml @@ -164,7 +164,7 @@ src = [ backend = "xpu" depends = [ "torch", - "cutlass_sycl", + "sycl_tla", ] src = [ "flash_attn_xpu/flash_api.cpp", diff --git a/megablocks/build.toml b/megablocks/build.toml index 36a13082..cb8edf74 100644 --- a/megablocks/build.toml +++ b/megablocks/build.toml @@ -53,7 +53,7 @@ src = [ backend = "xpu" depends = [ "torch", - "cutlass_sycl", + "sycl_tla", ] src = [ "csrc_xpu/utils.h", From 032964090ac40cbdfcb084320bbccfa0219dd67d Mon Sep 17 00:00:00 2001 From: YangKai0616 Date: Sat, 28 Feb 2026 02:36:13 +0000 Subject: [PATCH 2/3] Update kernels version --- flash-attn2/flake.lock | 7 +++---- flash-attn2/flake.nix | 2 +- megablocks/flake.lock | 7 +++---- megablocks/flake.nix | 2 +- 4 files changed, 8 insertions(+), 10 deletions(-) diff --git a/flash-attn2/flake.lock b/flash-attn2/flake.lock index 801c84bc..4b8dc5ac 100644 --- a/flash-attn2/flake.lock +++ b/flash-attn2/flake.lock @@ -41,16 +41,15 @@ "rust-overlay": "rust-overlay" }, "locked": { - "lastModified": 1769443799, - "narHash": "sha256-iSJyElXgv2SWMXwJuTAHFcZQI8ViWNJpMKFZ1JxyfPE=", + "lastModified": 1772049868, + "narHash": "sha256-WgKS95txMqqrX2oEntUDoZHwK1fwGaKYXdvexa4AT+4=", "owner": "huggingface", "repo": "kernels", - "rev": "30685a79203ed855b328cee874698d13e82fb3ae", + "rev": "f4c7babef0f2d4212f75aa469b63a41fce8df78c", "type": "github" }, "original": { "owner": "huggingface", - "ref": "v0.12.1", "repo": "kernels", "type": "github" } diff --git a/flash-attn2/flake.nix b/flash-attn2/flake.nix index ff53a773..22ad0199 100644 --- a/flash-attn2/flake.nix +++ b/flash-attn2/flake.nix @@ -2,7 +2,7 @@ description = "Flake for flash-attn kernel"; inputs = { - kernel-builder.url = "github:huggingface/kernels/v0.12.1"; + kernel-builder.url = "github:huggingface/kernels"; }; outputs = diff --git a/megablocks/flake.lock b/megablocks/flake.lock index 801c84bc..4b8dc5ac 100644 --- a/megablocks/flake.lock +++ b/megablocks/flake.lock @@ -41,16 +41,15 @@ "rust-overlay": "rust-overlay" }, "locked": { - "lastModified": 1769443799, - "narHash": "sha256-iSJyElXgv2SWMXwJuTAHFcZQI8ViWNJpMKFZ1JxyfPE=", + "lastModified": 1772049868, + "narHash": "sha256-WgKS95txMqqrX2oEntUDoZHwK1fwGaKYXdvexa4AT+4=", "owner": "huggingface", "repo": "kernels", - "rev": "30685a79203ed855b328cee874698d13e82fb3ae", + "rev": "f4c7babef0f2d4212f75aa469b63a41fce8df78c", "type": "github" }, "original": { "owner": "huggingface", - "ref": "v0.12.1", "repo": "kernels", "type": "github" } diff --git a/megablocks/flake.nix b/megablocks/flake.nix index 7359f3f0..e99aad20 100644 --- a/megablocks/flake.nix +++ b/megablocks/flake.nix @@ -2,7 +2,7 @@ description = "Flake for megablocks_moe kernel"; inputs = { - kernel-builder.url = "github:huggingface/kernels/v0.12.1"; + kernel-builder.url = "github:huggingface/kernels"; }; outputs = From a7857000063ed0f83891c80de4f5f1c9909d074a Mon Sep 17 00:00:00 2001 From: YangKai0616 Date: Thu, 5 Mar 2026 03:46:59 +0000 Subject: [PATCH 3/3] Update flake.lock --- flash-attn2/flake.lock | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/flash-attn2/flake.lock b/flash-attn2/flake.lock index 4b8dc5ac..f0e88e05 100644 --- a/flash-attn2/flake.lock +++ b/flash-attn2/flake.lock @@ -41,11 +41,11 @@ "rust-overlay": "rust-overlay" }, "locked": { - "lastModified": 1772049868, - "narHash": "sha256-WgKS95txMqqrX2oEntUDoZHwK1fwGaKYXdvexa4AT+4=", + "lastModified": 1772639128, + "narHash": "sha256-2tmxkshScPr/NK3JFJ5KVr/U1qhmdsqlut+Kfe7zLNY=", "owner": "huggingface", "repo": "kernels", - "rev": "f4c7babef0f2d4212f75aa469b63a41fce8df78c", + "rev": "20d9f69982d233bcd3fdb3d0daf35fe03a233413", "type": "github" }, "original": {