o i @spddlZddlmZeddZdZGdddZGdd d ZGd d d ZeZ d d Z dddZ ddZ dS)N) namedtuple _MemoryInfoz free,total)c@s eZdZddZeddZdS)FakeCUDADevicecCs d|_dS)Nz(GPU-00000000-0000-0000-0000-000000000000)uuidselfr b/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/cuda/simulator/cudadrv/devices.py__init__  zFakeCUDADevice.__init__cCtSN _SIMULATOR_CCrr r r compute_capability z!FakeCUDADevice.compute_capabilityN)__name__ __module__ __qualname__r propertyrr r r r r src@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|_dSr) _device_idr_device)r device_idr r r r s zFakeCUDAContext.__init__cCdSrr rr r r __enter__zFakeCUDAContext.__enter__cCrrr )r exc_typeexc_valexc_tbr r r __exit__rzFakeCUDAContext.__exit__cCs dj|dS)Nzr)formatrr r r __str__!s zFakeCUDAContext.__str__cC|jSr)rrr r r id$zFakeCUDAContext.idcCr%r)rrr r r device(r'zFakeCUDAContext.devicecCrrrrr r r r,rz"FakeCUDAContext.compute_capabilitycCrrr rr r r reset0rzFakeCUDAContext.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)rfloatrr r r get_memory_info3szFakeCUDAContext.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)r szr r r memalloc;szFakeCUDAContext.memallocFcCs ||S)zAllocates memory on the host)r2)r r1mappedportablewcr r r memhostallocCs zFakeCUDAContext.memhostallocN)FFF)rrr__doc__r rr"r$rr&r(rr)r,r2r6r r r r rs    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)rlstclosedrr r r r Ns  zFakeDeviceList.__init__cCsd|_|j|S)NFr:r9)r devnumr r r __getitem__Rs zFakeDeviceList.__getitem__cCsddd|jDS)Nz, cSsg|]}t|qSr )str).0dr r r Wsz*FakeDeviceList.__str__..)joinr9rr r r r$VszFakeDeviceList.__str__cC t|jSr)iterr9rr r r __iter__Yr zFakeDeviceList.__iter__cCrCr)lenr9rr r r __len__\r zFakeDeviceList.__len__cCs|jrdS|jdS)Nrr;rr r r current_s zFakeDeviceList.currentN) rrrr7r r=r$rErGrrHr r r r r8Hsr8cCsdtd_dS)NTr)gpusr:r r r r r)isr)cCst|Sr)r)r<r r r get_contextmsrJcCs|S)zP In the simulator, a context is always "available", so this is a no-op. r )funcr r r require_contextqsrL)r) numpyr/ collectionsrrrrrr8rIr)rJrLr r r r s   6