o Td@sdZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ej dkr2dZd Znej d krs r'cCs*t|dd}tj|st|ddS)NTrz not found)r"osr&isfileFileNotFoundErrorr%r r rcheck_static_libCs  r,cCs@|dkr tdjS|dkrtdjS|rdnd}t|jS)Nrr rr)rby)rrrr r r_get_source_variableIs     r.Tc Cs4d}z/t\}}d|}td|dtd|tdddt||\}}td|WntyL}ztd |d }WYd }~nd }~wwd } | D]V} t| |}td | t| |rltd|n tdt j ||d t j fvrztdddt| tdWqSty}ztd| |fd }WYd }~qSd }~wwqSd} t| |d d}td | t| d d|rtd|n tdt j |zt| Wnty}ztd| |fd }WYd }~nd }~wwtd} td| tdddt}|rtd| Stdd }| S)z:Test library lookup. Path info is printed to stdout. Fz, z Finding driver from candidates: z...z Using loader z trying to load driver)endz ok, loaded from z ERROR: failed to open driver: TNz nvvm cudartzFinding {} from {}z located atz named z trying to open libraryz okz ERROR: failed to open %s: %s cudadevrtr(z ERROR: failed to find %s: %sr zFinding libdevice from z ERROR: can't open libdevice)rjoinprintrrsplitr"formatr.r)r&basenamesysrr'OSErrorr,r+r) _platform print_pathsfaileddlloaderr locationsdllr&elibsrwherer r rtestSsr             rA)NF)F)NT)__doc__r)r6r#numba.misc.findlibrnumba.cuda.cuda_pathsrnumba.cuda.cudadrv.driverrrnumba.cuda.cudadrv.errorrrrrrrr"r'r,r.rAr r r rs.