Example script to run workflow with ML clusterizer

Author: sawenzel

MC/run/ANCHOR/tests/ML_clusterizer/run.sh

@@ -0,0 +1,68 @@
+#!/bin/bash
+
+#JDL_OUTPUT=*.txt@disk=1,AO2D*.root@disk=2,*.log@disk=1,*stat*@disk=1,*.json@disk=1,debug*tgz@disk=2,tf*/coll*.root
+#JDL_ERROROUTPUT=*.txt@disk=1,AO2D*.root@disk=2,*.log@disk=1,*.json@disk=1,debug*tgz@disk=2
+#JDL_PACKAGE=O2PDPSuite::daily-20251215-0000-1
+
+#
+# An **EXAMPLE** showing injection of GPU_proc_nn config_keys into the workflow creation, so
+# that TPC clusterizer runs with ML kernel.
+#
+
+# example anchoring
+export ALIEN_JDL_LPMANCHORPASSNAME=apass4
+export ALIEN_JDL_MCANCHOR=apass4
+export ALIEN_JDL_CPULIMIT=8
+export ALIEN_JDL_LPMRUNNUMBER=544124
+export ALIEN_JDL_LPMPRODUCTIONTYPE=MC
+export ALIEN_JDL_LPMINTERACTIONTYPE=PbPb
+export ALIEN_JDL_LPMPRODUCTIONTAG=MLClusterTest
+export ALIEN_JDL_LPMANCHORRUN=544124
+export ALIEN_JDL_LPMANCHORPRODUCTION=LHC23zz
+export ALIEN_JDL_LPMANCHORYEAR=2023
+
+export NTIMEFRAMES=1
+export PRODSPLIT=${ALIEN_O2DPG_GRIDSUBMIT_PRODSPLIT:-100}
+export SPLITID=${ALIEN_O2DPG_GRIDSUBMIT_SUBJOBID:-50}
+export CYCLE=0
+
+# this file modifies few config key values for GPU_proc_nn for application in TPC clusterization
+LOCAL_CONFIG="customize_ml_clusterizing.json"
+cat > ${LOCAL_CONFIG} <<EOF
+{
+  "ConfigParams": {
+    "GPU_proc_nn" : {
+      "nnInferenceDevice" : "CPU",
+      "nnInferenceAllocateDevMem" : "0",
+      "nnInferenceInputDType" : "FP16",
+      "nnInferenceOutputDType" : "FP16",
+      "nnSigmoidTrafoClassThreshold" : "1",
+      "nnClusterizerAddIndexData" : "1",
+      "nnInferenceIntraOpNumThreads" : "1",
+      "nnInferenceInterOpNumThreads" : "1",
+      "nnInferenceVerbosity" : "3",
+      "nnClusterizerVerbosity" : "2",
+      "nnClusterizerApplyCfDeconvolution" : "0",
+      "nnLoadFromCCDB" : "1",
+      "nnClusterizerBoundaryFillValue" : "0",
+      "nnInferenceOrtProfiling" : "0",
+      "nnClusterizerSetDeconvolutionFlags" : "1",
+      "nnClusterizerSizeInputRow" : "1",
+      "nnClusterizerSizeInputPad" : "4",
+      "nnClusterizerSizeInputTime" : "4",
+      "nnClassThreshold" : "0.05",
+      "nnClusterizerUseCfRegression" : "0",
+      "nnClusterizerBatchedMode" : "262144",
+      "applyNNclusterizer" : "1"
+    }
+  }
+  "Executables": {   
+  }
+}
+EOF
+
+# we use pp event gen just to get faster results
+export ALIEN_JDL_ANCHOR_SIM_OPTIONS="-gen pythia8pp -confKey \"GeometryManagerParam.useParallelWorld=1;GeometryManagerParam.usePwGeoBVH=1;GeometryManagerParam.usePwCaching=1\" ${LOCAL_CONFIG:+--overwrite-config ${LOCAL_CONFIG}}"
+
+export ALIEN_JDL_O2DPGWORKFLOWTARGET="tpcclus"
+${O2DPG_ROOT}/MC/run/ANCHOR/anchorMC.sh