
    i.                     l    S SK r S SKJrJrJrJrJrJrJrJ	r	J
r
JrJrJrJrJrJrJr  S\4S jrSS jrg)    N)AllocationDetailsChainContainerMetadataEValueExecutionPlanInstructionInt
KernelCallNullOperatorProgram
ScalarTypeStringSubsegmentOffsetsTensorTensorShapeDynamismreturnc                     [        S[        S[        [        S5      5      [        [        S5      5      [        [	        5       5      [        [        S5      5      [        [        [        R                  SSS/SS/SSS[        SSSS9[        R                  S	9	S
9/S/S/[        / / [        [        SSS/S95      /S S9/[        SSS9[!        SSS9// SS/S9	// / / [#        S/ S9/ S9$ )Nr   forward   pass   F   )	memory_idmemory_offset_highmemory_offset_low)	scalar_typestorage_offsetsizes	dim_orderrequires_gradlayoutdata_buffer_idxallocation_infoshape_dynamism)val)op_indexargs)inputsoutputsinstructions
stacktraceplace)encoded_inp_strencoded_out_strz	aten::addr   )nameoverloadi   )	r0   valuesr)   r*   chainscontainer_meta_type	operators	delegatesnon_const_buffer_sizes)segment_indexoffsets)versionexecution_planconstant_bufferbackend_delegate_datasegmentsconstant_segment
named_data)r   r   r   r	   r   r   r   r   FLOATr   r   STATICr   r   r
   r   r   r        [/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/exir/tests/common.pyget_test_programrF      s   3q6N3q6N46N6&>*"(2(8(8+,#$a&'(!f*/#$,-,=*+3424-
 ,?+E+E. s! "&1*aqRSf2U&V%W#'	 %6$+W% $xHI()4yQ)+
X  *BGe3 3rD   c                     [         R                  R                  n [         R                  R                  R
                  R                  R                  R                  U 5        [         R                  R                  R                  R                  R                  R                  U 5        [         R                  R                  R                  R                  R                  R                  U 5        [         R                  R                  R                  R                  R                  R                  U 5        [         R                  R                  R                  R                  R                  R                  U 5        [         R                  R                  R                  R                  R                  R                  U 5        [         R                  R                  R                  R                  R                  R                  U 5        g )N)torchTagcoreopsatenmaxdefaulttagsappendsumsearchsortedr   	ones_likeupsample_nearest2dindexaddbmm)	canonicals    rE   !register_additional_test_aten_opsrX   V   s)    		I	IINN##**95	IINN##**95	IINN&&++229=	IINN$$))00;	IINN%%--2299)D	IINN$$++I6	IINN!!&&--i8rD   )r   N)rH   executorch.exir.schemar   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   rF   rX   rC   rD   rE   <module>rZ      s5        (4' 4n
9rD   