From 5f7fab002a6040b1b2ee82485490a73d1ff3d9e7 Mon Sep 17 00:00:00 2001
From: Wuwei Lin <wuwei@apache.org>
Date: Wed, 20 Sep 2023 14:57:22 -0700
Subject: [PATCH] [MetaSchedule] Allow generating uint random data

---
 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 17064c64ab52..35db74a1a795 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]