Fix i386 test

apache · Jun 1, 2020 · 7bf2a77 · 7bf2a77
1 parent a52bb3f
commit 7bf2a77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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