1 files changed, 13 insertions, 0 deletions
diff --git a/extra/tensorflow/PKGBUILD b/extra/tensorflow/PKGBUILD
new file mode 100644
index 00000000..61820fe9
--- /dev/null
+++ b/extra/tensorflow/PKGBUILD
@@ -0,0 +1,13 @@
+eval "$(
+  declare -f prepare | \
+    sed '
+      s|export CC_OPT_FLAGS="-march=x86-64"|export CC_OPT_FLAGS="-march=${CARCH}"|
+    '
+)"
+
+makedepends=(
+  $(
+    printf '%s\n' "${makedepends[@]}" | \
+      sed 's/cuda//' | sed 's/cudnn//'
+  )
+)