Skip to content

Commit 7c4074d

Browse files
committed
[Metal] Enable Metal 4 shader compilation
Use the latest SDK language version when available so generated MPP tensor_ops shaders can compile on M5+ while preserving the existing fallback for older SDKs.
1 parent 4052880 commit 7c4074d

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/runtime/metal/metal_module.mm

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,11 @@ int GetPropertyMask() const final {
124124

125125
if (fmt_ == "metal") {
126126
MTLCompileOptions* opts = [MTLCompileOptions alloc];
127+
#if defined(MTLLanguageVersion4_0)
128+
opts.languageVersion = MTLLanguageVersion4_0;
129+
#else
127130
opts.languageVersion = MTLLanguageVersion2_3;
131+
#endif
128132
opts.fastMathEnabled = YES;
129133
// opts = nil;
130134
// Per-kernel payload is bytes; treat as UTF-8 MSL source.

0 commit comments

Comments
 (0)