PyTorch custom operators for graph operations

cmssw-tool-conf.spec

@@ -182,6 +182,7 @@ Requires: xtensor
 Requires: xtl
 Requires: xgboost
 Requires: pytorch
+Requires: pytorch-custom-ops
 
 ## INCLUDE cmssw-vectorization
 ## INCLUDE cmssw-drop-tools

pytorch-cluster.spec

@@ -0,0 +1,43 @@
+### RPM external pytorch-cluster 1.6.3
+## INCLUDE compilation_flags
+## INCLUDE cpp-standard
+%define tag f2d99195a0003ca2d2ba9ed50d0117e2f23360e0
+%define branch master
+%define github_user rusty1s
+
+Source: git+https://github.com/%{github_user}/pytorch_cluster.git?obj=%{branch}/%{tag}&export=%{n}-%{realversion}&submodules=1&output=/%{n}-%{realversion}.tgz
+
+BuildRequires: cmake
+Requires: pytorch cudnn
+%define build_flags -Wall -Wextra -pedantic %{?arch_build_flags} 
+
+%prep
+%setup -n %{n}-%{realversion}
+# Make sure the default c++sdt stand is c++14
+grep -q 'CMAKE_CXX_STANDARD  *14' CMakeLists.txt
+sed -i -e 's|CMAKE_CXX_STANDARD  *14|CMAKE_CXX_STANDARD %{cms_cxx_standard}|' CMakeLists.txt
+
+%build
+
+rm -rf ../build && mkdir ../build && cd ../build
+
+
+cmake ../%{n}-%{realversion} \
+    -DCMAKE_BUILD_TYPE=Release \
+    -DCMAKE_INSTALL_PREFIX=%{i} \
+    -DCMAKE_INSTALL_LIBDIR=lib \
+    -DCMAKE_PREFIX_PATH="%{cmake_prefix_path}" \
+    -DCMAKE_CXX_STANDARD=%{cms_cxx_standard} \
+    -DCMAKE_CXX_FLAGS="%{build_flags}" \
+    -DBUILD_TEST=OFF \
+    -DWITH_PYTHON=OFF \
+    -DWITH_CUDA=OFF \
+    -DBUILD_TEST=OFF \
+    -DBUILD_SHARED_LIBS=ON 
+
+
+make %{makeprocesses} VERBOSE=1
+
+%install
+cd ../build
+make %{makeprocesses} install VERBOSE=1

pytorch-custom-ops.spec

@@ -0,0 +1,8 @@
+### RPM external pytorch-custom-ops 1.0
+Source: none
+Requires: pytorch-scatter
+Requires: pytorch-cluster
+Requires: pytorch-sparse
+%prep
+%build
+%install

pytorch-scatter.spec

@@ -0,0 +1,43 @@
+### RPM external pytorch-scatter 2.1.2
+## INCLUDE compilation_flags
+## INCLUDE cpp-standard
+%define tag c095c62e4334fcd05e4ac3c4bb09d285960d6be6
+%define branch master
+%define github_user rusty1s
+
+Source: git+https://github.com/%{github_user}/pytorch_scatter.git?obj=%{branch}/%{tag}&export=%{n}-%{realversion}&submodules=1&output=/%{n}-%{realversion}.tgz
+
+BuildRequires: cmake
+Requires: pytorch cudnn
+%define build_flags -Wall -Wextra -pedantic %{?arch_build_flags} 
+
+%prep
+%setup -n %{n}-%{realversion}
+# Make sure the default c++sdt stand is c++14
+grep -q 'CMAKE_CXX_STANDARD  *14' CMakeLists.txt
+sed -i -e 's|CMAKE_CXX_STANDARD  *14|CMAKE_CXX_STANDARD %{cms_cxx_standard}|' CMakeLists.txt
+
+%build
+
+rm -rf ../build && mkdir ../build && cd ../build
+
+
+cmake ../%{n}-%{realversion} \
+    -DCMAKE_BUILD_TYPE=Release \
+    -DCMAKE_INSTALL_PREFIX=%{i} \
+    -DCMAKE_INSTALL_LIBDIR=lib \
+    -DCMAKE_PREFIX_PATH="%{cmake_prefix_path}" \
+    -DCMAKE_CXX_STANDARD=%{cms_cxx_standard} \
+    -DCMAKE_CXX_FLAGS="%{build_flags}" \
+    -DBUILD_TEST=OFF \
+    -DWITH_PYTHON=OFF \
+    -DWITH_CUDA=OFF \
+    -DBUILD_TEST=OFF \
+    -DBUILD_SHARED_LIBS=ON 
+
+
+make %{makeprocesses} VERBOSE=1
+
+%install
+cd ../build
+make %{makeprocesses} install VERBOSE=1

pytorch-sparse.spec

@@ -0,0 +1,43 @@
+### RPM external pytorch-sparse 0.6.18
+## INCLUDE compilation_flags
+## INCLUDE cpp-standard
+%define tag 2d559810c6af7f8b2cf88553dd5a5824a667a07d
+%define branch master
+%define github_user rusty1s
+
+Source: git+https://github.com/%{github_user}/pytorch_sparse.git?obj=%{branch}/%{tag}&export=%{n}-%{realversion}&submodules=1&output=/%{n}-%{realversion}.tgz
+
+BuildRequires: cmake
+Requires: pytorch cudnn
+%define build_flags -Wall -Wextra -pedantic %{?arch_build_flags} 
+
+%prep
+%setup -n %{n}-%{realversion}
+# Make sure the default c++sdt stand is c++14
+grep -q 'CMAKE_CXX_STANDARD  *14' CMakeLists.txt
+sed -i -e 's|CMAKE_CXX_STANDARD  *14|CMAKE_CXX_STANDARD %{cms_cxx_standard}|' CMakeLists.txt
+
+%build
+
+rm -rf ../build && mkdir ../build && cd ../build
+
+
+cmake ../%{n}-%{realversion} \
+    -DCMAKE_BUILD_TYPE=Release \
+    -DCMAKE_INSTALL_PREFIX=%{i} \
+    -DCMAKE_INSTALL_LIBDIR=lib \
+    -DCMAKE_PREFIX_PATH="%{cmake_prefix_path}" \
+    -DCMAKE_CXX_STANDARD=%{cms_cxx_standard} \
+    -DCMAKE_CXX_FLAGS="%{build_flags}" \
+    -DBUILD_TEST=OFF \
+    -DWITH_PYTHON=OFF \
+    -DWITH_CUDA=OFF \
+    -DBUILD_TEST=OFF \
+    -DBUILD_SHARED_LIBS=ON 
+
+
+make %{makeprocesses} VERBOSE=1
+
+%install
+cd ../build
+make %{makeprocesses} install VERBOSE=1

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

@@ -0,0 +1,10 @@
+<tool name="pytorch-cluster" version="@TOOL_VERSION@">
+  <client>
+    <environment name="PYTORCH_CLUSTER_BASE" default="@TOOL_ROOT@"/>
+    <environment name="INCLUDE" default="$PYTORCH_CLUSTER_BASE/include"/>
+    <environment name="INCLUDE" default="$PYTORCH_CLUSTER_BASE/include/torchcluster/"/>
+    <environment name="LIBDIR" default="$PYTORCH_CLUSTER_BASE/lib"/>
+  </client>
+  <use name="pytorch"/>
+  <lib name="torchcluster"/>
+</tool>

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

@@ -0,0 +1,6 @@
+<tool name="pytorch-custom-ops" version="1.0">
+    <use name="pytorch"/>
+    <use name="pytorch-scatter"/>
+    <use name="pytorch-cluster"/>
+    <use name="pytorch-sparse"/>
+</tool>

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

@@ -0,0 +1,10 @@
+<tool name="pytorch-scatter" version="@TOOL_VERSION@">
+  <client>
+    <environment name="PYTORCH_SCATTER_BASE" default="@TOOL_ROOT@"/>
+    <environment name="INCLUDE" default="$PYTORCH_SCATTER_BASE/include"/>
+    <environment name="INCLUDE" default="$PYTORCH_SCATTER_BASE/include/torchscatter/"/>
+    <environment name="LIBDIR" default="$PYTORCH_SCATTER_BASE/lib"/>
+  </client>
+  <use name="pytorch"/>
+  <lib name="torchscatter"/>
+</tool>

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

@@ -0,0 +1,10 @@
+<tool name="pytorch-sparse" version="@TOOL_VERSION@">
+  <client>
+    <environment name="PYTORCH_SPARSE_BASE" default="@TOOL_ROOT@"/>
+    <environment name="INCLUDE" default="$PYTORCH_SPARSE_BASE/include"/>
+    <environment name="INCLUDE" default="$PYTORCH_SPARSE_BASE/include/torchsparse/"/>
+    <environment name="LIBDIR" default="$PYTORCH_SPARSE_BASE/lib"/>
+  </client>
+  <use name="pytorch"/>
+  <lib name="torchsparse"/>
+</tool>