def METHOD_NAME(self, *torch_inputs: List[torch.Tensor]) -> List[torch.Tensor]:
    if self.rt_module is None:
        if torch_inputs[0].is_cuda:
            self.build(target="cuda")
        elif torch_inputs[0].device.type == "cpu":
            self.build()
        else:
            raise Exception(f"the target {torch_inputs[0].device.type} is not supported yet")
    return self.rt_module.METHOD_NAME(torch_inputs)