Skip to content

Evaluation functions acceleration

Sylvain Caillou requested to merge sc_fix_eval_utils into dev

This MR contains:

  • Fixes for residual errors following updates to the new naming scheme.
  • Modifications for events tensors device : Tensors have been moved to the accelerator device from the CPU, resulting in accelerated tensor operations.

It accelerates the function graph_construction_efficiency , graph_construction_efficiency_2D , graph_scoring_efficiency, gnn_efficiency_rz .

gnn_purity_rz was already properly accelerated.

  • Tests on Graph construction evaluation on CTD23 events (with and without target_tracks in the yaml eval configuration file)
  • Tests on Edge Classification evaluation on CTD23 events (with and without target_tracks in the yaml eval configuration file)

Warning: The function graph_scoring_efficiency_purity has not been thoroughly tested due to a lack of example use cases in the evaluation YAML file.

I would like this MR be a part of a 2.0.2 patch in addition with MR !142 (merged) .

Edited by Sylvain Caillou

Merge request reports