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]