o i(@sdZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ej dkr2dZd Znej d krs r$cCstj|s t|ddS)Nz not found)osr#isfileFileNotFoundError)r#r r rcheck_static_libCs r(cCs@|dkr tdjS|dkrtdjS|rdnd}t|jS)Nrr rr)rZby)rrrr r r_get_source_variableHs     r)c Cs d}z6t\}}td|D] }td|qtd|tdddt||\}}tdtd |WntyS}ztd |d }WYd }~nd }~wwtjd kr|st}tj tjj d|d}zt | } | } Wd n1swYWnt ytd|dYnwtdd| D} td| D] }td|qd} | D]A} t| }td| t| td|ztdddt| tdWqt y}ztd| |fd }WYd }~qd }~wwd} t| d d}td| t| d dtd|ztdddt|tdWnty@}ztd| |fd }WYd }~nd }~wwtd}td |t}td|ztdddt|tdW| Sty}ztd| |fd }WYd }~| Sd }~ww)!z:Test library lookup. Path info is printed to stdout. FzFinding driver from candidates: z Using loader z Trying to load driverz...)endz okz Loaded from z ERROR: failed to open driver: TNlinuxprocmapsz ERROR: Could not open z) to determine absolute path to libcuda.socss|] }d|vr|VqdS)z libcuda.soNr ).0sr r r ~sztest..z Mapped libcuda.so paths:z znvvm nvrtc cudartzFinding {} from {}z Located atz Trying to open libraryz ERROR: failed to open %s: %sZ cudadevrtrz Checking libraryz ERROR: failed to find %s: %sr zFinding libdevice from )rprintrrsysplatformr%getpidr#joinseprrOSErrorsetsplitr formatr)r$r(r'r)failedZdlloaderrlocationZdllr#epidZmapsfilefr. locationsZlibsrwherer r rtestRs                  rC)F)__doc__r%r3r!Znumba.misc.findlibrZnumba.cuda.cuda_pathsrZnumba.cuda.cudadrv.driverrrZnumba.cuda.cudadrv.errorrr4rrrrr r$r(r)rCr r r rs.