From 35c0347b9de02cec382c655cc19608d3026d6348 Mon Sep 17 00:00:00 2001 From: Brian Ryu Date: Mon, 8 Dec 2025 22:07:56 +0000 Subject: [PATCH 1/2] Update sm120 and 121 with min cuda capability 12.9 --- flashinfer/aot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flashinfer/aot.py b/flashinfer/aot.py index 609e1bcbcf..42410359a1 100644 --- a/flashinfer/aot.py +++ b/flashinfer/aot.py @@ -743,8 +743,8 @@ def has_sm(compute: str, version: str) -> bool: "sm100f": has_sm("compute_100", "12.9"), "sm103": has_sm("compute_103", "12.9"), "sm110": has_sm("compute_110", "12.9"), - "sm120": has_sm("compute_120", "13.0"), - "sm121": has_sm("compute_121", "13.0"), + "sm120": has_sm("compute_120", "12.9"), + "sm121": has_sm("compute_121", "12.9"), } From 1b95a22127c40e731298105fbe60c2ee88362774 Mon Sep 17 00:00:00 2001 From: Brian Ryu Date: Mon, 8 Dec 2025 22:22:42 +0000 Subject: [PATCH 2/2] Fix 110 and 120 --- flashinfer/aot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flashinfer/aot.py b/flashinfer/aot.py index 42410359a1..ad64ac2c11 100644 --- a/flashinfer/aot.py +++ b/flashinfer/aot.py @@ -742,8 +742,8 @@ def has_sm(compute: str, version: str) -> bool: "sm100": has_sm("compute_100", "12.8"), "sm100f": has_sm("compute_100", "12.9"), "sm103": has_sm("compute_103", "12.9"), - "sm110": has_sm("compute_110", "12.9"), - "sm120": has_sm("compute_120", "12.9"), + "sm110": has_sm("compute_110", "13.0"), + "sm120": has_sm("compute_120", "12.8"), "sm121": has_sm("compute_121", "12.9"), }