o )ip @s|ddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z dee e ffdd Zd d Zed kr#szprint_info..zgpu.compute_capabilityzgpu.namez not availabler Z dcgm_profilerz build.infozbuild.cuda_versionzbuild.hip_versionzbuild.python_versionzbuild.torch_versionz build.env.nonecsrrr)rrrrrr =s zbuild.nvcc_versionz open sourcezsource.privacyZ fairinternalz {:<50} {}:)rprintrtorchcudarZcurrent_deviceZget_device_capabilityjoinZget_device_namer rZ_build_metadata isinstanceZ_cpp_library_load_exceptionZxFormersInvalidLibException build_infoZ cuda_versionZ hip_versionpython_versionZ torch_version build_envrrZxformersZ_nvcc_build_version RuntimeErrorAttributeErrorrformat)rZdevicecapr(rrnamestatusrrr print_infosP             r1__main__)typingrr$rrrrrZ ops.commonr Zprofiler.profiler_dcgmr rrr1__name__rrrrs    1