64#h'ddlZddlZddlmZddlZdZhdZdZejs dZ ejjdZ e jD]Zej j#edZej j'es;ejj)d eZejj-eZej0j3eej4Z ee dd k\r ee_ ejrCej8s6ej:j=d dZed Zned z Zeej:d <yyy#YTxYw)N)argsctjdk(r?ddlGfddjjj fd}|St }tjddg}|jdD]C}t|dkDs|j|jd jd dE|S) NntrceZdZdWjfdWjdzfdWjdzfdWjfdWjdzfdWjdzfgZy ) &get_gpu_names..DISPLAY_DEVICEAcb DeviceName DeviceString StateFlagsDeviceID DeviceKeyN)__name__ __module__ __qualname__c_ulongc_char_fields_)ctypess2D:\ComfyUI_windows_portable\ComfyUI\cuda_malloc.pyDISPLAY_DEVICEAr sfv~~&v}}r12!45v~~.V]]S01fmmc12 HrcF}j||_d}t}jd|j |drS|dz }|j |j jdjd|j |drS|S)Nrutf-8)sizeofrsetEnumDisplayDevicesAbyrefaddr decode) device_info device_index gpu_namesrruser32s renum_display_devicesz+get_gpu_names..enum_display_devicess)+K#]];7KNLI,,T<kAZ\]^!  k66==gFG,,T<kAZ\]^ rz nvidia-smiz-L rz (UUID) osnamer Structurewindllr&r subprocess check_outputsplitlenr!r")r'r%outlrrr&s @@@r get_gpu_namesr3s ww$ f.. %% $%%E %%|T&:;5!A1vz ahhw/55h?BC"r>)Tesla M4Tesla M6 Tesla M10 Tesla M40 Tesla M60 Quadro K620 Quadro M500 Quadro M520 Quadro M600 Quadro M620 GeForce 830M GeForce 840M GeForce 910M GeForce 920M GeForce 930M GeForce 940M GeForce 945M Quadro K1200 Quadro K2200 Quadro M1000 Quadro M1200 Quadro M2000 Quadro M2200 Quadro M3000 Quadro M4000 Quadro M5000 Quadro M5500 Quadro M6000 GeForce MX110 GeForce MX130GeForce GTX 745GeForce GTX 750GeForce GTX 950GeForce GTX 960GeForce GTX 970GeForce GTX 980GeForce GTX 1630GeForce GTX 1650GeForce GTX 850MGeForce GTX 860MGeForce GTX TITAN Xcx t}|D]}d|vstD] }||vsyy#t}Y-xYw)NNVIDIAFT)r3r blacklist)namesxbs rcuda_malloc_supportedrc5sH q=6  s + 9torchz version.pytorch_version_importPYTORCH_CUDA_ALLOC_CONFzbackend:cudaMallocAsyncz,backend:cudaMallocAsync) r)importlib.util importlibcomfy.cli_argsrr-r3r_rc cuda_mallocversionutil find_spec torch_specsubmodule_search_locationsfolderpathjoinver_fileisfilespec_from_file_locationspecmodule_from_specmoduleloader exec_module __version__intdisable_cuda_mallocenvirongetenv_varrrrsD $L      ^^--g6  ;;Fww||FL9Hww~~h' ~~==>TV^_"88> ''/ ,, < wqz?a 46D  D44jjnn6=G+--,3BJJ()5   sA#EA:EE