replace dynamic cumsum test with dynamic argsort for now

apache · Mar 9, 2021 · c3b8c96 · c3b8c96
1 parent d7f983c
commit c3b8c96
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/unittest/test_target_codegen_spirv.py b/tests/python/unittest/test_target_codegen_spirv.py
@@ -97,9 +97,9 @@ def check_mod(mod, x_np, res_np):
     dtype = "int32"
     x = relay.var("x", shape=(relay.Any(),), dtype=dtype)
     mod = tvm.IRModule()
-    mod["main"] = relay.Function([x], relay.cumsum(x))
+    mod["main"] = relay.Function([x], relay.argsort(x))
     x_np = np.random.randint(0, high=10, size=(10,)).astype(dtype)
-    res_np = np.cumsum(x_np)
+    res_np = np.argsort(x_np)
 
     check_mod(mod, x_np, res_np)