xKg ddlZddlmZeddZdZGddZGdd ZGd d ZeZ d Z dd Z dZ y)N) namedtuple _MemoryInfoz free,total)c"eZdZdZedZy)FakeCUDADevicecd|_y)Nz(GPU-00000000-0000-0000-0000-000000000000)uuidselfs h/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/cuda/simulator/cudadrv/devices.py__init__zFakeCUDADevice.__init__ s > ctSN _SIMULATOR_CCr s r compute_capabilityz!FakeCUDADevice.compute_capability rN)__name__ __module__ __qualname__rpropertyrrr rr s?rrcreZdZdZdZdZdZdZedZ edZ edZ d Z d Z d Zdd Zy )FakeCUDAContextz` This stub implements functionality only for simulating a single GPU at the moment. c0||_t|_yr) _device_idr_device)r device_ids r rzFakeCUDAContext.__init__s#%' rcyrrr s r __enter__zFakeCUDAContext.__enter__ rcyrr)r exc_typeexc_valexc_tbs r __exit__zFakeCUDAContext.__exit__r#rc&dj|S)Nzr )formatr s r __str__zFakeCUDAContext.__str__!s+222==rc|jSr)rr s r idzFakeCUDAContext.id$s rc|jSr)rr s r devicezFakeCUDAContext.device(s ||rctSrrr s r rz"FakeCUDAContext.compute_capability,rrcyrrr s r resetzFakeCUDAContext.reset0r#rc>ttdtdS)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_info3s 5<u66rc0tj|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.memalloc;s zz"D))rc$|j|S)zAllocates memory on the host)r=)r r<mappedportablewcs r memhostalloczFakeCUDAContext.memhostallocCs}}R  rN)FFF)rrr__doc__rr"r(r+rr-r/rr2r6r=rBrrr rrsk(  > 7*!rrc>eZdZdZdZdZdZdZdZe dZ y) 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() c4tdf|_d|_y)NrF)rlstclosedr s r rzFakeDeviceList.__init__Ns#A&( rc.d|_|j|S)NFrHrG)r devnums r __getitem__zFakeDeviceList.__getitem__Rs xxrcpdj|jDcgc] }t|c}Scc}w)Nz, )joinrGstr)r ds r r+zFakeDeviceList.__str__Vs+yy$((3(Q#a&(3443s3c,t|jSr)iterrGr s r __iter__zFakeDeviceList.__iter__YsDHH~rc,t|jSr)lenrGr s r __len__zFakeDeviceList.__len__\s488}rc:|jry|jdS)NrrJr s r currentzFakeDeviceList.current_s ;;xx{rN) rrrrCrrLr+rSrVrrXrrr rErEHs4  5rrEc dtd_y)NTr)gpusrHrrr r2r2is DGNrct|Sr)r)rKs r get_contextr\ms 6 ""rc|S)zP In the simulator, a context is always "available", so this is a no-op. r)funcs r require_contextr_qs  Kr)r) numpyr: collectionsrrrrrrErZr2r\r_rrr rbsT" 5  3!3!l<#r