o e@sFddlZddlmZddZddZejjeddd d d d ZdS) N)nvrtccCs8t|tjr|tjjkrtd|dStd|)NzNvrtc Error: {}zUnknown error type: {}) isinstancerZ nvrtcResultZ NVRTC_SUCCESS RuntimeErrorformat)errrED:\Projects\ConvertPro\env\Lib\site-packages\cuda/tests/test_nvrtc.py ASSERT_DRV s  r cCs.t\}}}t|||kp||ko||kS)N)rZ nvrtcVersionr )majorminorr major_version minor_versionrrrnvrtcVersionLessThansr z*When nvrtcGetSupportedArchs was introduced)reasoncCs(t\}}t|t|dksJdS)Nr)rZnvrtcGetSupportedArchsr len)rZsupportedArchsrrrtest_nvrtcGetSupportedArchss r)Zpytestcudarr rmarkZskipifrrrrrs