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)devrrg/var/www/html/eduruby.in/lip-sync/lip-sync-env/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)rZ 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.rr0rrrrr+=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)fnr4r2r3rr jitwrapperbs zjit..jitwrapper)r2r4)NotImplementedErrorr isinstancelistr) Z func_or_sigr2r4argtypesinlinerestyper3linkZ boundscheckoptcacher7rr6rjitSsrAccrrrrrrr defer_cleanupks rBr1) NFFNFNFNNTN)r" contextlibrZcudadrv.devicesrrrZkernelrZnumba.core.sigutilsr warningsr r%r rrrrobjectrrr$r'r)r*r+r/rArBrrrrs.