
    wgy
                     v    d dl mZ d dlZd Z G d dej                        Zedk(  r ej                          yy)    )GUFuncEngineNc                     t        j                  |       }|j                  |      }|j                         D ])  \  }}t	        ||      }||k7  sd}t        ||||fz         y )Nz"error for %s: got=%s but expect=%s)r   from_signaturescheduleitemsgetattrAssertionError)		signatureshapesexpectsgufbschkvgotfmts	            s/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numba/cuda/tests/cudapy/test_gufunc_scheduling.pytemplater      sf    &&y1D
--
C 41c1o!86C 3{!233	4    c                   6    e Zd Zd Zd Zd Zd Zd Zd Zd Z	y)	TestGUFuncSchedulingc                 N    d}d}t        ddgdgdddg	      }t        |||       y )
N(m, n), (n, p) -> (m, p))d         )   r      r   r   r   r   r   r   r   FTishapesoshapesloopdimspinneddictr   selfr
   r   r   s       r   test_signature_1z%TestGUFuncScheduling.test_signature_1   s;    .	'V$H4=	
 	FG,r   c                 N    d}d}t        ddgdgdddg      }t        |||       y )	Nr   )r   )r   r   r   r    r!   r"   r#   Fr$   r)   r+   s       r   test_signature_2z%TestGUFuncScheduling.test_signature_2   s;    .	)V$H5>	
 	FG,r   c                 N    d}d}t        ddgdgdddg      }t        |||       y )	Nr   ))   "   r   r   )r1   r2   r   r   r    r!   r"   )r1   r2   Fr$   r)   r+   s       r   test_signature_3z%TestGUFuncScheduling.test_signature_3&   s;    .	/V$H5>	
 	FG,r   c                 N    d}d}t        ddgdgdddg      }t        |||       y )	Nr   )r    r!   r    r!   r"    Fr$   r)   r+   s       r   test_signature_4z%TestGUFuncScheduling.test_signature_41   s;    .	V$H5>	
 	FG,r   c                 N    d}d}t        ddgdgdddg      }t        |||       y )Nz(a), (a) -> (a)r   r9   r9   r5   Fr$   r)   r+   s       r   test_signature_5z%TestGUFuncScheduling.test_signature_5<   s:    %	4LF5>	
 	FG,r   c                 N    d}d}t        ddgdgdddg      }t        |||       y )N(), () -> ()r8   r5   r9   Fr$   r)   r+   s       r   test_signature_6z%TestGUFuncScheduling.test_signature_6G   s:    "	HD5>	
 	FG,r   c                 N    d}d}t        ddgdgdddg      }t        |||       y )Nr<   )r9   r5   r5   r9   FTr$   r)   r+   s       r   test_signature_7z%TestGUFuncScheduling.test_signature_7R   s:    "	HD4=	
 	FG,r   N)
__name__
__module____qualname__r-   r/   r3   r6   r:   r=   r?   r5   r   r   r   r      s%    	-	-	-	-	-	-	-r   r   __main__)numba.np.ufunc.deviceufuncr   unittestr   TestCaser   r@   mainr5   r   r   <module>rH      s?    3 4L-8,, L-^ zHMMO r   