adding example to test accelerator loading

apache · Jul 12, 2019 · 448e950 · 448e950
1 parent f9a6e8b
commit 448e950
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/example/accel_api/Makefile b/example/accel_api/Makefile
@@ -0,0 +1,5 @@
+all:
+	g++ -shared -fPIC myacc.cpp -o libmyacc.so -I ../../include/mxnet
+
+clean:
+	rm -rf libmyacc.so
diff --git a/example/accel_api/myacc.cpp b/example/accel_api/myacc.cpp
@@ -0,0 +1,6 @@
+#include <string>
+
+std::string getAccName() {
+  return std::string("myacc");
+}
+
diff --git a/example/accel_api/test.py b/example/accel_api/test.py
@@ -0,0 +1,6 @@
+import mxnet as mx
+
+ctx = mx.context.load_acc('libmyacc.so')
+print(ctx)
+
+mx.nd.empty(4,ctx=ctx)