diff options
Diffstat (limited to 'sci-libs/caffe2/caffe2-2.1.1.ebuild')
| -rw-r--r-- | sci-libs/caffe2/caffe2-2.1.1.ebuild | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/sci-libs/caffe2/caffe2-2.1.1.ebuild b/sci-libs/caffe2/caffe2-2.1.1.ebuild index e5e9a71069cd..65c5c2831826 100644 --- a/sci-libs/caffe2/caffe2-2.1.1.ebuild +++ b/sci-libs/caffe2/caffe2-2.1.1.ebuild @@ -135,7 +135,6 @@ src_configure() { -DUSE_CCACHE=OFF -DUSE_CUDA=$(usex cuda) -DUSE_CUDNN=$(usex cuda) - -DUSE_FAST_NVCC=$(usex cuda) -DTORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-3.5 7.0}" -DBUILD_NVFUSER=$(usex cuda) -DUSE_DISTRIBUTED=$(usex distributed) @@ -204,7 +203,12 @@ src_install() { mkdir -p python/torch/include || die mv "${ED}"/usr/lib/python*/site-packages/caffe2 python/ || die mv "${ED}"/usr/include/torch python/torch/include || die + mv "${ED}${S}"/nvfuser python/nvfuser || die + rm -r "${ED}${S}"/test || die + rm -r "${ED}${S}"/third_party || die cp torch/version.py python/torch/ || die python_domodule python/caffe2 python_domodule python/torch + python_domodule python/nvfuser + find "${ED}" -empty -delete } |