simple ffi example works

9445e1ae · Minjie Wang · 2694b127 · 9445e1ae
Commit 9445e1ae authored Sep 05, 2018 by Minjie Wang
Show whitespace changes
Inline Side-by-side

Showing with 19 additions and 0 deletions

src/graph/graph.cc src/graph/graph.cc +19 -0

No files found.
--- a/src/graph/graph.cc
+++ b/src/graph/graph.cc
+#include <dgl/runtime/packed_func.h>
+#include <dgl/runtime/registry.h>
+
+using namespace tvm;
+using namespace tvm::runtime;
+
+void MyAdd(TVMArgs args, TVMRetValue* rv) {
+  int a = args[0];
+  int b = args[1];
+  *rv = a + b;
+}
+
+void CallPacked() {
+  PackedFunc myadd = PackedFunc(MyAdd);
+  int c = myadd(1, 2);
+}
+
+TVM_REGISTER_GLOBAL("myadd")
+.set_body(MyAdd);