Skip to content

Define custom TensorRT plugin

Anthony Correia requested to merge anthonyc/custom_scatter_for_trt into main

This MR define a custom TensorRT scatter_add operator that can be activated for the GNN when the use_trt_scater ONNX option is used.

This MR must be merge AFTER !33 (merged). It is actually rebased w.r.t. to this MR.

Edited by Anthony Correia

Merge request reports

Loading