def METHOD_NAME(mod):
    """Offload ops based on the registered ops
    Paramters
    ---------
    mod : Module
        The input module.
    Returns
    -------
    mod : Module
        The output module with offloaded ops.
    """
    backend = "verilator"
    mod = transform.AnnotateTarget([backend])(mod)
    mod = transform.PartitionGraph()(mod)
    return mod