U hd< @sdZddlmZddlmZmZmZddlmZddl m Z ddl m Z dd l mZmZmZd#d d Zd d ZGdddeZddZddZddZddZddZGdddeZeZd$dd Zed!d"ZdS)%z Contains CUDA API functions )contextmanager)require_contextresetgpusFakeCUDAKernel) is_signaturewarn)InOutInOutcCs|dkstddS)Nrz/Only a single device supported by the simulator)AssertionError)devrQ/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/cuda/simulator/api.py select_devicesrcCsdSNTrrrrris_float16_supportedsrc@s$eZdZdZeddZddZdS)streamz The stream API is supported in the simulator - however, all execution occurs synchronously, so synchronization requires no operation. ccs dVdSNrselfrrrauto_synchronizeszstream.auto_synchronizecCsdSrrrrrr synchronize!szstream.synchronizeN)__name__ __module__ __qualname____doc__rrrrrrrrs rcCsdSrrrrrrr%srcCs dt_dSr)rclosedrrrrclose)sr"cOsdSrr)argskwargsrrrdeclare_device-sr%cCs$tdtddtdddS)NzFound 1 CUDA deviceszid %d %20s %40s)rZ SIMULATORz [SUPPORTED]z %40s: 5.0zcompute capability)printrrrrdetect1s r'cCstSr)rrrrr list_devices7sr(c@s4eZdZdZd ddZd ddZddZd d Zd S)Eventz The simulator supports the event API, but they do not record timing info, and all simulation is synchronous. Execution time is not recorded. rcCsdSrrrrrrrrecordBsz Event.recordcCsdSrrr*rrrwaitEsz Event.waitcCsdSrrrrrrrHszEvent.synchronizecCs tddS)NzSimulator timings are bogusgr )reventrrr elapsed_timeKszEvent.elapsed_timeN)r)r)rrrr r+r,rr.rrrrr)=s   r)NFTc sX|r td|dk rtd|dks6t|s6t|trJfdd} | St|dS)Nz)bounds checking is not supported for CUDAz Cannot link PTX in the simulatorcst|dS)N)devicefastmathdebugr)fnr1r/r0rr jitwrapperbs zjit..jitwrapper)r/r1)NotImplementedErrorr isinstancelistr) Z func_or_sigr/r1Zargtypesinlinerestyper0linkZ boundscheckoptcacher4rr3rjitSsr=ccs dVdSrrrrrr defer_cleanupksr>)r) NFFNFNFNNTN)r contextlibrZcudadrv.devicesrrrZkernelrZnumba.core.sigutilsr warningsr r#r rrrrobjectrrr"r%r'r(r)r-r=r>rrrrs6