rpms/python-torch
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Cleanup combo of rocm and caffe2

Browse source 
Signed-off-by: Tom Rix <trix@redhat.com>
This commit is contained in:
Tom Rix 2024-02-23 07:33:39 -05:00
commit 2752e0ea1d
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
python-torch.spec
View file

@ -567,9 +567,14 @@ sed -i -f br.sed devel.files
%{python3_sitearch}/torch/lib/libtorch_cuda_linalg.so
%endif
%if %{with caffe2}
%{python3_sitearch}/torch/lib/libcaffe2_detectron_ops.so
%{python3_sitearch}/torch/lib/libcaffe2_observers.so
%{python3_sitearch}/caffe2/python/caffe2_pybind11_state.cpython*.so
%if %{without rocm}
%{python3_sitearch}/torch/lib/libcaffe2_detectron_ops.so
%else
%{python3_sitearch}/torch/lib/libcaffe2_detectron_ops_hip.so
%{python3_sitearch}/caffe2/python/caffe2_pybind11_state_hip.cpython*.so
%endif
%endif
# misc