masahi commented on a change in pull request #10561:
URL: https://github.com/apache/tvm/pull/10561#discussion_r823538769
##########
File path: src/relay/backend/te_compiler_cache.cc
##########
@@ -750,9 +774,12 @@ std::string GetUniqueName(std::string name,
std::unordered_map<std::string, int>
}
TVM_REGISTER_GLOBAL("relay.backend.LowerToTE").set_body_typed([](Function
prim_func) {
- return ScheduleBuilder(tvm::Target("ext_dev"), false).Create(prim_func,
[&](std::string name) {
- return name;
- });
+ auto tgt = tvm::Target("ext_dev");
+ LowerToTECompute lower_te_compute(tgt);
+ auto outputs = lower_te_compute.Lower(prim_func, [&](std::string name) {
return name; });
+ return CachedFunc(tgt, GlobalVar(lower_te_compute.candidate_name_),
lower_te_compute.fn_inputs_,
+ outputs, te::Schedule(), tir::PrimFunc(), {},
+ IRModule(Map<GlobalVar, BaseFunc>({})),
lower_te_compute.constant_tensors_);
Review comment:
cc @mbaret for this change
--
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]
