diff --git a/flashinfer/aot.py b/flashinfer/aot.py index 609e1bcbcf..ad64ac2c11 100644 --- a/flashinfer/aot.py +++ b/flashinfer/aot.py @@ -742,9 +742,9 @@ 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", "13.0"), - "sm121": has_sm("compute_121", "13.0"), + "sm110": has_sm("compute_110", "13.0"), + "sm120": has_sm("compute_120", "12.8"), + "sm121": has_sm("compute_121", "12.9"), }