o
    
i                     @   s   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d Ze jje ddd	d
 Ze jje p:e dddd Ze jje pJe dddd ZdS )    N)has_cupy_gpu	has_torchhas_tensorflowrequire_gpu)use_pytorch_for_gpu_memoryuse_tensorflow_for_gpu_memorycupy_logger_v1c                  C   s   t  } | d \}}|S )Nr   )setup_loggerstep_ r   Q/home/ubuntu/.local/lib/python3.10/site-packages/spacy_loggers/tests/test_cupy.pylogger   s   r   zCuPy support required)reasonc                 C   s&   t   i }| | |d dksJ d S )Ncupy.pool.sourcedefault)r   r   infor   r   r   "test_cupy_allocator_source_default   s   r   zCuPy/PyTorch support requiredc                 C   ,   t   t  i }| | |d dksJ d S )Nr   pytorch)r   r   r   r   r   r    test_cupy_allocator_source_torch   
   r   z CuPy/TensorFlow support requiredc                 C   r   )Nr   
tensorflow)r   r   r   r   r   r   test_cupy_allocator_source_tf$   r   r   )pytest
thinc.utilr   r   r   r   thinc.backendsr   r   spacy_loggers.cupyr	   fixturer   markskipifr   r   r   r   r   r   r   <module>   s     

	
