diff --git a/ggml-metal.m b/ggml-metal.m index 4f3f14e24..cc353f170 100644 --- a/ggml-metal.m +++ b/ggml-metal.m @@ -1,4 +1,5 @@ #import "ggml-metal.h" +#include #import "ggml.h" @@ -174,9 +175,39 @@ struct ggml_metal_context * ggml_metal_init(int n_cb) { NSError * error = nil; //NSString * path = [[NSBundle mainBundle] pathForResource:@"../../examples/metal/metal" ofType:@"metal"]; - NSBundle * bundle = [NSBundle bundleForClass:[GGMLMetalClass class]]; - NSString * path = [bundle pathForResource:@"ggml-metal" ofType:@"metal"]; - metal_printf("%s: loading '%s'\n", __func__, [path UTF8String]); + const char* v = getenv("GGML_METAL_PATH"); + NSString * path; + BOOL doLegacyCode = false; + if (v) { + path = [NSString stringWithCString: v encoding: NSUTF8StringEncoding]; + NSFileManager *fileManager = [NSFileManager defaultManager]; + BOOL isDir; + if ([fileManager fileExistsAtPath:path isDirectory: &isDir]) { + if (isDir) { + path = [path stringByAppendingString: @"/ggml-metal.metal"]; + if ([fileManager fileExistsAtPath:path]) { + fprintf(stderr, "%s: loading '%s'\n", __func__, [path UTF8String]); + } else { + // Use original code + doLegacyCode = true; + } + } else { + fprintf(stderr, "%s: loading '%s'\n", __func__, [path UTF8String]); + } + } else { + // Use original code + doLegacyCode = true; + } + } else { + // Use original code + doLegacyCode = true; + } + + if (doLegacyCode) { + NSBundle * bundle = [NSBundle bundleForClass:[GGMLMetalClass class]]; + path = [bundle pathForResource:@"ggml-metal" ofType:@"metal"]; + fprintf(stderr, "%s: loading '%s'\n", __func__, [path UTF8String]); + } NSString * src = [NSString stringWithContentsOfFile:path encoding:NSUTF8StringEncoding error:&error]; if (error) {