sh SSKrSSKJr \"SS5rSr"SS5r"SS 5r"S S 5r\"5r S r SS jr Sr g)N) namedtuple _MemoryInfoz free,total)c*\rSrSrSr\S5rSrg)FakeCUDADevice cSUlg)Nz(GPU-00000000-0000-0000-0000-000000000000uuidselfs x/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/simulator/cudadrv/devices.py__init__FakeCUDADevice.__init__ s > c[$N _SIMULATOR_CCr s rcompute_capability!FakeCUDADevice.compute_capability rr N)__name__ __module__ __qualname____firstlineno__rpropertyr__static_attributes__rrrr s?rrc|\rSrSrSrSrSrSrSr\ S5r \ S5r \ S 5r S r S rS rSS jrSrg)FakeCUDAContextzT This stub implements functionality only for simulating a single GPU at the moment. c.Xl[5Ulgr) _device_idr_device)r device_ids rrFakeCUDAContext.__init__s#%' rcgrr r s r __enter__FakeCUDAContext.__enter__ rcgrr )rexc_typeexc_valexc_tbs r__exit__FakeCUDAContext.__exit__r,rc SRUS9$)Nzr )formatr s r__str__FakeCUDAContext.__str__!s+222==rcUR$r)r%r s ridFakeCUDAContext.id$s rcUR$r)r&r s rdeviceFakeCUDAContext.device(s ||rc[$rrr s rr"FakeCUDAContext.compute_capability,rrcgrr r s rresetFakeCUDAContext.reset0r,rc>[[S5[S55$)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 rget_memory_infoFakeCUDAContext.get_memory_info3s 5<u66rc,[R"USS9$)z Allocates memory on the simulated device At present, there is no division between simulated host memory and simulated device memory. u1)dtype)npndarray)rszs rmemallocFakeCUDAContext.memalloc;s zz"D))rc$URU5$)zAllocates memory on the host)rM)rrLmappedportablewcs r memhostallocFakeCUDAContext.memhostallocCs}}R  r)r&r%N)FFF)rrrr__doc__rr*r1r5rr8r;rr@rErMrSrr rrr"r"sk(  > 7*!rr"cF\rSrSrSrSrSrSrSrSr \ S5r S r g ) FakeDeviceListHz 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() c4[S54UlSUlg)NrF)r"lstclosedr s rrFakeDeviceList.__init__Ns#A&( rc.SUlURU$)NFr[rZ)rdevnums r __getitem__FakeDeviceList.__getitem__Rs xxrcvSRURVs/sHn[U5PM sn5$s snf)Nz, )joinrZstr)rds rr5FakeDeviceList.__str__Vs+yy$((3(Q#a&(3443s6c,[UR5$r)iterrZr s r__iter__FakeDeviceList.__iter__YsDHH~rc,[UR5$r)lenrZr s r__len__FakeDeviceList.__len__\s488}rcDUR(agURS$)Nrr^r s rcurrentFakeDeviceList.current_s ;;xx{rr^N) rrrrrUrr`r5rirmrrprr rrrWrWHs4  5rrWc S[Slg)NTr)gpusr[r rrr@r@is DGNrc[U5$r)r")r_s r get_contextrums 6 ""rcU$)zH In the simulator, a context is always "available", so this is a no-op. r )funcs rrequire_contextrxqs  Kr)r) numpyrJ collectionsrrrrr"rWrsr@rurxr rrr{sT" 5  3!3!l<#r