def get_trt_llm_mlperf_gpu_config()

in dags/inference/configs/trt_llm_mlperf_v41_config.py [0:0]

• 17 lines of code
• 1 McCabe index (conditional complexity)

def get_trt_llm_mlperf_gpu_config(
    machine_type: vm_resource.MachineVersion,
    image_project: vm_resource.ImageProject,
    image_family: vm_resource.ImageFamily,
    accelerator_type: vm_resource.GpuVersion,
    count: int,
    gpu_zone: vm_resource.Zone,
    time_out_in_min: int,
    test_name: str,
    project: Project,
    network: str,
    subnetwork: str,
    existing_instance_name: str = None,
    benchmark_configs: Dict = {},
    model_parameters: Dict = {},
    parameter_positions: Dict = {},
    binary_search_steps: int = 1,