hSSKrSSKJr SSKJrJr \R R\(+SS9\R R\(+SS9S55rg)N)CupyOps) has_torchhas_torch_cuda_gpuz needs PyTorch)reasonz needs a GPUcHSSKn[Rb[5nURS5R 5nUR U5nUR UR :XdeURRRUURSS55 gg)Nr) rr ) torchrxpzeroscudaasarrayshapetestingassert_allclosealloc2f)r opstas ^/home/james-whalen/.local/lib/python3.13/site-packages/thinc/tests/regression/test_issue564.py test_issue564rs~zzi KK % % ' KKNww!''!!! && KKA  ) pytest thinc.apir thinc.compatrrmarkskipifrrrrsY 6 M/:**=A  B;  r