o
    ÁÙ¾i*  ã                   @   s˜   d dl Z dde jdede jfdd„Zdde jdede jfdd	„Zdde jdede jfd
d„Zdde jdede jfdd„Zdde jdede jfdd„ZdS )é    Nç      ð?ÚxÚscaleÚreturnc                 C   ó   t jjj | |¡S ©N)ÚtorchÚopsÚ
sgl_kernelÚfast_hadamard_transformÚdefault©r   r   © r   úG/home/ubuntu/.local/lib/python3.10/site-packages/sgl_kernel/hadamard.pyÚhadamard_transform   ó   r   c                 C   r   r   )r   r	   r
   Úfast_hadamard_transform_12Nr   r   r   r   r   Úhadamard_transform_12n   r   r   c                 C   r   r   )r   r	   r
   Úfast_hadamard_transform_20Nr   r   r   r   r   Úhadamard_transform_20n   r   r   c                 C   r   r   )r   r	   r
   Úfast_hadamard_transform_28Nr   r   r   r   r   Úhadamard_transform_28n   r   r   c                 C   r   r   )r   r	   r
   Úfast_hadamard_transform_40Nr   r   r   r   r   Úhadamard_transform_40n   r   r   )r   )r   ÚTensorÚfloatr   r   r   r   r   r   r   r   r   Ú<module>   s     