diff options
| author |   Alfredo Tupone <tupone@gentoo.org> | 2023-08-11 09:26:35 +0200 |
|---|---|---|
| committer | Alfredo Tupone <tupone@gentoo.org> | 2023-08-11 09:31:10 +0200 |
| commit | 5088a2d38c87926670b7d18f75fc6b91da982caf (patch) | |
| tree | 1156a08a3e9036523aec17ab673e474beeb486cb /sci-libs/caffe2 | |
| parent | dev-python/pyaudio: disable py3.9 (diff) | |
| download | gentoo-5088a2d38c87926670b7d18f75fc6b91da982caf.tar.gz gentoo-5088a2d38c87926670b7d18f75fc6b91da982caf.tar.bz2 gentoo-5088a2d38c87926670b7d18f75fc6b91da982caf.zip | |
sci-libs/caffe2: build nvfuser for cuda users
Signed-off-by: Alfredo Tupone <tupone@gentoo.org>
Diffstat (limited to 'sci-libs/caffe2')
| -rw-r--r-- | sci-libs/caffe2/caffe2-2.0.0-r5.ebuild (renamed from sci-libs/caffe2/caffe2-2.0.0-r4.ebuild) | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/sci-libs/caffe2/caffe2-2.0.0-r4.ebuild b/sci-libs/caffe2/caffe2-2.0.0-r5.ebuild index f6a3684a96a0..eaf3e07d46ac 100644 --- a/sci-libs/caffe2/caffe2-2.0.0-r4.ebuild +++ b/sci-libs/caffe2/caffe2-2.0.0-r5.ebuild @@ -121,7 +121,7 @@ src_configure() { -DUSE_CUDNN=$(usex cuda) -DUSE_FAST_NVCC=$(usex cuda) -DTORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-3.5 7.0}" - -DBUILD_NVFUSER=OFF + -DBUILD_NVFUSER=$(usex cuda) -DUSE_DISTRIBUTED=$(usex distributed) -DUSE_MPI=$(usex mpi) -DUSE_FAKELOWP=OFF @@ -181,6 +181,8 @@ src_configure() { src_install() { cmake_src_install + use cuda && dolib.so "${BUILD_DIR}"/lib/libnvfuser_codegen.so + insinto "/var/lib/${PN}" doins "${BUILD_DIR}"/CMakeCache.txt |