mbs-octoml commented on a change in pull request #9134:
URL: https://github.com/apache/tvm/pull/9134#discussion_r719856466
##########
File path: src/relay/backend/vm/compiler.cc
##########
@@ -1083,13 +1082,21 @@ IRModule VMCompiler::OptimizeModule(IRModule mod, const
TargetsMap& targets_arg,
pass_seqs.push_back(transform::InferType());
pass_seqs.push_back(transform::LabelOps());
+ tec::TargetMap targets_;
+ tec::DeviceMap device_map;
+
+ targets_ = GetTecTargetMapFromTargetMap(targets);
+
+ pass_seqs.push_back(tec::LowerTEPass(targets_, device_map, "vm_mod",
[this](Function func) {}));
Review comment:
For others: This PR also needs to remove the direct calls to the
TECompiler Lower and LowerShapeFunc which are a bit fiddly to change over. I
believe Michalis is working on that.
--
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]
