Skip to content

Commit afa1772

Browse files
yzhaiustcyuzhai
and
yuzhai
authored
truncate name for cutlass profiler (#2124)
Co-authored-by: yuzhai <[email protected]>
1 parent 9b3772d commit afa1772

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/cutlass_library/library.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -900,9 +900,9 @@ class GemmKind(enum.Enum):
900900
GemmKind.PlanarComplex: "gemm_planar_complex",
901901
GemmKind.PlanarComplexArray: "gemm_planar_complex_array",
902902
GemmKind.Grouped: "gemm_grouped",
903-
GemmKind.BlockScaledUniversal3x: "gemm_block_scaled",
903+
GemmKind.BlockScaledUniversal3x: "gemm",
904904
GemmKind.GroupedUniversal3x: "gemm_grouped",
905-
GemmKind.GroupedBlockScaledUniversal3x: "gemm_grouped_block_scaled"
905+
GemmKind.GroupedBlockScaledUniversal3x: "gemm_grouped"
906906
}
907907

908908
#

0 commit comments

Comments
 (0)