
    Vh                     n   d dl mZ d dlZd dlmZ d dlmZ  ej                         sej                  j                  dd      Z
e
j                  dej                  j                         ej                  j                  dd	      Zd
D ]  Zej!                  dd e         G d de      ZdedefdZyy)    )AnyN)Tensor)Function_inductor_testDEFzrealize(Tensor self) -> Tensor)tagsIMPL)CPUCUDAMetarealizec                 "    | j                         S N)clonexs    N/home/dcms/DCMS/lib/python3.12/site-packages/torch/_inductor/test_operators.py<lambda>r      s         c                   H    e Zd Zedededefd       Zedededefd       Zy)Realizectxr   returnc                 T    t         j                  j                  j                  |      S r   )torchopsr   r   )r   r   s     r   forwardzRealize.forward   s    99++33A66r   grad_outputc                     |d   S )Nr    )r   r   s     r   backwardzRealize.backward   s     q>!r   N)	__name__
__module____qualname__staticmethodobjectr   r   r   r!   r    r   r   r   r      sR    		7 	7F 	7v 	7 
	7 
	"# 	"S 	"S 	" 
	"r   r   r   r   c                 ,    t         j                  |       S r   )r   applyr   s    r   r   r      s    }}Qr   )typingr   torch.libraryr   r   torch.autogradr   _running_with_deploylibraryLibrary_test_lib_defdefineTagpt2_compliant_tag_test_lib_impldispatch_keyimplr   r   r    r   r   <module>r6      s       # "u!!#MM))*:EBM(uyy/J/J   ]]**+;VDN/ JI':LIJ"( " 6  f  ) $r   