def METHOD_NAME():
    """Helper function for invoking Torch-TRT"""
    import torch_tensorrt
    module, example_inputs = model.get_module()
    torch_dtype_precision = torch.half if FP16 else torch.float32
    print(
        f"Compiling {model.name} with batch size {model.batch_size}, precision {model.dargs.precision}, "
        + f"and {'default' if 'ir' not in torch_trt_kwargs else torch_trt_kwargs['ir']} IR"
    )
    trt_module = torch_tensorrt.compile(
        module,
        inputs=example_inputs,
        enabled_precisions={torch_dtype_precision},
        **torch_trt_kwargs,
    )
    model.set_module(trt_module)