o
    پi                     @   sV   d dl Z d dlZd dlZd dlmZ defddZdefddZd	edefd
dZ	dS )    N)Callablereturnc                  C   s   t j \} }| S N)torchcudaget_device_capability)majorminor r
   K/home/ubuntu/.local/lib/python3.10/site-packages/deep_gemm/testing/utils.pyget_arch_major   s   r   	conditionc                    s    fdd}|S )Nc                    s   t   fdd}|S )Nc                     s@     r| i | d S t j d t d   t   d S )N:z > Filtered by )print__name__)argskwargs)r   funcr
   r   wrapper   s
   
z/test_filter.<locals>.decorator.<locals>.wrapper	functoolswrapsr   r   r   r   r   	decorator   s   ztest_filter.<locals>.decoratorr
   )r   r   r
   r   r   test_filter   s   
r   namec                    s    fdd}|S )Nc                    s   t   fdd}|S )Nc                     sN     rt jd }| i | |d ur|t j< d S d S | i | d S r   )osenvironpop)r   r   saved)r   r   r   r
   r   r      s   z.ignore_env.<locals>.decorator.<locals>.wrapperr   r   r   r   r   r   r      s   	zignore_env.<locals>.decoratorr
   )r   r   r   r
   r"   r   
ignore_env   s   r#   )
r   r   r   typingr   intr   r   strr#   r
   r
   r
   r   <module>   s    