From 7bf2a77c5eb4e389753a6319df287a64fe8a0be6 Mon Sep 17 00:00:00 2001 From: Matthew Brookhart Date: Mon, 1 Jun 2020 15:55:11 -0700 Subject: [PATCH] Fix i386 test --- tests/python/relay/test_op_level3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/relay/test_op_level3.py b/tests/python/relay/test_op_level3.py index ae13130c00b53..5a1184108c7ee 100644 --- a/tests/python/relay/test_op_level3.py +++ b/tests/python/relay/test_op_level3.py @@ -686,7 +686,7 @@ def verify_scatter(dshape, ishape, axis=0): data_np = np.random.uniform(size=dshape).astype("float32") updates_np = np.random.uniform(size=ishape).astype("float32") - indices_np = np.random.randint(-dshape[axis], dshape[axis] - 1, ishape) + indices_np = np.random.randint(-dshape[axis], dshape[axis] - 1, ishape).astype("int64") ref_res = ref_scatter(data_np, indices_np, updates_np, axis) # TODO(mbrookhart): expand testing when adding more backend schedules