Function register_operation_handler

ppq/executor/base.py:25–41

Regitser a custimized function as operation handler. Function should accept at least 3 input parameters, return one or more tensor as result: func(op: Operation, values: List[torch.Tensor], ctx: TorchBackendContext = None, **kwargs) -> torch.Tensor: If there is already another

(handler: Callable, operation_type: str, platform: TargetPlatform)

def register_operation_handler(handler: Callable, operation_type: str, platform: TargetPlatform):
    """Regitser a custimized function as operation handler.
    
    Function should accept at least 3 input parameters, return one or more tensor as result:
    func(op: Operation, values: List[torch.Tensor], ctx: TorchBackendContext = None, **kwargs) -> torch.Tensor:
    
    If there is already another operation handler for given operation_type,
        new handler will replace the old one without warrning.

    Args:
        handler (Callable): Callable function, which interface follows restriction above.
        operation_type (str): Operation type.
        platform (TargetPlatform): Register platform.
    """
    if platform not in OPERATION_FORWARD_TABLE:
        raise ValueError('Unknown Platform detected, Please check your platform setting.')
    OPERATION_FORWARD_TABLE[platform][operation_type] = handler


class RuntimeHook(metaclass=ABCMeta):