Add MKL based tensorflow back-end

openrndr · Oct 21, 2020 · 8f4c3d7 · 8f4c3d7
1 parent 2e40843
commit 8f4c3d7
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 1 deletion.
diff --git a/build.gradle b/build.gradle
@@ -25,6 +25,7 @@ project.ext {
     gsonVersion = "2.8.6"
     antlrVersion = "4.8-1"
     tensorflowVersion = "0.2.0"
+    mklDnnVersion = "0.21.5-1.5.4"
 }
 
 switch (org.gradle.internal.os.OperatingSystem.current()) {

diff --git a/orx-tensorflow-mkl-natives-linux-x64/build.gradle b/orx-tensorflow-mkl-natives-linux-x64/build.gradle
@@ -0,0 +1,4 @@
+dependencies {
+    runtimeOnly "org.bytedeco:mkl-dnn:$mklDnnVersion:linux-x86_64"
+    runtimeOnly "org.tensorflow:tensorflow-core-api:$tensorflowVersion:linux-x86_64-mkl"
+}
diff --git a/orx-tensorflow-mkl-natives-macos/build.gradle b/orx-tensorflow-mkl-natives-macos/build.gradle
@@ -0,0 +1,4 @@
+dependencies {
+    runtimeOnly "org.bytedeco:mkl-dnn:$mklDnnVersion:macosx-x86_64"
+    runtimeOnly "org.tensorflow:tensorflow-core-api:$tensorflowVersion:macosx-x86_64-mkl"
+}
diff --git a/orx-tensorflow-mkl-natives-windows/build.gradle b/orx-tensorflow-mkl-natives-windows/build.gradle
@@ -0,0 +1,4 @@
+dependencies {
+    runtimeOnly "org.bytedeco:mkl-dnn:$mklDnnVersion:windows-x86_64"
+    runtimeOnly "org.tensorflow:tensorflow-core-api:$tensorflowVersion:windows-x86_64"
+}
diff --git a/settings.gradle b/settings.gradle
@@ -40,9 +40,12 @@ include 'openrndr-demos',
         'orx-syphon',
         'orx-temporal-blur',
         'orx-tensorflow',
+        'orx-tensorflow-mkl-natives-linux-x64',
+        'orx-tensorflow-mkl-natives-macos',
+        'orx-tensorflow-mkl-natives-windows',
         'orx-tensorflow-natives-linux-x64',
-        'orx-tensorflow-natives-windows',
         'orx-tensorflow-natives-macos',
+        'orx-tensorflow-natives-windows',
         'orx-timer',
         'orx-time-operators',
         'orx-kinect-common',