Skip to content

Conversation

@Shlok-Saxena
Copy link

This PR adds support for the GELU activation function in TMVA::Experimental::SOFIE.

The implementation follows the ONNX-specified tanh-based approximation of GELU and
integrates with the existing SOFIE operator and ONNX parser infrastructure.

Changes include:

  • New SOFIE operator ROperator_Gelu
  • ONNX parser support via ParseGelu
  • Operator registration in the ONNX dispatcher

The implementation mirrors the existing Relu operator pattern and was validated by
building ROOT with TMVA and SOFIE enabled.

@Shlok-Saxena Shlok-Saxena changed the title TMVA(SOFIE): add GELU activation operator [tmva][sofie] Add GELU activation operator Dec 31, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants