From 658e213f7eec296159d970960483b05c59b066de Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Thu, 28 Aug 2025 09:04:40 +0000 Subject: [PATCH 1/2] Add `cutlass_4_0` as a dependency --- build2cmake/src/config/v2.rs | 2 ++ build2cmake/src/torch/cuda.rs | 11 +++++++++++ lib/deps.nix | 3 +++ 3 files changed, 16 insertions(+) diff --git a/build2cmake/src/config/v2.rs b/build2cmake/src/config/v2.rs index 936b5970..16385818 100644 --- a/build2cmake/src/config/v2.rs +++ b/build2cmake/src/config/v2.rs @@ -217,6 +217,8 @@ pub enum Dependencies { Cutlass3_8, #[serde(rename = "cutlass_3_9")] Cutlass3_9, + #[serde(rename = "cutlass_4_0")] + Cutlass4_0, #[serde(rename = "cutlass_sycl")] CutlassSycl, Torch, diff --git a/build2cmake/src/torch/cuda.rs b/build2cmake/src/torch/cuda.rs index f71df783..c58db682 100644 --- a/build2cmake/src/torch/cuda.rs +++ b/build2cmake/src/torch/cuda.rs @@ -276,6 +276,17 @@ fn render_deps(env: &Environment, build: &Build, write: &mut impl Write) -> Resu ) .wrap_err("Cannot render CUTLASS dependency template")?; } + Dependencies::Cutlass4_0 => { + env.get_template("cuda/dep-cutlass.cmake") + .wrap_err("Cannot get CUTLASS dependency template")? + .render_to_write( + context! { + version => "4.0.0", + }, + &mut *write, + ) + .wrap_err("Cannot render CUTLASS dependency template")?; + } Dependencies::Torch => (), _ => { eprintln!("Warning: CUDA backend doesn't need/support dependency: {dep:?}"); diff --git a/lib/deps.nix b/lib/deps.nix index 4f96d962..bffecf01 100644 --- a/lib/deps.nix +++ b/lib/deps.nix @@ -25,6 +25,9 @@ let "cutlass_3_9" = [ pkgs.cutlass_3_9 ]; + "cutlass_4_0" = [ + pkgs.cutlass_4_0 + ]; "torch" = [ torch torch.cxxdev From 1d91b7c28f07a9cca5f13bbfef326ce48f05aed2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Thu, 25 Sep 2025 08:14:31 +0000 Subject: [PATCH 2/2] CI: add rustfmt component --- .github/workflows/rust.yaml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/rust.yaml b/.github/workflows/rust.yaml index 467b19b0..f38c8b55 100644 --- a/.github/workflows/rust.yaml +++ b/.github/workflows/rust.yaml @@ -9,6 +9,8 @@ jobs: steps: - uses: actions/checkout@v4 - uses: dtolnay/rust-toolchain@stable + with: + components: rustfmt - name: Cargo fmt (kernel-abi-check) run: | ( cd kernel-abi-check/kernel-abi-check && cargo fmt --all -- --check )