[TypoFix] fix typo for SM120 (#1408)

ede9eaa3 · Cunxiao Ni · GitHub · 53be59dc · ede9eaa3
Unverified Commit ede9eaa3 authored Dec 11, 2025 by Cunxiao Ni Committed by GitHub Dec 11, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/op/gemm_py.cc src/op/gemm_py.cc +1 -1

No files found.
--- a/src/op/gemm_py.cc
+++ b/src/op/gemm_py.cc
@@ -135,7 +135,7 @@ GemmInst GemmPyNode::getGemmInst(int block_size, Target target) const {
    return GemmInst::kMFMA;
  } else if (TargetIsVolta(target) || TargetIsAmpere(target) ||
             TargetIsTuring(target) || TargetIsHopper(target) ||
-             TargetIsSm100(target)) {
+             TargetIsSm100(target) || TargetIsSM120(target)) {
    return GemmInst::kMMA;
  } else {
    ICHECK(0) << "Unsupported target for gemm: " << target->str();