Mila::Dnn::Compute::CudaMatMulBiasGeluOpRegistrar Class Reference

Class responsible for registering the CudaMatMulBiasGeluOp operation. More...

Static Public Member Functions
static void	registerOperations ()
	Registers the CudaMatMulBiasGeluOp operation with the OperationRegistry.

Static Public Attributes
static bool	isRegistered
	Self-registration mechanism that registers the operation during startup.

Detailed Description

Class responsible for registering the CudaMatMulBiasGeluOp operation.

The CudaMatMulBiasGeluOpRegistrar class registers the CudaMatMulBiasGeluOp operation with the OperationRegistry. It associates the operation name "Cuda::MatMulBiasGeluOp" with a factory function that creates instances of CudaMatMulBiasGeluOp.

Member Function Documentation

registerOperations()

static void Mila::Dnn::Compute::CudaMatMulBiasGeluOpRegistrar::registerOperations ( )

inlinestatic

Registers the CudaMatMulBiasGeluOp operation with the OperationRegistry.

This function registers the CudaMatMulBiasGeluOp operation for the CUDA device type with the OperationRegistry. It associates the operation name "Cuda::MatMulBiasGeluOp" with a factory function that creates instances of CudaMatMulBiasGeluOp. It also registers patterns of modules that can be fused using this operation.

Here is the call graph for this function:

Here is the caller graph for this function:

Member Data Documentation

isRegistered

bool Mila::Dnn::Compute::CudaMatMulBiasGeluOpRegistrar::isRegistered

inlinestatic

Initial value:

= []() {
            registerOperations();
            return true;
            }()

Self-registration mechanism that registers the operation during startup.

This static member ensures the operation is registered when the program starts without requiring explicit registration calls.

---

The documentation for this class was generated from the following file:

• /home/runner/work/Mila/Mila/Mila/Src/Dnn/Compute/Operations/Cuda/MatMulBiasActivation.ixx