iddlZddlmZddlmZmZej je dej je ddZy)N)CupyOps) has_torchhas_torch_cuda_gpuz needs PyTorch)reasonz needs a GPUcDddl}tjt}|jdj }|j |}|j |j k(sJ|jjj||jddyy)Nr) rr ) torchrxpzeroscudaasarrayshapetestingassert_allclosealloc2f)r opstas u/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/thinc/tests/regression/test_issue564.py test_issue564rs~zzi KK % % ' KKNww!''!!! && KKA  ) pytest thinc.apir thinc.compatrrmarkskipifrrrrsY 6 M/:**=A  B;  r