
    i                     P   d dl Z d dlmZmZmZmZ d dlmZmZ d dl	m
Z
 e j                  d        Ze j                  j                  e d      d        Ze j                  j                  e xs e d	      d
        Ze j                  j                  e xs e d      d        Zy)    N)has_cupy_gpu	has_torchhas_tensorflowrequire_gpu)use_pytorch_for_gpu_memoryuse_tensorflow_for_gpu_memorycupy_logger_v1c                  0    t               }  | d       \  }}|S )Nr	   )setup_loggerstep_s      n/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/spacy_loggers/tests/test_cupy.pyloggerr      s    !#L4 GD!K    zCuPy support required)reasonc                 @    t                i } | |       |d   dk(  sJ y )Ncupy.pool.sourcedefault)r   r   infos     r   "test_cupy_allocator_source_defaultr      s)     MD
4L"#y000r   zCuPy/PyTorch support requiredc                 T    t                t                i } | |       |d   dk(  sJ y )Nr   pytorch)r   r   r   s     r    test_cupy_allocator_source_torchr      s/     M D
4L"#y000r   z CuPy/TensorFlow support requiredc                 T    t                t                i } | |       |d   dk(  sJ y )Nr   
tensorflow)r   r   r   s     r   test_cupy_allocator_source_tfr   $   s/     M!#D
4L"#|333r   )pytest
thinc.utilr   r   r   r   thinc.backendsr   r   spacy_loggers.cupyr
   fixturer   markskipifr   r   r    r   r   <module>r'      s     K K T -   $-DE1 F1 %I.M  11 *N*3U  44r   