hi, bro
tvm-master\src\pass\split_host_device.cc:
…
Stmt Mutate_(const AttrStmt *op, const Stmt& s) final {
if (op->attr_key == attr::thread_extent ||
op->attr_key == attr::pipeline_exec_scope) {
return SplitDeviceFunc(s);
}
…
we want to split the code from host to device by split_host_device.cc,
but our code is not for GPU, so there is no thread info.
one world, attr::thread_extent and attr::pipeline_exec_scope can not be used for split the code,
so we want to split the code flexibly,
and we modify the code by adding one attribute,
so please help to check whether it is ok for the modification, and whether the code can be update to yours.
thank you.