Merge pull request #9061 from cms-sw/enable-vector-lwtnn-opencv

Enable vectorization of lwtnn, opencv and pytorch
cms-sw · Mar 12, 2024 · 6591182 · 6591182
2 parents 1f90040 + d73ee3d
commit 6591182
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/cmssw-vectorization.file b/cmssw-vectorization.file
@@ -1,2 +1,2 @@
-%define vectorized_packages zlib fastjet tensorflow-sources tensorflow OpenBLAS rivet gbl
+%define vectorized_packages zlib fastjet tensorflow-sources tensorflow OpenBLAS rivet gbl lwtnn opencv pytorch
 %{expand:%(for t in %{vectorized_packages} ; do echo Requires: $t; for v in %{package_vectorization}; do echo Requires: ${t}_${v}; done; done)}
diff --git a/scram-tools.file/tools/lwtnn/vectorized.tmpl b/scram-tools.file/tools/lwtnn/vectorized.tmpl
@@ -0,0 +1,5 @@
+<tool name="lwtnn_@TOOL_VECTORIZATION@" version="@TOOL_VERSION@">
+  <client>
+    <environment name="@TOOL_VECTORIZATION_KEY@_LIBDIR" default="@TOOL_ROOT@/lib"/>
+  </client>
+</tool>
diff --git a/scram-tools.file/tools/opencv/vectorized.tmpl b/scram-tools.file/tools/opencv/vectorized.tmpl
@@ -0,0 +1,5 @@
+<tool name="opencv_@TOOL_VECTORIZATION@" version="@TOOL_VERSION@">
+  <client>
+    <environment name="@TOOL_VECTORIZATION_KEY@_LIBDIR" default="@TOOL_ROOT@/lib"/>
+  </client>
+</tool>
diff --git a/scram-tools.file/tools/pytorch/vectorized.tmpl b/scram-tools.file/tools/pytorch/vectorized.tmpl
@@ -0,0 +1,5 @@
+<tool name="pytorch_@TOOL_VECTORIZATION@" version="@TOOL_VERSION@">
+  <client>
+    <environment name="@TOOL_VECTORIZATION_KEY@_LIBDIR" default="@TOOL_ROOT@/lib"/>
+  </client>
+</tool>