Skip to content

T
tic
Commit 6bf3a9d4 by Tianqi Chen Committed by GitHub
Options

[RUNTIME] Support nop (#913)

parent d4269913
Showing with 3 additions and 0 deletions
src/runtime/graph/graph_runtime.cc
...@@ -564,6 +564,9 @@ std::function<void()> GraphRuntime::CreateTVMOp( ...@@ -564,6 +564,9 @@ std::function<void()> GraphRuntime::CreateTVMOp(
t->shape = &(arg_ptr->shape_data[i]); t->shape = &(arg_ptr->shape_data[i]);
} }
} }
if (param.func_name == "__nop") {
return [](){};
}
// get compiled function from module. // get compiled function from module.
tvm::runtime::PackedFunc pf = module_.GetFunction(param.func_name, false); tvm::runtime::PackedFunc pf = module_.GetFunction(param.func_name, false);
CHECK(pf != nullptr) << "no such function in module: " << param.func_name; CHECK(pf != nullptr) << "no such function in module: " << param.func_name;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment