g ddlZddlmZeddZGddZGddZGd d ZeZd Z dd Z d Z dS)N) namedtuple _MemoryInfoz free,totalceZdZdZdS)FakeCUDADevicecd|_dS)Nz(GPU-00000000-0000-0000-0000-000000000000)uuidselfs Z/granary/edev/quantum/lib/python3.11/site-packages/numba/cuda/simulator/cudadrv/devices.py__init__zFakeCUDADevice.__init__s > N)__name__ __module__ __qualname__r r r rrs#?????r rceZdZdZdZdZdZdZedZ edZ edZ d Z d Z d Zdd ZdS)FakeCUDAContextz` This stub implements functionality only for simulating a single GPU at the moment. c:||_t|_dSN) _device_idr_device)r device_ids r r zFakeCUDAContext.__init__s#%'' r cdSrrr s r __enter__zFakeCUDAContext.__enter__ r cdSrr)r exc_typeexc_valexc_tbs r __exit__zFakeCUDAContext.__exit__rr c.d|S)Nzr )formatr s r __str__zFakeCUDAContext.__str__s+222===r c|jSr)rr s r idzFakeCUDAContext.ids r c|jSr)rr s r devicezFakeCUDAContext.device"s |r cdS)N)rr s r compute_capabilityz"FakeCUDAContext.compute_capability&svr cdSrrr s r resetzFakeCUDAContext.reset*rr cVttdtdS)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)rfloatr s r get_memory_infozFakeCUDAContext.get_memory_info-s 5<<u666r c.tj|dS)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 szs r memalloczFakeCUDAContext.memalloc5s z"D))))r Fc,||S)zAllocates memory on the host)r8)r r7mappedportablewcs r memhostalloczFakeCUDAContext.memhostalloc=s}}R   r N)FFF)rrr__doc__r rr r#propertyr%r'r+r-r1r8r=rr r rr s(((      >>>XXX   777***!!!!!!r rcFeZdZdZdZdZdZdZdZe dZ 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() c>tdf|_d|_dS)NrF)rlstclosedr s r r zFakeDeviceList.__init__Hs#A&&( r c*d|_|j|S)NFrDrC)r devnums r __getitem__zFakeDeviceList.__getitem__Ls xr cJdd|jDS)Nz, c,g|]}t|Sr)str).0ds r z*FakeDeviceList.__str__..Qs333Q#a&&333r )joinrCr s r r#zFakeDeviceList.__str__Ps%yy33$(333444r c*t|jSr)iterrCr s r __iter__zFakeDeviceList.__iter__SsDH~~r c*t|jSr)lenrCr s r __len__zFakeDeviceList.__len__Vs48}}r c.|jrdS|jdS)NrrFr s r currentzFakeDeviceList.currentYs ; 4x{r N) rrrr>r rHr#rRrUr?rWrr r rArABs    555Xr rAc*dtd_dS)NTr)gpusrDrr r r-r-csDGNNNr c t|Sr)r)rGs r get_contextr[gs 6 " ""r c|S)zP In the simulator, a context is always "available", so this is a no-op. r)funcs r require_contextr^ks  Kr )r) numpyr5 collectionsrrrrrArYr-r[r^rr r ras""""""j 55 ???????? 3!3!3!3!3!3!3!3!l<~####r