zhiics commented on a change in pull request #6353:
URL: https://github.com/apache/incubator-tvm/pull/6353#discussion_r483136682
##########
File path: tests/python/relay/dyn/test_dynamic_op_level2.py
##########
@@ -52,30 +52,84 @@ def verify_upsampling(dshape, scale_h, scale_w, layout,
method, align_corners=Fa
func = relay.Function([x, scale_h_var, scale_w_var], z)
for target, ctx in ctx_list():
- if "llvm" not in target: continue
- for kind in ["vm", "debug"]:
- mod = tvm.ir.IRModule.from_expr(func)
- intrp = relay.create_executor(kind, mod=mod, ctx=ctx,
target=target)
- op_res = intrp.evaluate()(x_data,
np.array(scale_h).astype("float32"), np.array(scale_w).astype("float32"))
- tvm.testing.assert_allclose(op_res.asnumpy(), ref_res,
rtol=1e-4, atol=1e-6)
-
- verify_upsampling((1, 16, 32, 32), 2.0, 2.0,"NCHW", "nearest_neighbor")
- verify_upsampling((1, 16, 32, 32), 2.0, 2.0, "NCHW", "bilinear", True)
- verify_upsampling((1, 16, 32, 32), 2.0, 2.0, "NHWC", "nearest_neighbor")
+ ##TODO(mbrookhart)(electriclilies): remove when VM supports
heterogeneous execution
Review comment:
ctx_list has been removed. Use enabled_targets and we can enable GPU
tests now
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
