+
    "Ý³j§  ã                   ó˜   € ^ RI t ^ RIt^ RIHt ^ RIHt ^ RIHt  ! R R] P                  4      t]	R8X  d   ] P                  ! 4        R# R# )é    N)Úparameterized)Úcreate_model_and_input_data)Úget_available_devicesc                   óL  a€ ] tR t^t o]P
                  ! ]! 4        U u. uF  q 3NK  	  up 4      R 4       t]P
                  ! ]! 4        U u. uF  q 3NK  	  up 4      R 4       t]P
                  ! ]! 4        U u. uF  q 3NK  	  up 4      R 4       t	Rt
VtR# u up i u up i u up i )Ú
TestModelsc                óê   € VR 8X  d6   \         P                  P                  4       '       g   V P                  R4       \	        R^
^@^ VR7      w  r#V! V4      pV P                  VP                  R4       R# )ÚcudaúCUDA not availableÚlinear©ÚdeviceN©é
   é    ©Útorchr	   Úis_availableÚskipTestr   ÚassertEqualÚshape©Úselfr   ÚmodelÚ
input_dataÚoutputs   &&   Ú]/home/luca/screen-checking/.venv/lib/python3.14/site-packages/test/test_model_architecture.pyÚtest_toy_linear_modelÚ TestModels.test_toy_linear_model   sc   € ð �VÔ¤E§J¡J×$;Ñ$;×$=Ò$=Ø�M‰MÐ.Ô/ä7Ø�b˜"˜b¨ô
Ñˆñ �zÓ"ˆØ×Ñ˜Ÿ™ xÖ0ó    c                óê   € VR 8X  d6   \         P                  P                  4       '       g   V P                  R4       \	        R^
^@^ VR7      w  r#V! V4      pV P                  VP                  R4       R# )r	   r
   Úln_linear_sigmoidr   Nr   r   r   s   &&   r   Útest_ln_linear_activation_modelÚ*TestModels.test_ln_linear_activation_model   sc   € ð �VÔ¤E§J¡J×$;Ñ$;×$=Ò$=Ø�M‰MÐ.Ô/ä7Ø  R¨°Fô
Ñˆñ �zÓ"ˆØ×Ñ˜Ÿ™ xÖ0r   c                óê   € VR 8X  d6   \         P                  P                  4       '       g   V P                  R4       \	        R^
^@^ VR7      w  r#V! V4      pV P                  VP                  R4       R# )r	   r
   Útransformer_blockr   N)r   é   é@   r   r   s   &&   r   Útest_transformer_blockÚ!TestModels.test_transformer_block)   sc   € ð �VÔ¤E§J¡J×$;Ñ$;×$=Ò$=Ø�M‰MÐ.Ô/ä7Ø  R¨°Fô
Ñˆñ �zÓ"ˆØ×Ñ˜Ÿ™ |Ö4r   © N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   Úexpandr   r   r"   r(   Ú__static_attributes__Ú__classdictcell__)r   Ú__classdict__s   0@r   r   r      s´   ø‡ € Ø×ÒÑ3HÔ3JÓKÑ3J¨›9Ñ3JÑKÓLñ	1ó Mð	1ð ×ÒÑ3HÔ3JÓKÑ3J¨›9Ñ3JÑKÓLñ	1ó Mð	1ð ×ÒÑ3HÔ3JÓKÑ3J¨›9Ñ3JÑKÓLñ	5ó Mö	5ùò3 Lùò Lùò Ls   ŸBÁBÁ9B!r   Ú__main__)Úunittestr   r   Ú#torchao.testing.model_architecturesr   Útorchao.utilsr   ÚTestCaser   r+   Úmainr*   r   r   Ú<module>r9      sA   ðó ã Ý 'å KÝ /ô#5�×"Ñ"ô #5ðL ˆzÔØ‡M‚M†Oñ r   