mbs-octoml commented on code in PR #12085:
URL: https://github.com/apache/tvm/pull/12085#discussion_r920537111
##########
tests/python/relay/test_pass_plan_devices.py:
##########
@@ -241,6 +243,70 @@ def ref(a, b, c, d):
exercise(input(), expected(), ref, rands((5, 7), 4))
+def test_left_add_on_cpu_via_copy_as_map():
+ metatable = {"VirtualDevice": [CPU, GPU]}
+
+ # As for test_left_add_on_cpu, but with an explicit device_copy.
+ def input():
+ return tvm.parser.parse(
+ """
+ #[version = "0.0.5"]
+ def @main(%a: Tensor[(5, 7), float32], %b: Tensor[(5, 7), float32],
+ %c: Tensor[(5, 7), float32], %d: Tensor[(5, 7),
float32]) {
+ %0 = add(%a, %b);
+ %1 = device_copy(%0, src_virtual_device=meta[VirtualDevice][0],
dst_virtual_device=meta[VirtualDevice][1]);
+ %2 = add(%c, %d);
+ subtract(%1, %2)
+ }
+ """,
+ "from_string",
+ None,
+ metatable,
+ )
+
+ config = tvm.target.make_compilation_config(CTXT, TARGETS, HOST_TARGET)
+ actual_mod = relay.transform.InferType()(input())
+ actual_mod = relay.transform.PlanDevices(config)(actual_mod)
+ actual_mod = relay.transform.CapturePostDfsIndexInSpans()(actual_mod)
+
+ # Actual looks like:
Review Comment:
It doesn't hurt though it is a repeat of the above test. (Tho I parse and
structural assert not str assert...). Thanks.
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
