
    O j                     Z    S SK r S SKJrJr  S\S\ R                  R
                  SS4S jrg)    N)
OpOverloadOpOverloadPacketopgraphreturnc                     [        U [        5      (       a  [        SU  35      e[        U [        5      (       d  [        S[	        U 5       35      e[
        R                  R                  U R                  U5        g )Nz;Must pass specific op overload, not overload packet, found zExpected OpOverload, got )	
isinstancer   AssertionErrorr   typetorch_C&_jit_register_decomposition_for_schema_schema)r   r   s     o/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/torch/jit/_decomposition_utils.py_register_decompositionr      se    "&''I"N
 	
 b*%%8b
CDD	HH33BJJF    )r   
torch._opsr   r   r   Graphr    r   r   <module>r      s1     3G
 G588>> Gd Gr   