xKg< dZddlmZddlmZmZmZddlmZddl m Z ddl m Z dd l mZmZmZdd Zd ZGd d eZdZdZdZdZdZGddeZeZ ddZedZy)z Contains CUDA API functions )contextmanager)require_contextresetgpusFakeCUDAKernel) is_signaturewarn)InOutInOutc|dk(sJdy)Nrz/Only a single device supported by the simulator)devs \/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/cuda/simulator/api.py select_devicers !8FFF8cyNTrrrris_float16_supportedrs rc&eZdZdZedZdZy)streamz The stream API is supported in the simulator - however, all execution occurs synchronously, so synchronization requires no operation. c#KdywNrselfs rauto_synchronizezstream.auto_synchronizes  cyrrrs r synchronizezstream.synchronize! rN)__name__ __module__ __qualname____doc__rr r#rrrrrs  rrcyrrrrrr#r#%rcdt_yr)rclosedrrrcloser-)s DKrcyrr)argskwargss rdeclare_devicer1-r*rcRtdtddztddzy)NzFound 1 CUDA deviceszid %d %20s %40s)r SIMULATORz [SUPPORTED]z %40s: 5.0zcompute capability)printrrrdetectr51s( ! !@ @A +, ,-rctSr)rrrr list_devicesr77s Krc,eZdZdZddZddZdZdZy)Eventz The simulator supports the event API, but they do not record timing info, and all simulation is synchronous. Execution time is not recorded. cyrrrrs rrecordz Event.recordBr$rcyrrr;s rwaitz Event.waitEr$rcyrrrs rr#zEvent.synchronizeHr$rctdy)NzSimulator timings are bogusgr )revents r elapsed_timezEvent.elapsed_timeKs *+rNr)r%r&r'r(r<r>r#rBrrrr9r9=s   rr9Nc |r td| td|t|st|tr fd} | St |S)Nz)bounds checking is not supported for CUDAz Cannot link PTX in the simulatorc"t|S)N)devicefastmathdebugr)fnrHrFrGs r jitwrapperzjit..jitwrapperbs!")/+3(-/ /r)rFrH)NotImplementedErrorr isinstancelistr ) func_or_sigrFrHargtypesinlinerestyperGlink boundscheckoptcacherJs `` ` rjitrVSs[ !"MNN !"DEE |K8+t, /  +fE BBrc#Kdywrrrrr defer_cleanuprXks  r!rC) NFFNFNFNNTN)r( contextlibrcudadrv.devicesrrrkernelr numba.core.sigutilsr warningsr r/rrrrrobjectrr#r-r1r5r7r9rArVrXrrrr_s &99",!!G  V    .  F& ?C9=*.C0  r