metal : enable tensor API for A19 (#17087)
This commit is contained in:
parent
df70bedda7
commit
c27efd2bd1
|
|
@ -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
|
// TODO: try to update the tensor API kernels to at least match the simdgroup performance
|
||||||
if (getenv("GGML_METAL_TENSOR_ENABLE") == NULL &&
|
if (getenv("GGML_METAL_TENSOR_ENABLE") == NULL &&
|
||||||
![[dev->mtl_device name] containsString:@"M5"] &&
|
![[dev->mtl_device name] containsString:@"M5"] &&
|
||||||
![[dev->mtl_device name] containsString:@"M6"]) {
|
![[dev->mtl_device name] containsString:@"M6"] &&
|
||||||
GGML_LOG_WARN("%s: tensor API disabled for pre-M5 device\n", __func__);
|
![[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;
|
dev->props.has_tensor = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue