U ,d @sdZddlmZddlmZmZmZddlmZddl m Z ddl m Z dd l mZmZmZd!d d ZGd d d eZddZddZddZddZddZGdddeZeZd"ddZedd ZdS)#z Contains CUDA API functions )contextmanager)require_contextresetgpusFakeCUDAKernel) Signaturewarn)InOutInOutcCs|dkstddS)Nrz/Only a single device supported by the simulator)AssertionError)devr.jitwrapper)r-r/)NotImplementedError isinstancestrtupler r) Z func_or_sigr-r/argtypesinlinerestyper.linkZ boundscheckoptcacher2rr1rjitNsr=ccs dVdSrrrrrr defer_cleanupdsr>)r) NFFNFNFNNTN)r contextlibrZcudadrv.devicesrrrkernelrZnumba.core.typingr warningsr r!r rrrobjectrrr r#r%r&r'r+r=r>rrrrs4