diff --git a/src/op/gemm_py.cc b/src/op/gemm_py.cc
index 6953779a1..8f6201c64 100644
--- a/src/op/gemm_py.cc
+++ b/src/op/gemm_py.cc
@@ -133,9 +133,7 @@ GemmInst GemmPyNode::getGemmInst(int block_size, Target target) const {
     return GemmInst::kWGMMA;
   } else if (TargetIsCDNA(target)) {
     return GemmInst::kMFMA;
-  } else if (TargetIsVolta(target) || TargetIsAmpere(target) ||
-             TargetIsTuring(target) || TargetIsHopper(target) ||
-             TargetIsSm100(target) || TargetIsSM120(target)) {
+  } else if (TargetIsCuda(target)) {
     return GemmInst::kMMA;
   } else {
     ICHECK(0) << "Unsupported target for gemm: " << target->str();