remove print

apache · Dec 2, 2020 · b61ac98 · b61ac98
1 parent 7fffa99
commit b61ac98
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/tests/python/topi/python/test_topi_argwhere.py b/tests/python/topi/python/test_topi_argwhere.py
@@ -52,14 +52,11 @@ def check_device(device, ctx):
 
         func = tvm.build(sch, [out_shape, condition, out], device, name="argwhere")
 
-        print(func.imported_modules[0].get_source())
-
         args = [tvm.nd.array(np_shape, ctx)]
         args.append(tvm.nd.array(np_data, ctx))
         args.append(tvm.nd.empty(out.shape, ctx=ctx, dtype=condition.dtype))
         func(*args)
         np.set_printoptions(threshold=np.inf)
-        # print(args[-1].asnumpy())
         tvm.testing.assert_allclose(args[-1].asnumpy(), np.array(np_out))
 
     for target, ctx in tvm.testing.enabled_targets():