Skip to content

Commit 31e5272

Browse files
committed
fix dtype in test
1 parent 1b3a90e commit 31e5272

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/python/integration/test_meta_schedule_auto_tensorize.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,8 @@
3636

3737
config = ms.TuneConfig(
3838
strategy="evolutionary",
39-
num_trials_per_iter=64,
40-
max_trials_per_task=64,
39+
num_trials_per_iter=32,
40+
max_trials_per_task=32,
4141
max_trials_global=20000,
4242
)
4343

@@ -227,7 +227,7 @@ def _test_conv2d(data_dtype, sch_rules, postprocs, target):
227227

228228
relay_mod = tvm.IRModule.from_expr(conv2d)
229229

230-
data_np = np.random.uniform(1, 10, d_shape).astype("uint8")
230+
data_np = np.random.uniform(1, 10, d_shape).astype(data_dtype)
231231
weight_np = np.random.uniform(1, 10, size=w_shape).astype("int8")
232232

233233
tune_and_test(relay_mod, data_np, weight_np, "conv2d", target, sch_rules, postprocs)

0 commit comments

Comments
 (0)