[Target] Add MetaSchedule-compatible attributes to OpenCL (#15166)

MetaSchedule prefers `max_threads_per_block` and `max_shared_memory_per_block` for GPU targets. Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
apache · Jun 27, 2023 · 48f295f · 48f295f
1 parent 55df810
commit 48f295f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/target/target_kind.cc b/src/target/target_kind.cc
@@ -362,6 +362,8 @@ TVM_REGISTER_TARGET_KIND("rocm", kDLROCM)
     .set_target_parser(UpdateROCmAttrs);
 
 TVM_REGISTER_TARGET_KIND("opencl", kDLOpenCL)
+    .add_attr_option<Integer>("max_threads_per_block", Integer(256))
+    .add_attr_option<Integer>("max_shared_memory_per_block", Integer(16384))
     .add_attr_option<Integer>("max_num_threads", Integer(256))
     .add_attr_option<Integer>("thread_warp_size", Integer(1))
     .add_attr_option<Integer>("texture_spatial_limit", Integer(16384))