[FIX][LLVM] Workaround -mcpu=apple-latest for llvm above 18.0 (apache…

…#17492)
cbalint13 · Nov 30, 2024 · 872e66f · 872e66f
1 parent 7ae7ea8
commit 872e66f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/target/tag.cc b/src/target/tag.cc
@@ -429,6 +429,17 @@ TVM_REGISTER_TAG_AWS_C5("aws/cpu/c5.24xlarge", 48, "cascadelake");
 
 #undef TVM_REGISTER_TAG_AWS_C5
 
+#if TVM_LLVM_VERSION >= 190
+#define TVM_REGISTER_METAL_GPU_TAG(Name, ThreadsPerBlock, SharedMem, WarpSize)   \
+  TVM_REGISTER_TARGET_TAG(Name).set_config(                                      \
+      {{"kind", String("metal")},                                                \
+       {"max_threads_per_block", runtime::Int(ThreadsPerBlock)},                 \
+       {"max_shared_memory_per_block", runtime::Int(SharedMem)},                 \
+       {"thread_warp_size", runtime::Int(WarpSize)},                             \
+       {"host", Map<String, ObjectRef>{{"kind", String("llvm")},                 \
+                                       {"mtriple", String("arm64-apple-macos")}, \
+                                       {"mcpu", String("apple-m4")}}}});
+#else
 #define TVM_REGISTER_METAL_GPU_TAG(Name, ThreadsPerBlock, SharedMem, WarpSize)   \
   TVM_REGISTER_TARGET_TAG(Name).set_config(                                      \
       {{"kind", String("metal")},                                                \
@@ -438,6 +449,7 @@ TVM_REGISTER_TAG_AWS_C5("aws/cpu/c5.24xlarge", 48, "cascadelake");
        {"host", Map<String, ObjectRef>{{"kind", String("llvm")},                 \
                                        {"mtriple", String("arm64-apple-macos")}, \
                                        {"mcpu", String("apple-latest")}}}});
+#endif
 
 #if TVM_LLVM_HAS_AARCH64_TARGET
 TVM_REGISTER_METAL_GPU_TAG("apple/m1-gpu", 1024, 32768, 32);