[tvm] branch main updated: [MetaSchedule] Allow generating uint random data (#15792)

[sanirudh]

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

sanirudh 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 88e590a129 [MetaSchedule] Allow generating uint random data (#15792)
88e590a129 is described below

commit 88e590a1296451379874ab11c5b13b01a0428193
Author: Wuwei Lin <wu...@apache.org>
AuthorDate: Thu Sep 21 01:42:07 2023 -0700

    [MetaSchedule] Allow generating uint random data (#15792)
---
 python/tvm/meta_schedule/testing/tune_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/tvm/meta_schedule/testing/tune_utils.py 
b/python/tvm/meta_schedule/testing/tune_utils.py
index 17064c64ab..35db74a1a7 100644
--- a/python/tvm/meta_schedule/testing/tune_utils.py
+++ b/python/tvm/meta_schedule/testing/tune_utils.py
@@ -56,7 +56,9 @@ def generate_input_data(
         "uint8": (0, 255),
         "int8": (-128, 127),
         "int32": (0, 10000),
+        "uint32": (0, 10000),
         "int64": (0, 10000),
+        "uint64": (0, 10000),
     }
     if input_dtype in range_map:
         _low, _high = range_map[input_dtype]