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

…8202)
apache · Jun 7, 2021 · 3ab4a6b · 3ab4a6b
1 parent 072a3d2
commit 3ab4a6b
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)