electriclilies commented on a change in pull request #9312:
URL: https://github.com/apache/tvm/pull/9312#discussion_r744014493
##########
File path: src/relay/transforms/device_domains.cc
##########
@@ -47,20 +49,22 @@ constexpr size_t mix(size_t h1, size_t h2) {
* See te_compiler.cc for where this rewriting occurs.
*/
DeviceCopyProps GetPrimitiveDeviceCopyProps(const CallNode* call_node) {
- auto tir_call_attrs = call_node->attrs.as<TIRCallAttrs>();
- if (tir_call_attrs == nullptr) {
- return {};
- }
- if (tir_call_attrs->metadata.count("source_device") != 1 ||
- tir_call_attrs->metadata.count("dst_device") != 1) {
- return {};
+ if (call_node->op == CallLoweredOp()) {
Review comment:
we decided we don't need that actually so this is OK as is.
--
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]
