U hd @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)uuidselfr]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/cuda/simulator/cudadrv/devices.py__init__szFakeCUDADevice.__init__N)__name__ __module__ __qualname__r rrrr rsrc@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)rZ device_idrrr r szFakeCUDAContext.__init__cCsdSrrrrrr __enter__szFakeCUDAContext.__enter__cCsdSrr)rexc_typeexc_valexc_tbrrr __exit__szFakeCUDAContext.__exit__cCs dj|dS)Nzr)formatrrrr __str__szFakeCUDAContext.__str__cCs|jSr)rrrrr idszFakeCUDAContext.idcCs|jSr)rrrrr device"szFakeCUDAContext.devicecCsdS)N)rrrrr compute_capability&sz"FakeCUDAContext.compute_capabilitycCsdSrrrrrr reset*szFakeCUDAContext.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)Zdtype)npZndarray)rszrrr memalloc5szFakeCUDAContext.memallocFcCs ||S)zAllocates memory on the host)r%)rr$ZmappedZportablewcrrr memhostalloc=szFakeCUDAContext.memhostallocN)FFF)r r r __doc__r rrrpropertyrrrrr!r%r'rrrr 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)NFr,r+)rdevnumrrr __getitem__LszFakeDeviceList.__getitem__cCsddd|jDS)Nz, cSsg|] }t|qSr)str).0drrr Qsz*FakeDeviceList.__str__..)joinr+rrrr rPszFakeDeviceList.__str__cCs t|jSr)iterr+rrrr __iter__SszFakeDeviceList.__iter__cCs t|jSr)lenr+rrrr __len__VszFakeDeviceList.__len__cCs|jr dS|jdS)Nrr-rrrr currentYszFakeDeviceList.currentN) r r r r(r r/rr6r8r)r9rrrr r*Bsr*cCsdtd_dS)NTr)gpusr,rrrr rcsrcCst|Sr)r)r.rrr get_contextgsr;cCs|S)zP In the simulator, a context is always "available", so this is a no-op. r)funcrrr require_contextksr=)r) numpyr# collectionsrrrrr*r:rr;r=rrrr s  6