Skip to content

Commit c27efd2

Browse files
authored
metal : enable tensor API for A19 (#17087)
1 parent df70bed commit c27efd2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -564,8 +564,10 @@ ggml_metal_device_t ggml_metal_device_init(void) {
564564
// TODO: try to update the tensor API kernels to at least match the simdgroup performance
565565
if (getenv("GGML_METAL_TENSOR_ENABLE") == NULL &&
566566
![[dev->mtl_device name] containsString:@"M5"] &&
567-
![[dev->mtl_device name] containsString:@"M6"]) {
568-
GGML_LOG_WARN("%s: tensor API disabled for pre-M5 device\n", __func__);
567+
![[dev->mtl_device name] containsString:@"M6"] &&
568+
![[dev->mtl_device name] containsString:@"A19"] &&
569+
![[dev->mtl_device name] containsString:@"A20"]) {
570+
GGML_LOG_WARN("%s: tensor API disabled for pre-M5 and pre-A19 devices\n", __func__);
569571
dev->props.has_tensor = false;
570572
}
571573

0 commit comments

Comments
 (0)