Skip to content

Commit

Permalink
Correctly set support_simdgroup_reduction and support_simdgroup_mm on…
Browse files Browse the repository at this point in the history
… iPhone/iPad
  • Loading branch information
azarovalex committed Jan 13, 2024
1 parent 76484fb commit 04c99e8
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions ggml-metal.m
Original file line number Diff line number Diff line change
Expand Up @@ -330,7 +330,6 @@ static void ggml_metal_log(enum ggml_log_level level, const char * format, ...){
}
}

#if TARGET_OS_OSX
// print MTL GPU family:
GGML_METAL_LOG_INFO("%s: GPU name: %s\n", __func__, [[ctx->device name] UTF8String]);

Expand Down Expand Up @@ -366,7 +365,7 @@ static void ggml_metal_log(enum ggml_log_level level, const char * format, ...){
ctx->support_simdgroup_reduction |= [ctx->device supportsFamily:MTLGPUFamilyMetal3];

ctx->support_simdgroup_mm = [ctx->device supportsFamily:MTLGPUFamilyApple7];

#if TARGET_OS_OSX
GGML_METAL_LOG_INFO("%s: simdgroup reduction support = %s\n", __func__, ctx->support_simdgroup_reduction ? "true" : "false");
GGML_METAL_LOG_INFO("%s: simdgroup matrix mul. support = %s\n", __func__, ctx->support_simdgroup_mm ? "true" : "false");
GGML_METAL_LOG_INFO("%s: hasUnifiedMemory = %s\n", __func__, ctx->device.hasUnifiedMemory ? "true" : "false");
Expand Down

0 comments on commit 04c99e8

Please sign in to comment.