We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9b3772d commit afa1772Copy full SHA for afa1772
python/cutlass_library/library.py
@@ -900,9 +900,9 @@ class GemmKind(enum.Enum):
900
GemmKind.PlanarComplex: "gemm_planar_complex",
901
GemmKind.PlanarComplexArray: "gemm_planar_complex_array",
902
GemmKind.Grouped: "gemm_grouped",
903
- GemmKind.BlockScaledUniversal3x: "gemm_block_scaled",
+ GemmKind.BlockScaledUniversal3x: "gemm",
904
GemmKind.GroupedUniversal3x: "gemm_grouped",
905
- GemmKind.GroupedBlockScaledUniversal3x: "gemm_grouped_block_scaled"
+ GemmKind.GroupedBlockScaledUniversal3x: "gemm_grouped"
906
}
907
908
#
0 commit comments