"tests/vscode:/vscode.git/clone" did not exist on "8e6cbd621f34c0892ab5adac5259cd61401a6d2b"
graph.cc 357 Bytes
Newer Older
Minjie Wang's avatar
Minjie Wang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#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);