commit: 17596353d6219013a251b16c096aae3f42c90ac0 Author: Antonin Riha <antonin.riha <AT> bluedynamic <DOT> cz> AuthorDate: Fri Jan 14 14:52:19 2022 +0000 Commit: Andrew Ammerlaan <andrewammerlaan <AT> gentoo <DOT> org> CommitDate: Fri Jan 14 15:19:52 2022 +0000 URL: https://gitweb.gentoo.org/proj/sci.git/commit/?id=17596353
Fixed thrust include path for rocm platform Signed-off-by: Antonín Říha <riha.antonin <AT> outlook.com> Closes: https://github.com/gentoo/sci/pull/1133 Signed-off-by: Andrew Ammerlaan <andrewammerlaan <AT> gentoo.org> sci-libs/pytorch/pytorch-1.10.1.ebuild | 1 + 1 file changed, 1 insertion(+) diff --git a/sci-libs/pytorch/pytorch-1.10.1.ebuild b/sci-libs/pytorch/pytorch-1.10.1.ebuild index 948d595a7..59947e530 100644 --- a/sci-libs/pytorch/pytorch-1.10.1.ebuild +++ b/sci-libs/pytorch/pytorch-1.10.1.ebuild @@ -230,6 +230,7 @@ src_prepare() { export PYTORCH_ROCM_ARCH="${AMDGPU_TARGETS}" sed -e "/set(roctracer_INCLUDE_DIRS/s,\${ROCTRACER_PATH}/include,${EPREFIX}/usr/include/roctracer," \ -e "/PYTORCH_HIP_HCC_LIBRARIES/s,\${HIP_PATH}/lib,${EPREFIX}/usr/lib/hip/lib," \ + -e "/set(roctracer_INCLUDE_DIRS/a\ set(thrust_INCLUDE_DIRS ${EPREFIX}/usr/include/rocthrust)" \ -e "s,\${ROCTRACER_PATH}/lib,${EPREFIX}/usr/lib64/roctracer," \ -e "/READ.*\.info\/version-dev/c\ set(ROCM_VERSION_DEV_RAW ${ROCM_VERSION})" \ -i cmake/public/LoadHIP.cmake || die
