h! %SSKrSSKrSSKJr SSKJrJrJrJrJ r J r SSK J r SSK JrJr SSKJrJrJrJrJr SS KJrJr SS KJr SS KJr SS KJr SS KJ r SSK!J"r" \"SSS9r#\\\"\$S'\"S0S9r%\\&\$S'SS0r'S\(SS4Sjr)S%Sjr*S%Sjr+Sr,S\(S\"4Sjr-Sr.\R^S\(4Sj5r0S\"4Sjr1S\"SS4Sjr2S\34Sjr4S r5\Rl4S!\\(\4S"\ \Rl4S#jjr7/S$Qr8g)&N) ContextVar)AnyCallableDictOptionalTypecast)registry)cupyhas_cupy)assert_pytorch_installedassert_tensorflow_installedget_torch_default_device is_cupy_array require_cpu)cupy_pytorch_allocatorcupy_tensorflow_allocator) ParamServer)CupyOps)MPSOps)NumpyOps)Ops context_ops)default context_poolsops allocatorreturncdUS:Xa [5 gUS:Xa [5 g[SUS35e)z~Route GPU memory allocation via PyTorch or tensorflow. Raise an error if the given argument does not match either of the two. pytorch tensorflowz#Invalid 'gpu_allocator' argument: 'z4'. Available allocators are: 'pytorch', 'tensorflow'N)use_pytorch_for_gpu_memoryuse_tensorflow_for_gpu_memory ValueErrorrs Q/home/james-whalen/.local/lib/python3.13/site-packages/thinc/backends/__init__.pyset_gpu_allocatorr)s:I"$ l "%'1)

>r*attrs local_classc^U"5nUR5Hup4[X#U5 M U$)N)itemssetattr)rarbobjr<values r(r^r^s, -C{{} 5!% Jr*) rOrNrQrrrrrr )r N)9 contextlibr[ contextvarsrtypingrrrrrr r compatr r utilrrrrr_cupy_allocatorsrr _param_serverrcupy_opsrmps_opsr numpy_opsrrrr__annotations__rdictr_strr)r$r%r;rIrLcontextmanagerrQrNrOboolrXrUlocalr^__all__rBr*r(rzs7"<<#P&)3M4)P Z &P",_b"I z$I       5*8" #C, %#%%((""" 4AJ S>(,Y__(= r*