[tvm] branch main updated: [Hotfix] A line is accidentally removed in `Verify-GPU-Code`

[junrushao]

This is an automated email from the ASF dual-hosted git repository.

junrushao pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 5b5bf75  [Hotfix] A line is accidentally removed in `Verify-GPU-Code`
5b5bf75 is described below

commit 5b5bf75a089a243b2f947f4d7d05e7c1e879b7e0
Author: Junru Shao <junrushao1...@gmail.com>
AuthorDate: Tue Mar 22 00:17:37 2022 -0700

    [Hotfix] A line is accidentally removed in `Verify-GPU-Code`
---
 python/tvm/meta_schedule/testing/run_subgraph_auto_scheduler.py | 2 ++
 src/meta_schedule/postproc/verify_gpu_code.cc                   | 1 +
 2 files changed, 3 insertions(+)

diff --git a/python/tvm/meta_schedule/testing/run_subgraph_auto_scheduler.py 
b/python/tvm/meta_schedule/testing/run_subgraph_auto_scheduler.py
index 4649a8b..00edb7d 100644
--- a/python/tvm/meta_schedule/testing/run_subgraph_auto_scheduler.py
+++ b/python/tvm/meta_schedule/testing/run_subgraph_auto_scheduler.py
@@ -91,6 +91,8 @@ def main():
             cache_line_bytes=64,
             
max_shared_memory_per_block=int(ARGS.target.attrs["max_shared_memory_per_block"]),
             
max_threads_per_block=int(ARGS.target.attrs["max_threads_per_block"]),
+            # The value `max_local_memory_per_block` is not used in 
AutoScheduler,
+            # but is required by the API.
             max_local_memory_per_block=12345678,
             max_vthread_extent=8,
             warp_size=32,
diff --git a/src/meta_schedule/postproc/verify_gpu_code.cc 
b/src/meta_schedule/postproc/verify_gpu_code.cc
index 7d4a716..6743598 100644
--- a/src/meta_schedule/postproc/verify_gpu_code.cc
+++ b/src/meta_schedule/postproc/verify_gpu_code.cc
@@ -110,6 +110,7 @@ class VerifyGPUCodeNode : public PostprocNode {
     Target target = context->target.value();
     this->target_constraints_ = Map<String, PrimExpr>{
         {"max_shared_memory_per_block", Extract(target, 
"max_shared_memory_per_block")},
+        {"max_threads_per_block", Extract(target, "max_threads_per_block")},
         {"max_vthread", Integer(8)},
         {"max_vector_bytes", Integer(16)},
     };