o i< @sdZddlmZddlmZmZmZddlmZddl m Z ddl m Z dd l mZmZmZd#d d Zd d ZGdddeZddZddZddZddZddZGdddeZeZ   d$dd Zed!d"ZdS)%z Contains CUDA API functions )contextmanager)require_contextresetgpusFakeCUDAKernel) is_signaturewarn)InOutInOutcCs|dksJddS)Nrz/Only a single device supported by the simulator)devrrU/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/cuda/simulator/api.py select_devicesrcCsdSNTrrrrris_float16_supportedrc@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 synchronize!rzstream.synchronizeN)__name__ __module__ __qualname____doc__rrrrrrrrs   rcCrrrrrrrr%rrcCs dt_dSr)rclosedrrrrclose)s r$cOrrr)argskwargsrrrdeclare_device-rr'cCs$tdtddtdddS)NzFound 1 CUDA deviceszid %d %20s %40s)r SIMULATORz [SUPPORTED]z %40s: 5.0zcompute capability)printrrrrdetect1s r*cCstSr)rrrrr list_devices7rr+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. rcCrrrrrrrrrecordBrz Event.recordcCrrrr-rrrwaitErz Event.waitcCrrrrrrrrHrzEvent.synchronizecCs tddS)NzSimulator timings are bogusgr )reventrrr elapsed_timeKszEvent.elapsed_timeNr)rr r!r"r.r/rr1rrrrr,=s    r,NFTc sX|rtd|durtd|dust|st|tr%fdd} | St|dS)Nz)bounds checking is not supported for CUDAz Cannot link PTX in the simulatorcst|dS)N)devicefastmathdebugr)fnr5r3r4rr jitwrapperbs zjit..jitwrapper)r3r5)NotImplementedErrorr isinstancelistr) func_or_sigr3r5argtypesinlinerestyper4link boundscheckoptcacher8rr7rjitSsrDccrrrrrrr defer_cleanupks rEr2) NFFNFNFNNTN)r" contextlibrcudadrv.devicesrrrkernelrnumba.core.sigutilsr warningsr r%r rrrrobjectrrr$r'r*r+r,r0rDrErrrrs.