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

…pache#8202)
neo-ai · Jun 17, 2021 · 9578c1a · 9578c1a
1 parent 10874d6
commit 9578c1a
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)