GPU TPC NN Clusterizer: Fix compilation without ONNX

davidrohr · davidrohr · commit 23781677b66c · 2025-05-21T08:45:23.000+02:00
diff --git a/GPU/GPUTracking/Global/GPUChainTrackingClusterizer.cxx b/GPU/GPUTracking/Global/GPUChainTrackingClusterizer.cxx
@@ -1198,13 +1198,15 @@ int32_t GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)
     }
   }
   for (int32_t i = 0; i < GetProcessingSettings().nTPCClustererLanes; i++) {
+#ifdef GPUCA_HAS_ONNX
     if (GetProcessingSettings().nn.applyNNclusterizer) {
       LOG(info) << "(ORT) Environment releasing...";
       GPUTPCNNClusterizerHost& nnApplication = nnApplications[i];
       nnApplication.mModelClass.release(true);
       nnApplication.mModelReg1.release(true);
       nnApplication.mModelReg2.release(true);
     }
+#endif
     if (transferRunning[i]) {
       ReleaseEvent(mEvents->stream[i], doGPU);
     }

Original file line number	Diff line number	Diff line change
`@@ -1198,13 +1198,15 @@ int32_t GPUChainTracking::RunTPCClusterizer(bool synchronizeOutput)`
`1198`	`1198`	`}`
`1199`	`1199`	`}`
`1200`	`1200`	`for (int32_t i = 0; i < GetProcessingSettings().nTPCClustererLanes; i++) {`
	`1201`	`+#ifdef GPUCA_HAS_ONNX`
`1201`	`1202`	`if (GetProcessingSettings().nn.applyNNclusterizer) {`
`1202`	`1203`	`LOG(info) << "(ORT) Environment releasing...";`
`1203`	`1204`	`GPUTPCNNClusterizerHost& nnApplication = nnApplications[i];`
`1204`	`1205`	`nnApplication.mModelClass.release(true);`
`1205`	`1206`	`nnApplication.mModelReg1.release(true);`
`1206`	`1207`	`nnApplication.mModelReg2.release(true);`
`1207`	`1208`	`}`
	`1209`	`+#endif`
`1208`	`1210`	`if (transferRunning[i]) {`
`1209`	`1211`	`ReleaseEvent(mEvents->stream[i], doGPU);`
`1210`	`1212`	`}`