o 8ήc @sdZddlmZddlmZmZmZddlmZddl m Z ddl m Z dd l mZmZmZd!d d ZGd d d eZddZddZddZddZddZGdddeZeZ   d"ddZedd ZdS)#z Contains CUDA API functions )contextmanager)require_contextresetgpusFakeCUDAKernel) Signaturewarn)InOutInOutcCs|dksJddS)Nrz/Only a single device supported by the simulator)devrr?/tmp/pip-target-vg8gfxp4/lib/python/numba/cuda/simulator/api.py select_devicesrc@s$eZdZdZeddZddZdS)streamz The stream API is supported in the simulator - however, all execution occurs synchronously, so synchronization requires no operation. cc dVdSNrselfrrrauto_synchronizes zstream.auto_synchronizecCdSrrrrrr synchronizezstream.synchronizeN)__name__ __module__ __qualname____doc__rrrrrrrrs   rcCrrrrrrrr rrcCs dt_dS)NT)rclosedrrrrclose$s r"cOrrr)argskwargsrrrdeclare_device(rr%cCs$tdtddtdddS)NzFound 1 CUDA deviceszid %d %20s %40s)r SIMULATORz [SUPPORTED]z %40s: 5.3zcompute capability)printrrrrdetect,s r(cCstSr)rrrrr list_devices2rr)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. rcCrrrrrrrrrecord=rz Event.recordcCrrrr+rrrwait@rz Event.waitcCrrrrrrrrCrzEvent.synchronizecCs tddS)NzSimulator timings are bogusgr )reventrrr elapsed_timeFszEvent.elapsed_timeNr)rrrr r,r-rr/rrrrr*8s    r*NFTc sV|rtd|durtd|dust|tttfr$fdd} | St|dS)Nz)bounds checking is not supported for CUDAz Cannot link PTX in the simulatorcst|dS)N)devicefastmathdebugr)fnr3r1r2rr jitwrapper[s zjit..jitwrapper)r1r3)NotImplementedError isinstancestrtupler r) func_or_sigr1r3argtypesinlinerestyper2link boundscheckoptcacher6rr5rjitNsrCccrrrrrrr defer_cleanupds rDr0) NFFNFNFNNTN)r contextlibrcudadrv.devicesrrrkernelrnumba.core.typingr warningsr r#r rrrobjectrrr"r%r(r)r*r.rCrDrrrrs,