Enable dynamic shapes for CuTeDSL GroupGemm

[ghstack-poisoned]
2025-11-16 23:44:53 +08:00 · 2025-11-11 09:37:43 -08:00
1 changed files with 0 additions and 3 deletions
--- a/torch/_inductor/utils.py
+++ b/torch/_inductor/utils.py
@ -1974,9 +1974,6 @@ def use_blackwell_cutedsl_grouped_mm(
    if not can_use_tma(mat_a, mat_b, output_layout=layout):
        return False

-    if any(is_dynamic(x) for x in [mat_a, mat_b]):
-        return False
-
    if not a_is_2d or b_is_2d:
        return False