Commit 6bf3a9d4 by Tianqi Chen Committed by GitHub

[RUNTIME] Support nop (#913)

parent d4269913
......@@ -564,6 +564,9 @@ std::function<void()> GraphRuntime::CreateTVMOp(
t->shape = &(arg_ptr->shape_data[i]);
}
}
if (param.func_name == "__nop") {
return [](){};
}
// get compiled function from module.
tvm::runtime::PackedFunc pf = module_.GetFunction(param.func_name, false);
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