Cleanup and add new pytorch packages to cmssw-tools

Author: iarspider

cmssw-tools.spec

@@ -188,7 +188,10 @@ Requires: oracle-fake
 Requires: xtensor
 Requires: xtl
 Requires: xgboost
-Requires: pytorch
+Requires: pytorch-cpu
+# TODO
+#Requires: pytorch-cuda
+#Requires: pytorch-rocm
 Requires: pytorch-custom-ops
 
 ## INCLUDE tfaot-models

pytorch-ignore-different-cuda-include-dir.patch

@@ -5,7 +5,7 @@
     <environment name="INCLUDE" default="$@TOOL_BASE@/include/torchcluster/"/>
     <environment name="LIBDIR" default="$@TOOL_BASE@/lib"/>
   </client>
-  <use name="pytorch"/>
+  <use name="pytorch-cpu"/>
   <lib name="torchcluster"/>
   <flags FORCE_LINK="1"/>
 </tool>

pytorch.spec

@@ -1,5 +1,5 @@
 <tool name="@TOOL@" version="1.0" revision="2">
-    <use name="pytorch"/>
+    <use name="pytorch-cpu"/>
     <use name="pytorch-scatter"/>
     <use name="pytorch-cluster"/>
     <use name="pytorch-sparse"/>

scram-tools.file/tools/pytorch-cluster/pytorch-cluster.xml

@@ -5,7 +5,7 @@
     <environment name="INCLUDE" default="$@TOOL_BASE@/include/torchscatter/"/>
     <environment name="LIBDIR" default="$@TOOL_BASE@/lib"/>
   </client>
-  <use name="pytorch"/>
+  <use name="pytorch-cpu"/>
   <lib name="torchscatter"/>
   <flags FORCE_LINK="1"/>
 </tool>

scram-tools.file/tools/pytorch-custom-ops/pytorch-custom-ops.xml

@@ -5,7 +5,7 @@
     <environment name="INCLUDE" default="$@TOOL_BASE@/include/torchsparse/"/>
     <environment name="LIBDIR" default="$@TOOL_BASE@/lib"/>
   </client>
-  <use name="pytorch"/>
+  <use name="pytorch-cpu"/>
   <lib name="torchsparse"/>
   <flags FORCE_LINK="1"/>
 </tool>