o ip @s|ddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z dee e ffdd Zd d Zed kr#szprint_info..zgpu.compute_capabilityzgpu.namez not availabler dcgm_profilerz build.infozbuild.cuda_versionzbuild.hip_versionzbuild.python_versionzbuild.torch_versionz build.env.nonecsrr r!)r"rrrrr$=s zbuild.nvcc_versionz open sourcezsource.privacy fairinternalz {:<50} {}:)rprintrtorchcudarcurrent_deviceget_device_capabilityjoinget_device_namer r_build_metadata isinstance_cpp_library_load_exceptionxFormersInvalidLibException build_info cuda_version hip_versionpython_version torch_version build_envrrxformers_nvcc_build_version RuntimeErrorAttributeErrorrformat)rdevicecapr4rrnamestatusrrr print_infosP             rC__main__)typingrr*rrrrr ops.commonr profiler.profiler_dcgmr rrrC__name__rrrrs    1