Skip to content

Commit

Permalink
fix #2
Browse files Browse the repository at this point in the history
  • Loading branch information
PIPIPIG233666 committed Jul 7, 2023
1 parent 2e60c4b commit da73c77
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion sci-libs/caffe2/caffe2-2.0.1-r2.ebuild
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

EAPI=8

PYTHON_COMPAT=( python3_{9..11} )
PYTHON_COMPAT=( python3_{10..12} )
ROCM_VERSION=5.6.0
inherit python-single-r1 cmake flag-o-matic rocm

Expand Down Expand Up @@ -200,6 +200,7 @@ src_configure() {
-DLIBSHM_INSTALL_LIB_SUBDIR="${EPREFIX}"/usr/$(get_libdir)
)
if use rocm; then
export HIP_CLANG_PATH="/usr/lib/llvm/16/bin"
mycmakeargs+=(
-DPYTORCH_ROCM_ARCH="$(get_amdgpu_flags)"
-DROCM_VERSION_DEV_RAW=${ROCM_VERSION}
Expand All @@ -208,6 +209,7 @@ src_configure() {
fi

use cuda && addpredict "/dev/nvidiactl" # bug 867706

use rocm && export HIP_PATH="${EPREFIX}/usr"
cmake_src_configure
}
Expand Down

0 comments on commit da73c77

Please sign in to comment.