Class GraphDeployCommand

ppq/IR/base/command.py:125–136

class GraphDeployCommand(GraphCommand):
    def __init__(self, device: str) -> None:
        if device.startswith('cuda'):
            super().__init__(GraphCommandType.DEPLOY_TO_CUDA)
        elif device.startswith('cpu'):
            super().__init__(GraphCommandType.DEPLOY_TO_CPU)
        else:
            raise ValueError(f'Device type {device} not understand.')
        self._device = device

    def __str__(self) -> str:
        return super().__str__()


class QuantizeOperationCommand(GraphCommand):