From 2752e0ea1d222eb60a7ef6806613837bb7a1001e Mon Sep 17 00:00:00 2001
From: Tom Rix <trix@redhat.com>
Date: Fri, 23 Feb 2024 07:33:39 -0500
Subject: [PATCH] Cleanup combo of rocm and caffe2

Signed-off-by: Tom Rix <trix@redhat.com>
---
 python-torch.spec | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/python-torch.spec b/python-torch.spec
index 626ca0a..d1dbbd4 100644
--- a/python-torch.spec
+++ b/python-torch.spec
@@ -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