o 8ήc @slddlZddlmZeddZGdddZGdddZGd d d ZeZd d Z dd dZ ddZ dS)N) namedtuple _MemoryInfoz free,totalc@seZdZddZdS)FakeCUDADevicecCs d|_dS)Nz(GPU-00000000-0000-0000-0000-000000000000)uuidselfrK/tmp/pip-target-vg8gfxp4/lib/python/numba/cuda/simulator/cudadrv/devices.py__init__ zFakeCUDADevice.__init__N)__name__ __module__ __qualname__r rrrr rs rc@sveZdZdZddZddZddZdd Zed d Z ed d Z eddZ ddZ ddZ ddZdddZdS)FakeCUDAContextz` This stub implements functionality only for simulating a single GPU at the moment. cCs||_t|_dSN) _device_idr_device)r device_idrrr r s zFakeCUDAContext.__init__cCdSrrrrrr __enter__zFakeCUDAContext.__enter__cCrrr)rexc_typeexc_valexc_tbrrr __exit__rzFakeCUDAContext.__exit__cCs dj|dS)Nzr)formatrrrr __str__s zFakeCUDAContext.__str__cC|jSr)rrrrr idzFakeCUDAContext.idcCrr)rrrrr device"rzFakeCUDAContext.devicecCsdS)N)rrrrr compute_capability&sz"FakeCUDAContext.compute_capabilitycCrrrrrrr reset*rzFakeCUDAContext.resetcCsttdtdS)z Cross-platform free / total host memory is hard without external dependencies, e.g. `psutil` - so return infinite memory to maintain API type compatibility inf)rfloatrrrr get_memory_info-szFakeCUDAContext.get_memory_infocCstj|ddS)z Allocates memory on the simulated device At present, there is no division between simulated host memory and simulated device memory. u1)dtype)npndarray)rszrrr memalloc5szFakeCUDAContext.memallocFcCs ||S)zAllocates memory on the host)r-)rr,mappedportablewcrrr memhostalloc=s zFakeCUDAContext.memhostallocN)FFF)r r r__doc__r rrrpropertyrr r#r$r'r-r1rrrr r s    rc@sDeZdZdZddZddZddZdd Zd d Ze d d Z dS)FakeDeviceListz This stub implements a device list containing a single GPU. It also keeps track of the GPU status, i.e. whether the context is closed or not, which may have been set by the user calling reset() cCstdf|_d|_dS)NrF)rlstclosedrrrr r Hs  zFakeDeviceList.__init__cCsd|_|j|S)NFr6r5)rdevnumrrr __getitem__Ls zFakeDeviceList.__getitem__cCsddd|jDS)Nz, cSsg|]}t|qSr)str).0drrr Qsz*FakeDeviceList.__str__..)joinr5rrrr rPszFakeDeviceList.__str__cC t|jSr)iterr5rrrr __iter__Sr zFakeDeviceList.__iter__cCr?r)lenr5rrrr __len__Vr zFakeDeviceList.__len__cCs|jrdS|jdS)Nrr7rrrr currentYs zFakeDeviceList.currentN) r r rr2r r9rrArCr3rDrrrr r4Bsr4cCsdtd_dS)NTr)gpusr6rrrr r$csr$cCst|Sr)r)r8rrr get_contextgsrFcCs|S)zP In the simulator, a context is always "available", so this is a no-op. r)funcrrr require_contextksrH)r) numpyr* collectionsrrrrr4rEr$rFrHrrrr s  6