Add thread_warp_size for Metal device in default target attributes (a…

…pache#8202)
trevor-m · Jun 17, 2021 · 5bf3de2 · 5bf3de2
1 parent 1b32467
commit 5bf3de2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/target/target_kind.cc b/src/target/target_kind.cc
@@ -350,6 +350,7 @@ TVM_REGISTER_TARGET_KIND("opencl", kDLOpenCL)
 TVM_REGISTER_TARGET_KIND("metal", kDLMetal)
     .add_attr_option<Bool>("system-lib")
     .add_attr_option<Integer>("max_num_threads", Integer(256))
+    .add_attr_option<Integer>("thread_warp_size", Integer(16))
     .set_default_keys({"metal", "gpu"});
 
 TVM_REGISTER_TARGET_KIND("vulkan", kDLVulkan)