fix: guard MTLLanguageVersion4_0 with SDK version check

MTLLanguageVersion4_0 is only available in macOS 26+ SDK. Fall back
to 3_1 (macOS 14+) or 3_0 for older SDKs to fix CI builds.

Author: oraluben

python/tvm/script/ir_builder/tir/ir.py

@@ -1432,311 +1432,158 @@ def func(
 
 
 if TYPE_CHECKING:
-
     class int8: ...
-
     class int16: ...
-
     class int32: ...
-
     class int64: ...
-
     class int8x4: ...
-
     class int16x4: ...
-
     class int32x4: ...
-
     class int64x4: ...
-
     class int8x8: ...
-
     class int16x8: ...
-
     class int32x8: ...
-
     class int64x8: ...
-
     class int8x16: ...
-
     class int16x16: ...
-
     class int32x16: ...
-
     class int64x16: ...
-
     class int8x32: ...
-
     class int16x32: ...
-
     class int32x32: ...
-
     class int64x32: ...
-
     class int8x64: ...
-
     class int16x64: ...
-
     class int32x64: ...
-
     class int64x64: ...
-
     class uint8: ...
-
     class uint16: ...
-
     class uint32: ...
-
     class uint64: ...
-
     class uint8x4: ...
-
     class uint16x4: ...
-
     class uint32x4: ...
-
     class uint64x4: ...
-
     class uint8x8: ...
-
     class uint16x8: ...
-
     class uint32x8: ...
-
     class uint64x8: ...
-
     class uint8x16: ...
-
     class uint16x16: ...
-
     class uint32x16: ...
-
     class uint64x16: ...
-
     class uint8x32: ...
-
     class uint16x32: ...
-
     class uint32x32: ...
-
     class uint64x32: ...
-
     class uint8x64: ...
-
     class uint16x64: ...
-
     class uint32x64: ...
-
     class uint64x64: ...
-
     class float16: ...
-
     class float32: ...
-
     class float64: ...
-
     class float16x2: ...
-
     class float32x2: ...
-
     class float64x2: ...
-
     class float16x4: ...
-
     class float32x4: ...
-
     class float64x4: ...
-
     class float16x8: ...
-
     class float32x8: ...
-
     class float64x8: ...
-
     class float16x16: ...
-
     class float32x16: ...
-
     class float64x16: ...
-
     class float16x32: ...
-
     class float32x32: ...
-
     class float64x32: ...
-
     class float16x64: ...
-
     class float32x64: ...
-
     class float64x64: ...
-
     class float8_e3m4: ...
-
     class float8_e3m4x2: ...
-
     class float8_e3m4x4: ...
-
     class float8_e3m4x8: ...
-
     class float8_e3m4x16: ...
-
     class float8_e3m4x32: ...
-
     class float8_e3m4x64: ...
-
     class float8_e4m3: ...
-
     class float8_e4m3x2: ...
-
     class float8_e4m3x4: ...
-
     class float8_e4m3x8: ...
-
     class float8_e4m3x16: ...
-
     class float8_e4m3x32: ...
-
     class float8_e4m3x64: ...
-
     class float8_e4m3b11fnuz: ...
-
     class float8_e4m3b11fnuzx2: ...
-
     class float8_e4m3b11fnuzx4: ...
-
     class float8_e4m3b11fnuzx8: ...
-
     class float8_e4m3b11fnuzx16: ...
-
     class float8_e4m3b11fnuzx32: ...
-
     class float8_e4m3b11fnuzx64: ...
-
     class float8_e4m3fn: ...
-
     class float8_e4m3fnx2: ...
-
     class float8_e4m3fnx4: ...
-
     class float8_e4m3fnx8: ...
-
     class float8_e4m3fnx16: ...
-
     class float8_e4m3fnx32: ...
-
     class float8_e4m3fnx64: ...
-
     class float8_e4m3fnuz: ...
-
     class float8_e4m3fnuzx2: ...
-
     class float8_e4m3fnuzx4: ...
-
     class float8_e4m3fnuzx8: ...
-
     class float8_e4m3fnuzx16: ...
-
     class float8_e4m3fnuzx32: ...
-
     class float8_e4m3fnuzx64: ...
-
     class float8_e5m2: ...
-
     class float8_e5m2x2: ...
-
     class float8_e5m2x4: ...
-
     class float8_e5m2x8: ...
-
     class float8_e5m2x16: ...
-
     class float8_e5m2x32: ...
-
     class float8_e5m2x64: ...
-
     class float8_e5m2fnuz: ...
-
     class float8_e5m2fnuzx2: ...
-
     class float8_e5m2fnuzx4: ...
-
     class float8_e5m2fnuzx8: ...
-
     class float8_e5m2fnuzx16: ...
-
     class float8_e5m2fnuzx32: ...
-
     class float8_e5m2fnuzx64: ...
-
     class float8_e8m0fnu: ...
-
     class float8_e8m0fnux2: ...
-
     class float8_e8m0fnux4: ...
-
     class float8_e8m0fnux8: ...
-
     class float8_e8m0fnux16: ...
-
     class float8_e8m0fnux32: ...
-
     class float8_e8m0fnux64: ...
-
     class float6_e2m3fn: ...
-
     class float6_e2m3fnx2: ...
-
     class float6_e2m3fnx4: ...
-
     class float6_e2m3fnx8: ...
-
     class float6_e2m3fnx16: ...
-
     class float6_e2m3fnx32: ...
-
     class float6_e2m3fnx64: ...
-
     class float6_e3m2fn: ...
-
     class float6_e3m2fnx2: ...
-
     class float6_e3m2fnx4: ...
-
     class float6_e3m2fnx8: ...
-
     class float6_e3m2fnx16: ...
-
     class float6_e3m2fnx32: ...
-
     class float6_e3m2fnx64: ...
-
     class float4_e2m1fn: ...
-
     class float4_e2m1fnx2: ...
-
     class float4_e2m1fnx4: ...
-
     class float4_e2m1fnx8: ...
-
     class float4_e2m1fnx16: ...
-
     class float4_e2m1fnx32: ...
-
     class float4_e2m1fnx64: ...
-
     class bfloat16: ...
-
     class bfloat16x2: ...
-
     class bfloat16x4: ...
-
     class bfloat16x8: ...
-
     class bfloat16x16: ...
-
     class bfloat16x32: ...
-
     class bfloat16x64: ...
     class tfloat32: ...
     class tfloat32x2: ...

src/runtime/metal/metal_module.mm

@@ -103,7 +103,13 @@ void WriteToFile(const ffi::String& file_name, const ffi::String& format) const
 
     if (fmt_ == "metal") {
       MTLCompileOptions* opts = [MTLCompileOptions alloc];
+#if defined(__MAC_OS_X_VERSION_MAX_ALLOWED) && __MAC_OS_X_VERSION_MAX_ALLOWED >= 260000
       opts.languageVersion = MTLLanguageVersion4_0;
+#elif defined(__MAC_OS_X_VERSION_MAX_ALLOWED) && __MAC_OS_X_VERSION_MAX_ALLOWED >= 140000
+      opts.languageVersion = MTLLanguageVersion3_1;
+#else
+      opts.languageVersion = MTLLanguageVersion3_0;
+#endif
       opts.fastMathEnabled = YES;
       // opts = nil;
       lib =