From 8f4c3d746d1d1c13f258d4216e77318038ecd928 Mon Sep 17 00:00:00 2001 From: Edwin Jakobs Date: Wed, 21 Oct 2020 14:02:51 +0200 Subject: [PATCH] Add MKL based tensorflow back-end --- build.gradle | 1 + orx-tensorflow-mkl-natives-linux-x64/build.gradle | 4 ++++ orx-tensorflow-mkl-natives-macos/build.gradle | 4 ++++ orx-tensorflow-mkl-natives-windows/build.gradle | 4 ++++ settings.gradle | 5 ++++- 5 files changed, 17 insertions(+), 1 deletion(-) create mode 100644 orx-tensorflow-mkl-natives-linux-x64/build.gradle create mode 100644 orx-tensorflow-mkl-natives-macos/build.gradle create mode 100644 orx-tensorflow-mkl-natives-windows/build.gradle diff --git a/build.gradle b/build.gradle index 854e18700..cc20692ab 100644 --- 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 new file mode 100644 index 000000000..d269a5f4b --- /dev/null +++ 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" +} \ No newline at end of file diff --git a/orx-tensorflow-mkl-natives-macos/build.gradle b/orx-tensorflow-mkl-natives-macos/build.gradle new file mode 100644 index 000000000..5433c1b9b --- /dev/null +++ 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" +} \ No newline at end of file diff --git a/orx-tensorflow-mkl-natives-windows/build.gradle b/orx-tensorflow-mkl-natives-windows/build.gradle new file mode 100644 index 000000000..e4f8e4214 --- /dev/null +++ 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" +} \ No newline at end of file diff --git a/settings.gradle b/settings.gradle index 163b1d77b..9d95ad152 100644 --- 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',