1@j ddlZddlmZeddZdZGddZGdd ZGd d ZeZ d Z dd Z dZ dS)N) namedtuple _MemoryInfoz free,total)c*eZdZdZedZdS)FakeCUDADevicecd|_dS)Nz(GPU-00000000-0000-0000-0000-000000000000)uuidselfs UC:\Projects\Historical\venv\Lib\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>???XrrceZdZdZdZdZdZdZedZ 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|_dSr) _device_idr_device)r device_ids r rzFakeCUDAContext.__init__s#%'' rcdSrrr s r __enter__zFakeCUDAContext.__enter__ rcdSrr)r exc_typeexc_valexc_tbs r __exit__zFakeCUDAContext.__exit__r#rc.d|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,rrcdSrrr s r resetzFakeCUDAContext.reset0r#rcVttdtdS)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<<u666rc.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.memalloc;s z"D))))rFc,||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 rrs(((      >>>XXX   777***!!!!!!rrcFeZdZdZdZdZdZdZdZe 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 rzFakeDeviceList.__init__Ns#A&&( rc*d|_|j|S)NFrHrG)r devnums r __getitem__zFakeDeviceList.__getitem__Rs xrcJdd|jDS)Nz, c,g|]}t|Sr)str).0ds r z*FakeDeviceList.__str__..Ws333Q#a&&333r)joinrGr s r r+zFakeDeviceList.__str__Vs%yy33$(333444rc*t|jSr)iterrGr s r __iter__zFakeDeviceList.__iter__YsDH~~rc*t|jSr)lenrGr s r __len__zFakeDeviceList.__len__\s48}}rc.|jrdS|jdS)NrrJr s r currentzFakeDeviceList.current_s ; 4x{rN) rrrrCrrLr+rVrYrr[rrr rErEHs    555XrrEc*dtd_dS)NTr)gpusrHrrr r2r2isDGNNNrc t|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_contextrbqs  Kr)r) numpyr: collectionsrrrrrrEr]r2r_rbrrr res""""""j 55  3!3!3!3!3!3!3!3!l<~####r