o i@sTddlZddlmZddlmZmZejje ddejje ddddZdS) N)CupyOps) has_torchhas_torch_cuda_gpuz needs PyTorch)reasonz needs a GPUcCs`ddl}tjdur.t}|d}||}|j|jks J|jj|| dddSdS)Nr) rr) torchrxpzeroscudaasarrayshapetestingassert_allclosealloc2f)ropstarX/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/regression/test_issue564.py test_issue564s   r) pytest thinc.apir thinc.compatrrmarkskipifrrrrrs