From afa1772203677c5118fcd82537a9c8fefbcc7008 Mon Sep 17 00:00:00 2001 From: Yujia Zhai Date: Thu, 20 Feb 2025 21:16:56 -0800 Subject: [PATCH] truncate name for cutlass profiler (#2124) Co-authored-by: yuzhai --- python/cutlass_library/library.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/cutlass_library/library.py b/python/cutlass_library/library.py index d5f02db07..13a579a84 100644 --- 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" } #