o ‡ZŽh2 ã@s<ddlZddlmZmZmZmZmZmZmZm Z m Z mZmZm Z ddlmZmZgd¢ZeejejejejdZeejejdZeejejejejdZeejejejejdd Zeejejejejdd ZeejejejejdZeejejejd Zeejejejd Zdefdd „Zdefdd„Z dd„Z!dd„Z"dS)éNé)Ú_get_binary_op_configsÚ_get_bn_configsÚ_get_cat_configÚ_get_conv_configsÚ_get_default_op_configsÚ_get_embedding_op_configsÚ_get_fixed_qparams_op_configsÚ_get_linear_configsÚ_get_ln_configsÚ_get_rnn_op_configsÚ_get_share_qparams_op_configsÚ_get_tensor_info_op_configs)Ú BackendConfigÚDTypeConfig)Ú*get_test_only_legacy_native_backend_configÚdefault_op_quint8_dtype_configÚdefault_op_fp16_dtype_configÚ!default_dynamic_int8_dtype_configÚ$default_dynamic_float16_dtype_configÚ%input_output_only_quint8_dtype_configÚweight_only_quint8_dtype_configÚ!weight_only_quint4x2_dtype_configÚget_native_backend_configÚget_native_backend_config_dictÚ/get_test_only_legacy_native_backend_config_dict)Úinput_dtypeÚoutput_dtypeÚweight_dtypeÚ bias_dtype)rrT)rrrrZ is_dynamic)rrrÚreturnc CsÌtg}ttttg}ttg}tg}ttg}ttg}tg}ttg}ttg}tg} tdƒ t |ƒ¡ t|ƒ¡ t|ƒ¡ t|ƒ¡ t|ƒ¡ t|ƒ¡ t|ƒ¡ t|ƒ¡ t|ƒ¡ t| ƒ¡ t|ƒ¡ t|ƒ¡S)zr Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) with various additional fp16 ops. Z_native_and_fp16)Úweighted_op_quint8_dtype_configrrrrrrrrÚset_backend_pattern_configsrr rÚset_backend_pattern_configrrr r rrrrr© Zconv_dtype_configsZlinear_dtype_configsZbinary_op_dtype_configsZdefault_op_dtype_configsZfixed_qparams_op_dtype_configsZshare_qparams_op_dtype_configsZtensor_info_op_dtype_configsZrnn_op_dtype_configsZembedding_op_dtype_configsZlayer_norm_op_dtype_configs©r%úZ/var/www/auris/lib/python3.10/site-packages/torch/ao/quantization/backend_config/native.pyrls^üþþþÿþþ ÿÿÿ ÿírc CsÄtg}tttg}tg}tg}tg}tg}tg}ttg}ttg}tg} tdƒ t |ƒ¡ t |ƒ¡ t|ƒ¡ t |ƒ¡ t|ƒ¡ t|ƒ¡ t|ƒ¡ t|ƒ¡ t|ƒ¡ t| ƒ¡ t|ƒ¡ t|ƒ¡S)zQ Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack). Únative)r!rrrrrrrr"rr rr#rrr r rrrrrr$r%r%r&r©sNýþþ ÿÿÿ ÿírcCó tƒ ¡S)zd Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) in dictionary form. )rÚto_dictr%r%r%r&rÛs rcCr()z‰ Return the `BackendConfig` for PyTorch Native backend (fbgemm/qnnpack) with various additional fp16 ops in dictionary form. )rr)r%r%r%r&râs r)#ZtorchZ_common_operator_config_utilsrrrrrrr r rrr rZbackend_configrrÚ__all__Zquint8Zqint8Úfloatr!rZfloat16rrrrrZquint4x2rrrrrr%r%r%r&Úsl8üþüøøüýý=2