Skip to content

Commit a057551

Browse files
authored
metal : Fix dimension constraint violation in matmul2d descriptor (ggml-org#21048)
Updates Metal tensor API test probe to fix the dimension constraint violation in the matmul2d descriptor (at least one value must be a multiple of 16).
1 parent 003e533 commit a057551

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

ggml/src/ggml-metal/ggml-metal-device.m

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -690,7 +690,7 @@ ggml_metal_device_t ggml_metal_device_init(int device) {
690690
" auto tB = B.slice((int)tgid.x, 0); \n"
691691
" \n"
692692
" matmul2d< \n"
693-
" matmul2d_descriptor(8, 8, dynamic_extent), \n"
693+
" matmul2d_descriptor(16, 16, dynamic_extent), \n"
694694
" execution_simdgroups<4>> mm; \n"
695695
" \n"
696696
" auto cT = mm.get_destination_cooperative_tensor<decltype(tA), decltype(tB), float>(); \n"
@@ -740,7 +740,7 @@ ggml_metal_device_t ggml_metal_device_init(int device) {
740740
" auto tB = B.slice((int)tgid.x, 0); \n"
741741
" \n"
742742
" matmul2d< \n"
743-
" matmul2d_descriptor(8, 8, dynamic_extent), \n"
743+
" matmul2d_descriptor(16, 16, dynamic_extent), \n"
744744
" execution_simdgroups<4>> mm; \n"
745745
" \n"
746746
" auto cT = mm.get_destination_cooperative_tensor<decltype(tA), decltype(tB), float>(); \n"

0 commit comments

Comments
 (0)