B \ @sddlmZddlZddlZddlZddlZddlZddlmZm Z ejdkrVej Z nejdkrhej Z nej Z ddZ dd Zdd d Zdd dZdddZddZdS))print_functionN)find_lib find_fileZwin32darwincCsXtjdptjd}d|}tt||}|sHd}tt||}|rTt|SdS)NZNUMBAPRO_LIBDEVICENUMBAPRO_CUDALIBzlibdevice\.%s(\.\d+)*\.bc$zlibdevice(\.\d+)*\.bc$)osenvirongetrrecompilemax)archlibdirZpat candidatesr6lib/python3.7/site-packages/numba/cuda/cudadrv/libs.py get_libdevices  rc Cs"tt|d }|SQRXdS)Nrb)openrread)r Zbcfilerrropen_libdevice srcCsH|dkr tjdr tjdStjd}t|||}|rDt|SdS)NZnvvmZ NUMBAPRO_NVVMr)rrr rr )libplatformrrrrr get_cudalib%s    rFcCs2t|}|dkrtd||r*t|St|S)Nzlibrary %s not found)rOSErrorctypesCDLL _dllopener)rcccpathrrr open_cudalib-s   r Tc Cs(d}d}x|D]}t||}td||rR|r>td|q^tdtj|n tdd}|s|dtjfkry$tdd d t|dd td Wqt k r}z"td ||ft sdnd}Wdd}~XYqXqWd}xB|D]:}td|d d t |}|rtd qtd|d}qW| S)NFz!cublas cusparse cufft curand nvvmZFindingz located atz named z ERROR: can't locate libTz trying to open libraryz...)end)rz okz ERROR: failed to open %s: %s)Z compute_20Z compute_30Z compute_35Z compute_50z finding libdevice forz# ERROR: can't open libdevice for %s) splitrprintrrbasenamesysrr r _if_osx_10_5r) Z _platformZ print_pathsZfailedZlibsrreZarchsr rrrtest6s8       $   r(cCs6tjdkr2ttttdd}|dkr2dSdS)Nrr.) TF)r%rtuplemapintZmac_verr")Zversrrrr&[s  r&)N)F)NT)Z __future__rr rr%rrZ numba.findlibrrZWinDLLrrrrrr r(r&rrrrs"     %