§
    v-j·  ã                  óT   — d dl mZ h d£Zh d£Zeez  Zh d£Zh d£ZeZeZdd„Z	dd	„Z
d
S )é    )Úannotations>
   ÚmulÚconv2dÚeinsumÚmatmulÚ	linear_v2Ú	matmul_v2Ú
flash_attnÚfused_gemm_epilogueÚmax_pool2d_with_indexÚfused_rotary_position_embedding>   Úfused_attentionÚfused_feedforwardÚ fake_quantize_dequantize_abs_maxÚ/fake_quantize_dequantize_moving_average_abs_max>-   ÚexpÚlogÚpowÚtanÚacosÚasinÚcoshÚdistÚlog2ÚmeanÚsinhÚatanhÚexpm1Úlog10ÚpnormÚrsqrtÚcumsumÚerfinvÚrenormÚsquareÚcumprodÚsoftmaxÚsoftminÚlog_lossÚnll_lossÚsoftplusÚ
group_normÚ
huber_lossÚ
layer_normÚ
reciprocalÚ
reduce_sumÚlog_softmaxÚreduce_meanÚreduce_prodÚtanh_shrinkÚcross_entropyÚhsigmoid_lossÚcross_entropy2Úfrobenius_normÚtriplet_margin_lossÚmargin_cross_entropyÚsoftmax_with_cross_entropyÚc_softmax_with_cross_entropyÚ!sigmoid_cross_entropy_with_logitsÚ(c_softmax_with_multi_label_cross_entropy>   ÚscatterÚlookup_tableÚlookup_table_v2Úlinear_interp_v2Úbicubic_interp_v2Únearest_interp_v2Úbilinear_interp_v2Útrilinear_interp_v2Úreturnúdict[str, dict[str, set[str]]]c                 ó\   — t           t           t           dœt          t          t          dœdœ} | S ©N)ÚODÚO1ÚO2)Úfloat16Úbfloat16)ÚFP16_WHITE_LISTÚBF16_WHITE_LIST)Ú
white_lists    úT/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/amp/amp_lists.pyrR   rR   n   s@   € õ "Ý!Ý!ð
ð 
õ "Ý!Ý!ð
ð 
ðð €Jð Ðó    c                 ó˜   — t          ¦   «         t          t          z  t          dœt          ¦   «         t          t          z  t          dœdœ} | S rJ   )ÚsetÚFP16_BLACK_LISTÚEXTRA_BLACK_LISTÚBF16_BLACK_LIST)Ú
black_lists    rS   rZ   rZ   ~   sT   € õ ‘%”%Ý!Õ$4Ñ4Ý"ð
ð 
õ ‘%”%Ý!Õ$4Ñ4Ý"ð
ð 
ðð €Jð ÐrT   N)rG   rH   )Ú
__future__r   Ú
WHITE_LISTÚONLY_FP16_WHITE_LISTrP   rW   rX   rQ   rY   rR   rZ   © rT   rS   ú<module>r_      s«   ðð$ #Ð "Ð "Ð "Ð "Ð "ðð ð €
ðð ð Ð ð Ð3Ñ3€ð.ð .ð .€ðb	ð 	ð 	Ð ð €Ø!€ðð ð ð ð ð ð ð ð ð rT   