From 4ba7668168da691b05f09e01c744ef43f0f4779e Mon Sep 17 00:00:00 2001 From: Masahiro Ogawa Date: Mon, 23 Mar 2026 16:18:27 +0900 Subject: [PATCH] fix: add sm_89 (Ada Lovelace) gencode and fix Python/dep versions - Add -gencode arch=compute_89,code=sm_89 for RTX 40xx series (Ada Lovelace) under CUDA >= 11.8, alongside existing sm_75/80/87/90 - Bump requires-python to >=3.10 (quack-kernels requires it) - Sync quack-kernels to 0.3.4 to match setup.py Co-Authored-By: Claude Opus 4.6 (1M context) --- pyproject.toml | 4 ++-- setup.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 39a49bb21..39af57456 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ authors = [ { name = "Tri Dao", email = "tri@tridao.me" }, { name = "Albert Gu", email = "agu@cs.cmu.edu" } ] -requires-python = ">= 3.9" +requires-python = ">= 3.10" dynamic = ["version"] license = { file = "LICENSE" } # Include a LICENSE file in your repo keywords = ["cuda", "pytorch", "state-space model"] @@ -18,7 +18,7 @@ classifiers = [ dependencies = [ "torch", "tilelang==0.1.8", - "quack-kernels==0.3.1", + "quack-kernels==0.3.4", "triton>=3.5.0", "ninja", "einops", diff --git a/setup.py b/setup.py index 80d57ef7f..327c12245 100755 --- a/setup.py +++ b/setup.py @@ -185,6 +185,8 @@ def append_nvcc_threads(nvcc_extra_args): cc_flag.append("-gencode") cc_flag.append("arch=compute_87,code=sm_87") if bare_metal_version >= Version("11.8"): + cc_flag.append("-gencode") + cc_flag.append("arch=compute_89,code=sm_89") cc_flag.append("-gencode") cc_flag.append("arch=compute_90,code=sm_90") if bare_metal_version >= Version("12.8"):