Fixes hip_hcc library path (#6)

Co-authored-by: James Letendre <James.Letendre@gmail.com>

3 files changed, 24 insertions, 4 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 658b143d2ae9..65eed5aa1c71 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,7 +1,7 @@
 pkgbase = tensorflow-rocm
 	pkgdesc = Library for computation using data flow graphs for scalable machine learning
 	pkgver = 2.3.0
-	pkgrel = 1
+	pkgrel = 2
 	url = https://www.tensorflow.org/
 	arch = x86_64
 	license = APACHE
@@ -26,9 +26,11 @@ pkgbase = tensorflow-rocm
 	source = tensorflow-rocm-2.3.0.tar.gz::https://github.com/tensorflow/tensorflow/archive/v2.3.0.tar.gz
 	source = numpy1.20.patch::https://github.com/tensorflow/tensorflow/commit/75ea0b31477d6ba9e990e296bbbd8ca4e7eebadf.patch
 	source = build-against-actual-mkl.patch
+	source = fix_hip_hcc_path.patch
 	sha512sums = 86aa087ea84dac1ecc1023b23a378100d41cc6778ccd20404a4b955fc67cef11b3dc08abcc5b88020124d221e6fb172b33bd5206e9c9db6bc8fbeed399917eac
 	sha512sums = df2e0373e2f63b8766f31933f7db57f6a7559b8f03af1db51644fba87731451a7cd3895529a3192e5394612fcb42f245b794b1c9ca3c05881ca03a547c8c9acc
 	sha512sums = e51e3f3dced121db3a09fbdaefd33555536095584b72a5eb6f302fa6fa68ab56ea45e8a847ec90ff4ba076db312c06f91ff672e08e95263c658526582494ce08
+	sha512sums = 2ee01810efb836a746bb46376f1a0463019ff3433c2761fa82bbcf8dc805a07225a8626e7eb2a302b95f244d2d037e1795bc50194ac41f26c456fc6293ca8a74
 
 pkgname = tensorflow-rocm
 	pkgdesc = Library for computation using data flow graphs for scalable machine learning (with ROCM)
diff --git a/PKGBUILD b/PKGBUILD
index 7841c3bad69a..30e4c9979c4d 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -8,7 +8,7 @@ pkgbase=tensorflow-rocm
 pkgname=(tensorflow-rocm tensorflow-opt-rocm python-tensorflow-rocm python-tensorflow-opt-rocm)
 pkgver=2.3.0
 _pkgver=2.3.0
-pkgrel=1
+pkgrel=2
 pkgdesc="Library for computation using data flow graphs for scalable machine learning"
 url="https://www.tensorflow.org/"
 license=('APACHE')
@@ -20,10 +20,12 @@ makedepends=('bazel' 'python-numpy' 'rocm' 'rocm-libs' 'miopen' 'rccl' 'git' 'gc
 optdepends=('tensorboard: Tensorflow visualization toolkit')
 source=("$pkgname-$pkgver.tar.gz::https://github.com/tensorflow/tensorflow/archive/v${_pkgver}.tar.gz"
         numpy1.20.patch::https://github.com/tensorflow/tensorflow/commit/75ea0b31477d6ba9e990e296bbbd8ca4e7eebadf.patch
-        build-against-actual-mkl.patch)
+        build-against-actual-mkl.patch
+        fix_hip_hcc_path.patch)
 sha512sums=('86aa087ea84dac1ecc1023b23a378100d41cc6778ccd20404a4b955fc67cef11b3dc08abcc5b88020124d221e6fb172b33bd5206e9c9db6bc8fbeed399917eac'
             'df2e0373e2f63b8766f31933f7db57f6a7559b8f03af1db51644fba87731451a7cd3895529a3192e5394612fcb42f245b794b1c9ca3c05881ca03a547c8c9acc'
-            'e51e3f3dced121db3a09fbdaefd33555536095584b72a5eb6f302fa6fa68ab56ea45e8a847ec90ff4ba076db312c06f91ff672e08e95263c658526582494ce08')
+            'e51e3f3dced121db3a09fbdaefd33555536095584b72a5eb6f302fa6fa68ab56ea45e8a847ec90ff4ba076db312c06f91ff672e08e95263c658526582494ce08'
+            '2ee01810efb836a746bb46376f1a0463019ff3433c2761fa82bbcf8dc805a07225a8626e7eb2a302b95f244d2d037e1795bc50194ac41f26c456fc6293ca8a74')
 
 get_pyver () {
   python -c 'import sys; print(str(sys.version_info[0]) + "." + str(sys.version_info[1]))'
@@ -52,6 +54,9 @@ prepare() {
 
   patch -Np1 -d tensorflow-${_pkgver} -i "$srcdir"/numpy1.20.patch
 
+  # Fix hip_hcc path
+  patch -Np1 -d tensorflow-${_pkgver} -i "$srcdir"/fix_hip_hcc_path.patch
+
   cp -r tensorflow-${_pkgver} tensorflow-${_pkgver}-rocm
   cp -r tensorflow-${_pkgver} tensorflow-${_pkgver}-opt-rocm
 
diff --git a/fix_hip_hcc_path.patch b/fix_hip_hcc_path.patch
new file mode 100644
index 000000000000..8f1a3d80288d
--- /dev/null
+++ b/fix_hip_hcc_path.patch
@@ -0,0 +1,13 @@
+diff --git a/third_party/gpus/rocm_configure.bzl b/third_party/gpus/rocm_configure.bzl
+index 24829eb..4cfec24 100644
+--- a/third_party/gpus/rocm_configure.bzl
++++ b/third_party/gpus/rocm_configure.bzl
+@@ -440,7 +440,7 @@ def _find_libs(repository_ctx, rocm_config, bash_bin):
+     libs_paths = [
+         (name, _rocm_lib_paths(repository_ctx, name, path))
+         for name, path in [
+-            ("hip_hcc", rocm_config.rocm_toolkit_path),
++            ("hip_hcc", rocm_config.rocm_toolkit_path + "/hip"),
+             ("rocblas", rocm_config.rocm_toolkit_path + "/rocblas"),
+             ("rocfft", rocm_config.rocm_toolkit_path + "/rocfft"),
+             ("hiprand", rocm_config.rocm_toolkit_path + "/hiprand"),