Allow mma gemm for all cuda (#1047)

e3742d33 · Yichen Yan · GitHub · 0ff4f427 · e3742d33
Unverified Commit e3742d33 authored Oct 16, 2025 by Yichen Yan Committed by GitHub Oct 16, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

src/op/gemm.cc src/op/gemm.cc +1 -3

No files found.
--- a/src/op/gemm.cc
+++ b/src/op/gemm.cc
@@ -194,9 +194,7 @@ GemmInst GemmNode::GetGemmInst(int block_size, Target target) const {
    return GemmInst::kWGMMA;
  } else if (TargetIsCDNA(target)) {
    return GemmInst::kMFMA;
-  } else if (TargetIsVolta(target) || TargetIsAmpere(target) ||
+  } else if (TargetIsCuda(target)) {
-             TargetIsTuring(target) || TargetIsHopper(target) ||
-             TargetIsSm100(target)) {
    return GemmInst::kMMA;
  } else {
    ICHECK(0) << "Unsupported target for gemm: " << target->str();