Skip to content

Commit e6373fc

Browse files
committed
metal : add env for disabling tensor API
1 parent 49c1ac0 commit e6373fc

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -476,14 +476,16 @@ ggml_metal_device_t ggml_metal_device_init(void) {
476476
dev->props.has_bfloat |= [dev->mtl_device supportsFamily:MTLGPUFamilyApple6];
477477

478478
dev->props.has_tensor = [dev->mtl_device supportsFamily:MTLGPUFamilyMetal4_GGML];
479+
if (getenv("GGML_METAL_TENSOR_DISABLE") != NULL) {
480+
dev->props.has_tensor = false;
481+
}
479482

480483
dev->props.use_residency_sets = true;
481484
#if defined(GGML_METAL_HAS_RESIDENCY_SETS)
482485
dev->props.use_residency_sets = getenv("GGML_METAL_NO_RESIDENCY") == nil;
483486
#endif
484487

485488
dev->props.use_shared_buffers = dev->props.has_unified_memory;
486-
487489
if (getenv("GGML_METAL_SHARED_BUFFERS_DISABLE") != NULL) {
488490
dev->props.use_shared_buffers = false;
489491
}

0 commit comments

Comments
 (0)