Просмотр исходного кода

metal : enable tensor API for A19 (#17087)

Georgi Gerganov 2 месяцев назад
Родитель
Сommit
c27efd2bd1
1 измененных файлов с 4 добавлено и 2 удалено
  1. 4 2
      ggml/src/ggml-metal/ggml-metal-device.m

+ 4 - 2
ggml/src/ggml-metal/ggml-metal-device.m

@@ -564,8 +564,10 @@ ggml_metal_device_t ggml_metal_device_init(void) {
             // TODO: try to update the tensor API kernels to at least match the simdgroup performance
             if (getenv("GGML_METAL_TENSOR_ENABLE") == NULL &&
                 ![[dev->mtl_device name] containsString:@"M5"] &&
-                ![[dev->mtl_device name] containsString:@"M6"]) {
-                GGML_LOG_WARN("%s: tensor API disabled for pre-M5 device\n", __func__);
+                ![[dev->mtl_device name] containsString:@"M6"] &&
+                ![[dev->mtl_device name] containsString:@"A19"] &&
+                ![[dev->mtl_device name] containsString:@"A20"]) {
+                GGML_LOG_WARN("%s: tensor API disabled for pre-M5 and pre-A19 devices\n", __func__);
                 dev->props.has_tensor = false;
             }