
    `iu
                     t    d dl mZ d dlZd Z G d dej                  Zedk    r ej                     dS dS )    )GUFuncEngineNc                     t          j        |           }|                    |          }|                                D ]2\  }}t	          ||          }||k    rd}t          ||||fz            3d S )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	            /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/numba/cuda/tests/cudapy/test_gufunc_scheduling.pytemplater      s    &y11D
--

C 4 41c1oo!886C 3{!2333 4 4    c                   8    e Zd Zd Zd Zd Zd Zd Zd Zd Z	dS )	TestGUFuncSchedulingc                 `    d}d}t          ddgdgdddg	          }t          |||           d S )
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   sQ    .	'V$H4=	
 
 
 	FG,,,,,r   c                 `    d}d}t          ddgdgdddg          }t          |||           d S )	Nr   )r   )r   r   r   r    r!   r"   r#   Fr$   r)   r+   s       r   test_signature_2z%TestGUFuncScheduling.test_signature_2   sQ    .	)V$H5>	
 
 
 	FG,,,,,r   c                 `    d}d}t          ddgdgdddg          }t          |||           d S )	Nr   ))   "   r   r   )r1   r2   r   r   r    r!   r"   )r1   r2   Fr$   r)   r+   s       r   test_signature_3z%TestGUFuncScheduling.test_signature_3&   sQ    .	/V$H5>	
 
 
 	FG,,,,,r   c                 `    d}d}t          ddgdgdddg          }t          |||           d S )	Nr   )r    r!   r    r!   r"    Fr$   r)   r+   s       r   test_signature_4z%TestGUFuncScheduling.test_signature_41   sQ    .	V$H5>	
 
 
 	FG,,,,,r   c                 `    d}d}t          ddgdgdddg          }t          |||           d S )Nz(a), (a) -> (a)r   r9   r9   r5   Fr$   r)   r+   s       r   test_signature_5z%TestGUFuncScheduling.test_signature_5<   sP    %	4LF5>	
 
 
 	FG,,,,,r   c                 `    d}d}t          ddgdgdddg          }t          |||           d S )N(), () -> ()r8   r5   r9   Fr$   r)   r+   s       r   test_signature_6z%TestGUFuncScheduling.test_signature_6G   sP    "	HD5>	
 
 
 	FG,,,,,r   c                 `    d}d}t          ddgdgdddg          }t          |||           d S )Nr<   )r9   r5   r5   r9   FTr$   r)   r+   s       r   test_signature_7z%TestGUFuncScheduling.test_signature_7R   sP    "	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.cuda.deviceufuncr   unittestr   TestCaser   r@   mainr5   r   r   <module>rH      s    / / / / / / 4 4 4L- L- L- L- L-8, L- L- L-^ zHMOOOOO r   