truncate name for cutlass profiler (#2124)

Co-authored-by: yuzhai <[email protected]>
NVIDIA · Feb 21, 2025 · afa1772 · afa1772
1 parent 9b3772d
commit afa1772
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/cutlass_library/library.py b/python/cutlass_library/library.py
@@ -900,9 +900,9 @@ class GemmKind(enum.Enum):
   GemmKind.PlanarComplex: "gemm_planar_complex",
   GemmKind.PlanarComplexArray: "gemm_planar_complex_array",
   GemmKind.Grouped: "gemm_grouped",
-  GemmKind.BlockScaledUniversal3x: "gemm_block_scaled",
+  GemmKind.BlockScaledUniversal3x: "gemm",
   GemmKind.GroupedUniversal3x: "gemm_grouped",
-  GemmKind.GroupedBlockScaledUniversal3x: "gemm_grouped_block_scaled"
+  GemmKind.GroupedBlockScaledUniversal3x: "gemm_grouped"
 }
 
 #