[RUNTIME] Support nop (#913)

6bf3a9d4 · Tianqi Chen · GitHub · d4269913 · 6bf3a9d4
Commit 6bf3a9d4 authored Feb 20, 2018 by Tianqi Chen Committed by GitHub Feb 20, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/runtime/graph/graph_runtime.cc
+3 -0

No files found.
--- a/src/runtime/graph/graph_runtime.cc
+++ b/src/runtime/graph/graph_runtime.cc
@@ -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;