tico 0.1.0.dev251106__py3-none-any.whl → 0.2.0.dev260122__py3-none-any.whl

{tico-0.1.0.dev251106.dist-info → tico-0.2.0.dev260122.dist-info}/RECORD

@@ -1,4 +1,5 @@
-tico/__init__.py,sha256=zFizlBKVmXPeQuRPHlpsmWOePEoPowkPdrlRwoxPs4k,1883
+tico/__init__.py,sha256=6avfcofxPPv7HMPOr8C-r79A9KoVIhHNZUXhZVulEhM,1861
+tico/_version.py,sha256=98m_MDqJurV2D6AnSpzIp5i8c1hXtGO-yyjGtMnGWbs,32
 tico/pt2_to_circle.py,sha256=gu3MD4Iqc0zMZcCZ2IT8oGbyj21CTSbT3Rgd9s2B_9A,2767
 tico/config/__init__.py,sha256=xZzCXjZ84qE-CsBi-dfaL05bqpQ3stKKfTXhnrJRyVs,142
 tico/config/base.py,sha256=q5xMqGxTUZs4mFqt5c7i_y9U00fYgdMGl9nUqIVMlCo,1248
@@ -14,13 +15,15 @@ tico/passes/cast_clamp_mixed_type_args.py,sha256=m3_HpXLywWmWERfE5lM5PgvjBod7C4B
 tico/passes/cast_mixed_type_args.py,sha256=Wd3sCDKJZwdb8GiMWKljm8X5CLFRd8eCz-dmWks15Hc,7763
 tico/passes/const_prop_pass.py,sha256=hDxGgJNiRjsgOArdaoeAOcOOA-nKBvA1W1zcMZQA5yg,11531
 tico/passes/convert_conv1d_to_conv2d.py,sha256=ktS3h158y9rg1sQiW8BZZbflV_dk_UdjBPQnuiOKyzg,5303
+tico/passes/convert_conv3d_to_conv2d.py,sha256=mv6G-e18PBBJIanaQs4QwH4dK0agaQabQzNLlh3cV4c,22755
 tico/passes/convert_expand_to_slice_cat.py,sha256=Fa6b5pqiQNq-QBiEC0e3WkQYf2UEhMgzSTIt4hlzdjc,5470
 tico/passes/convert_layout_op_to_reshape.py,sha256=sCAFjkmVtiKjvDQSAgnjNBHl3_hWXJZElGDXQiTH-7s,2963
 tico/passes/convert_matmul_to_linear.py,sha256=WATtsHk_GzsU0HYovc3UMyEj8ApF2qLbInAsNlQj0nE,9759
 tico/passes/convert_repeat_to_expand_copy.py,sha256=JbtFTmWyfJS2SSd_higP1IEhQeh7wHdN5dmTbbiFVCs,3237
+tico/passes/convert_sym_size_to_circle_shape.py,sha256=ItWD9eJ-Ia-sFg8HNHrwd0ar058ouLF9FgcTNGdu1sI,3845
 tico/passes/convert_to_relu6.py,sha256=9B6OLyF72tMvD-ugV7aBx6l1szwERufNBUaX34pkZ4c,6445
 tico/passes/decompose_addmm.py,sha256=KjnpZjSuA0uvNmKaTN_EMwobcOi3CAB81buORzTDxro,3979
-tico/passes/decompose_batch_norm.py,sha256=06LAxhSmpTxFZJmUelwB3I_GipNWrLoM7PfM6ZkxOZY,6512
+tico/passes/decompose_batch_norm.py,sha256=fuYX-Z3jrKwYrvgD5pxh3QHjf3uob9DtMl3N7bqJggQ,6610
 tico/passes/decompose_fake_quantize.py,sha256=736srs8SM8K_mLR0WG10LVMMLRkYkBM9OF0k1GCkAW0,5218
 tico/passes/decompose_fake_quantize_tensor_qparams.py,sha256=CalubQ1OYC2l59_TNPOcAnl4VxvameYWIQcy57Z6yjI,13985
 tico/passes/decompose_group_norm.py,sha256=6BqvYtMTPzeIgp8cPA8OFMwEBvb7odcg04IUgwtp7NQ,10120
@@ -32,11 +35,12 @@ tico/passes/fuse_leading_unsqueeze_reshape.py,sha256=88jwTP35yRyXOk9xdO6YW2OEfdK
 tico/passes/fuse_redundant_reshape_to_mean.py,sha256=GhJS1ZKB6Ns4AhwcW3uUQ6q-0N-AzlD32B2EwusUJHg,3761
 tico/passes/legalize_causal_mask_value.py,sha256=0nfUKGd7XSe9Hg5TAi4dUi6Nn6-JRTWCwhULR5AEgqs,4079
 tico/passes/legalize_predefined_layout_operators.py,sha256=3gILn38jzIMDXtMTWpjdROgwmavDC64w115W171encg,18641
+tico/passes/lower_copy.py,sha256=TaeDnztjv9ZfxeOTEUuuFKQNXc676ou1YGNRMbTMDa0,3198
 tico/passes/lower_pow2_to_mul.py,sha256=nfJXa9ZTZMiLg6ownSyvkM4KF2z9tZW34Q3CCWI_vmQ,2402
 tico/passes/lower_to_resize_nearest_neighbor.py,sha256=gbrvTmWSXDPdJ1XJtWGI5mo-uEiauXEG3ELwbKYVPLI,9013
 tico/passes/lower_to_slice.py,sha256=OzlFzK3lBYyYwC3WThsWd94Ob4JINIJF8UaLAtnumzU,7262
 tico/passes/merge_consecutive_cat.py,sha256=ayZNLDA1DFM7Fxxi2Dmk1CujkgUuaVCH1rhQgLrvvOQ,2701
-tico/passes/ops.py,sha256=7IGRnxIJl-nLO4huVk_mgBfD4VGUNQRyeuM8K1L2u1U,2934
+tico/passes/ops.py,sha256=902FF_GjJH9DTXjIL1DWHPqSfkTMrVlWfUDFkTE_LII,3054
 tico/passes/remove_nop.py,sha256=Hf91p_EJAOC6DyWNthash0_UWtEcNc_M7znamQfYQ5Y,2686
 tico/passes/remove_redundant_assert_nodes.py,sha256=rYbTCyuNIXIC-2NreHKBVCuaSUkEQvB_iSRzb26P_EA,1821
 tico/passes/remove_redundant_expand.py,sha256=8yhlMnbog-T9gIK6LKIU0tu0__gfhZzO36g_fJIVVP4,2162
@@ -47,44 +51,26 @@ tico/passes/remove_redundant_to_copy.py,sha256=tKy4XKkO2l33fMxVPQ_iFkUeFvP15kbPv
 tico/passes/restore_linear.py,sha256=xGJdNb-1CrkOKS9BnLbcblkZc6P2vVjKIi-7lRcs7Bk,4111
 tico/passes/segment_index_select.py,sha256=VVCKNLtYRkr9n5lGnlzEuQsQ0WVxEYXGchFrDnB1C40,5189
 tico/quantization/__init__.py,sha256=xYeYEQWeJ6Le7vZVxV208XtukAW68nNQqX9bbjCrBaM,109
-tico/quantization/public_interface.py,sha256=YlE4re0HkkEDcq8IeXhPJUtveLIiDjAlChLvS_-254k,4153
+tico/quantization/public_interface.py,sha256=eZEe7LUs7CxlaM-UjJsgQTtPdDNDlw37WD-9gmhdvUs,3595
 tico/quantization/quantizer.py,sha256=FYNiqUqoH9vz1bda0I6yuKqJi2KdIfLEBd4EgeC-_t4,2357
 tico/quantization/quantizer_registry.py,sha256=MxVE1_hj1p8FjdAqkLzUhdez3Cqc-V25k6XKOcTkei0,2414
 tico/quantization/algorithm/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
+tico/quantization/algorithm/fpi_gptq/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
+tico/quantization/algorithm/fpi_gptq/fpi_gptq.py,sha256=FIqfz0a9LklRs3DH66NmWbobiDLW-4GzzwVVq4iH9FI,9502
+tico/quantization/algorithm/fpi_gptq/quantizer.py,sha256=pXrkm8mfdDvY-mgQh4svgAUaiqY82M1ZhcVRV-aRpSk,6811
 tico/quantization/algorithm/gptq/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
-tico/quantization/algorithm/gptq/gptq.py,sha256=x7wM9_OgOrcs6WmkVCDLn2bF7YuUAR_k6vLG2l593sk,6235
+tico/quantization/algorithm/gptq/gptq.py,sha256=OWWXvvYvt-j3iBTQJT4_1pJpxvG4goUOrDJJCxQrhBc,16035
 tico/quantization/algorithm/gptq/quant.py,sha256=Rl4wAOCmlE0U09BtNCDbccaSNohRHCNLwFi3zCqZfNo,5127
-tico/quantization/algorithm/gptq/quantizer.py,sha256=OvR9sHgosGYofwYcDhye84FBl55cNY7-UlfBt9gXbDY,11734
+tico/quantization/algorithm/gptq/quantizer.py,sha256=dOlaej5vPuNitgDP_cK-gPaoNKkpht7rRfjyyPSiSzc,12189
 tico/quantization/algorithm/gptq/utils.py,sha256=leGKayf-xbSjVwwAGTA5RsxUKrhDiklOQdlsLifjdrs,1811
-tico/quantization/algorithm/pt2e/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
-tico/quantization/algorithm/pt2e/quantizer.py,sha256=9K8SGwxi67DA8Hdwc_25ResJiSGLIMDkNyAwtQu3PGM,2673
-tico/quantization/algorithm/pt2e/utils.py,sha256=U9kf3J-1IJdlmFr5EQRcgWKX7AI8Z-tt_H0edKQ0ctQ,4784
-tico/quantization/algorithm/pt2e/annotation/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
-tico/quantization/algorithm/pt2e/annotation/annotator.py,sha256=_elaNHBM_7IZdMlhqH_BElJpNijV6AHQU-1MeK8KQ9g,7464
-tico/quantization/algorithm/pt2e/annotation/config.py,sha256=8x6YvixIogXh_PwqhKNb3L6lJyBDFQ_OHotonG_hWD0,977
-tico/quantization/algorithm/pt2e/annotation/spec.py,sha256=76cEij6xFNp51ZO5DOeYbRyWonhgHJO8mi47Q43ll8s,1415
-tico/quantization/algorithm/pt2e/annotation/utils.py,sha256=W6_p_EWPDKbvzfqktZfMm1NqiQeeQME5Pr6bzRPqXuo,3148
-tico/quantization/algorithm/pt2e/annotation/op/__init__.py,sha256=IlBNBqXeopMqHRkR-TPiuN_IkwXaSMXlSVO_9vKceB0,834
-tico/quantization/algorithm/pt2e/annotation/op/adaptive_avg_pool2d.py,sha256=ThDTMHFmGp-sFc4YMMTvNwrZtt2R7RNGaaWxVSZV-Yo,2408
-tico/quantization/algorithm/pt2e/annotation/op/add.py,sha256=-glHoBh49e6RM6W1L3raVjtZ_fLhsmOuzHsoBJUFcY8,2153
-tico/quantization/algorithm/pt2e/annotation/op/conv2d.py,sha256=hYE0R6ZrFwp4GABnsFac_F-gPgfG22G56S8fhSAnorM,3631
-tico/quantization/algorithm/pt2e/annotation/op/div.py,sha256=KKh5C30kCpMIpLV94CE7R8yh51K0hb7t4Owa9yeZ4TU,2135
-tico/quantization/algorithm/pt2e/annotation/op/linear.py,sha256=fOr8Ow0Y61xVQg4IZYSmYFG7CT28UAs1dXP0YQ1498M,3430
-tico/quantization/algorithm/pt2e/annotation/op/mean.py,sha256=ZLfCh3wFftV4gBYe5FeCo0tC8JOZ-5Hde64Sw2gfU_8,1959
-tico/quantization/algorithm/pt2e/annotation/op/mul.py,sha256=_R9dngri5wLsxqeLLeNYaJnTcuPQXtXBkfl23to_0Zs,2150
-tico/quantization/algorithm/pt2e/annotation/op/relu6.py,sha256=1Us4pBf2BSD3ICCWZPNGiBmaiRn_vnKopV_To7bpL7A,1956
-tico/quantization/algorithm/pt2e/annotation/op/rsqrt.py,sha256=Y2Z0mB-8Gk9tkvR4wNGnY0sM6q19YidZd5idbN8ZXTo,1966
-tico/quantization/algorithm/pt2e/annotation/op/sub.py,sha256=u4hg47dVCOCUqJbZV0GFZ5EKDUNu7NV1TMhxUnW_1vA,2135
-tico/quantization/algorithm/pt2e/transformation/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
-tico/quantization/algorithm/pt2e/transformation/convert_scalars_to_attrs.py,sha256=Idtoya2RcGKlgUJgC9WqNz0jH3gf6ViuPmsD9ySHbls,2253
 tico/quantization/algorithm/smoothquant/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/quantization/algorithm/smoothquant/observer.py,sha256=OWBKQ3ox6PqeqgevxOjpXvb7uApoqE4YbUBelGhVSN8,3435
 tico/quantization/algorithm/smoothquant/quantizer.py,sha256=pvf6HwW7VzyNFhfEDGwG-YPdPaEoGQfo4nfaeS9Qg_E,3686
 tico/quantization/algorithm/smoothquant/smooth_quant.py,sha256=fxCy4m-BsSjraciSVPFlPhgsOT46RjrOgczQGb7B9TA,11561
 tico/quantization/config/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/quantization/config/base.py,sha256=xg_HCDSuMgYvMd6ENZe4Sm2SYJgMaCBj4cmqaz_lhAs,816
-tico/quantization/config/gptq.py,sha256=O3NEPYMJdgMJQB--blw3WI8FGbK9nDlSqSo2ZHvNwb8,960
-tico/quantization/config/pt2e.py,sha256=vSfULljHEnypadUyo-zjVoPSbP8Y2eDzSD_kRTcv6bk,837
+tico/quantization/config/fpi_gptq.py,sha256=s3IqGG_9KGFvSI4vUOdBlsqfHlJ-0OpEKoBzm2WLdGY,1040
+tico/quantization/config/gptq.py,sha256=H9aCm7QsP0_Gwc_aC3ZkFjitbDMe87WWj9gs5L7urGQ,1700
 tico/quantization/config/ptq.py,sha256=zbLQbuiEpO-qlDgyUYTZ3hkVxr3boq5TX0n0QTBHic4,4540
 tico/quantization/config/smoothquant.py,sha256=ntrqjYf6EbO4AE2IA5zn1A_f0AQgU6UqTtVkw6IiUsw,1401
 tico/quantization/evaluation/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
@@ -107,7 +93,7 @@ tico/quantization/wrapq/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9d
 tico/quantization/wrapq/dtypes.py,sha256=xfCBtq6mQmUYRwsoFgII6gvRl1raQi0Inj9pznDuKwQ,2236
 tico/quantization/wrapq/mode.py,sha256=lT-T8vIv8YWcwrjT7xXVhOw1g7aoAdh_3PWB-ptPKaI,1052
 tico/quantization/wrapq/qscheme.py,sha256=uwhv7bCxOOXB3I-IKlRyr_u4eXOq48uIqGy4TLDqGxY,1301
-tico/quantization/wrapq/quantizer.py,sha256=J1nH5FfJ_sKOAKmYjHAZ9zgnitPJ7fcOhO09E2CBIbw,6577
+tico/quantization/wrapq/quantizer.py,sha256=5RtggN1lTnA1a7-TEvcq2KVEPhzv4Ec2idQHJDgSxBI,6625
 tico/quantization/wrapq/examples/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/quantization/wrapq/examples/compare_ppl.py,sha256=iJJEWC7IJpfBHwX0MEiKXBTbxqLHsNQ3Bq9Hh0R57lg,7737
 tico/quantization/wrapq/examples/debug_quant_outputs.py,sha256=kVp_nf9tr4wM5w18ILtYxVueY5bjYl_Mb4v0Em_M3Aw,7665
@@ -129,7 +115,7 @@ tico/quantization/wrapq/utils/metrics.py,sha256=ZnEQOd9fzDDxdXl32PFl3jMQv5ycz9nF
 tico/quantization/wrapq/utils/reduce_utils.py,sha256=3kWawLB91EcvvHlCrNqqfZF7tpgr22htBSA049mKw_4,973
 tico/quantization/wrapq/wrappers/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/quantization/wrapq/wrappers/ptq_wrapper.py,sha256=6zcVZ-vVhPCvFHQw6UlN7iizElrIHNkpAraeMaA0DDU,2388
-tico/quantization/wrapq/wrappers/quant_elementwise.py,sha256=trchhUknmZTcoCwVA62uzBP_mWuCjjuZjF0jb7TZpfA,3550
+tico/quantization/wrapq/wrappers/quant_elementwise.py,sha256=2mpDljmROSIG3DI4TpNuy6gVEZ294aT1rKww-ZSI96o,4880
 tico/quantization/wrapq/wrappers/quant_module_base.py,sha256=SgyUlFYxDx39CAvcN2q4lsTedbEVPmetIigrllmvvD4,5915
 tico/quantization/wrapq/wrappers/registry.py,sha256=QJcOD9gEGB_DJowdTTqemcRDcYxQa4tHv2CDFgZDnA0,5168
 tico/quantization/wrapq/wrappers/fairseq/__init__.py,sha256=K4R7rbxHosx9LBLk2WKlL8gFuZTYTws41TW47AsSUPM,149
@@ -162,19 +148,20 @@ tico/serialize/operators/op_alias_copy.py,sha256=Xu9OiILbGf8oddh8yTqovvLfgVs8XYV
 tico/serialize/operators/op_any.py,sha256=wrTXFQ1TWl-2ET2NGXAXI1dzfDDJsYtA71pyj2numPE,4968
 tico/serialize/operators/op_arange_start_step.py,sha256=0T5lWwh3TfsFStmVv0v5qG03KENRDBmMix08RXQ4D-U,2132
 tico/serialize/operators/op_argmax.py,sha256=ARyGHlmWVmzwCct93V5x1-VyKqhxMOvV8GuM8yQWXdo,2290
+tico/serialize/operators/op_attention.py,sha256=h9kEaW7w5tDPqPmWb3w5WbjhlCwQGEinhr-NaQ3iXi8,2090
 tico/serialize/operators/op_avg_pool2d.py,sha256=eZl232FqYQsn2jUN_XjombHq_lzp2hf_AKwCLbZBxh8,7720
 tico/serialize/operators/op_bmm.py,sha256=AELjHC9ISFPIzEEl5Kr1s4GSNLZElwZmVZJWkEyCEoA,2189
 tico/serialize/operators/op_cat.py,sha256=XDYOh0XAyrM0TlxVm6Sa0OFFGrKk7aSDcGXC-hYX4gs,2204
+tico/serialize/operators/op_circle_shape.py,sha256=6rvFvz7zw0hqW8jvUoBkOBW54nMmsE9KOmLtrnQaxhM,2220
 tico/serialize/operators/op_clamp.py,sha256=RRQVrzayDfN3PioCVJqa_yYOtcYwb5HHwkMe4E_YPmE,4408
 tico/serialize/operators/op_clone.py,sha256=vzDYJ8TS3tc2BAyd_z8nt5VqT1inpymSseMEhd9dva0,2394
 tico/serialize/operators/op_constant_pad_nd.py,sha256=nGWqYWNbj2E9ChQuoHsN-d8AO7UyVexnPil7qTqWZp8,3444
 tico/serialize/operators/op_conv2d.py,sha256=1_vouWXaF51gDLYg8z5Zlup0Tecq_ggAzvguiHzFffw,6828
-tico/serialize/operators/op_copy.py,sha256=boXHfl0bcvdBVl0tpzPMA_KBonh80vVqv61N3H5-PRU,6941
 tico/serialize/operators/op_cos.py,sha256=N12bNyuTQIxRnD0eHRPdFVzRQPMy1NFM4iM8oQ4lYzw,2034
 tico/serialize/operators/op_cumsum.py,sha256=px9ZGUDDsdWjrql8Z1FdXfF-7CJhditxyNz5QRZbLiM,3948
 tico/serialize/operators/op_depthwise_conv2d.py,sha256=U6_nX2V31Evm-HLN9b3RKIVg-m8jyD-Nw1GdePUPPjY,7284
-tico/serialize/operators/op_dequantize_per_channel.py,sha256=aPcVxjdgvfSFoLnv9NL-RxO5vZYj8ulqriMP5LHIWs0,3133
-tico/serialize/operators/op_dequantize_per_tensor.py,sha256=u9aK_Xle9rDN0EHLE0YrCTlXY4Q53Ch9Di4qmx7ynps,2304
+tico/serialize/operators/op_dequantize_per_channel.py,sha256=ugXpZQl0qEGJ2mgwzD_T5WoZwaVvXgs4qCgmKlpF9zk,3214
+tico/serialize/operators/op_dequantize_per_tensor.py,sha256=Rqz7KQY0pQ_QffLn4gxiV2iJY4asRs6BxvLm38hxciY,2384
 tico/serialize/operators/op_div.py,sha256=WjeM2Ux7TyGlSNx2aVC783JvcL0xnY6FBYo1Q_kdb5Q,2201
 tico/serialize/operators/op_embedding.py,sha256=OL5E5kIUbVPd9ihvBh8CNxGPj7GPGA-up2VRrYYlqeY,2262
 tico/serialize/operators/op_eq.py,sha256=g17_K6IkWvnop_LaiUJzcGPUSFElz6UUrf7T0bor5Bo,2133
@@ -231,15 +218,17 @@ tico/serialize/operators/op_sub.py,sha256=yZskQJF0ylXVk02Uid8djPNIWDJ-0uHJar4UYh
 tico/serialize/operators/op_sum.py,sha256=B5aSwQMhyoBe2JYdE5nVQ3QeVDSzL-yuZZujsG08OdQ,2294
 tico/serialize/operators/op_tanh.py,sha256=rs7FsbQeUQ7Ak8RoQV9ymNGXHXRObojfY_SiqJiyqdA,1846
 tico/serialize/operators/op_to_copy.py,sha256=a8T0uPMavMO_md1a-4_0dlvDHyZS_xew0qB6xjf69rI,3934
-tico/serialize/operators/op_transpose_conv.py,sha256=9NLnWpitfQzSDF-iAgw2fBA3YHL5y2Y8DQipeo8OvYA,5826
+tico/serialize/operators/op_transpose_conv.py,sha256=N5AuAUuh9BA4M2Y2-ZSkV1q82P0j7yYpB6g3qiaDmLk,6542
 tico/serialize/operators/op_unsqueeze.py,sha256=ZHhfVXSWEiwb2VDYX5uhxbGQyzZjKT7CrbBpVGxVHBU,2310
 tico/serialize/operators/op_view.py,sha256=xxE-GvTJ1UpcHst5KXYz3qKY-eJQvXKKrSZiA2O7E40,2593
 tico/serialize/operators/op_where.py,sha256=doE81GSwygrPBm3JIfN9w7kKXxeIYKxgk0eoY22QIcg,2845
 tico/serialize/operators/utils.py,sha256=lXGpEJW1h8U_-gfc6EWjvvSiq3yJ9P-v1v3EMRT_pSk,2954
 tico/serialize/operators/adapters/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/serialize/operators/adapters/llama_rmsnorm.py,sha256=6t3dhfNpR03eIjsmhymF2JKd6lCf7PvInqMf77c_BOE,1139
+tico/serialize/operators/adapters/onert/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
+tico/serialize/operators/adapters/onert/llama_attention.py,sha256=jPA6n1lwln1pf8PHkqpNa71W8BQaC6yWwP8xbcVw77w,1827
 tico/utils/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
-tico/utils/convert.py,sha256=wdamhQQPyiipL9vb_E_txDR9kmUyXmevYuYXcTeTAWE,13752
+tico/utils/convert.py,sha256=ejTuKlTw2fnngcPw-3gLcrmy4nTu2fn-FiwPKeofgNE,14405
 tico/utils/define.py,sha256=Ypgp7YffM4pgPl4Zh6TmogSn1OxGBMRw_e09qYGflZk,1467
 tico/utils/diff_graph.py,sha256=_eDGGPDPYQD4b--MXX0DLoVgSt_wLfNPt47UlolLLR4,5272
 tico/utils/dtype.py,sha256=L5Qb7qgbt0eQ5frUTvHYrRtTJb1dg4-JNEopcxCNg1U,1389
@@ -248,24 +237,24 @@ tico/utils/graph.py,sha256=jD6m58m5JmN9mPfaROA9CW3406iJxmnukke00AuwRqI,9131
 tico/utils/installed_packages.py,sha256=J0FTwnkCGs0MxRWoCMYAqiwH7Z0GWFDLV--x-IndSp4,1017
 tico/utils/logging.py,sha256=IlbBWscsaHidI0dNqro1HEXAbIcbkR3BD5ukLy2m95k,1286
 tico/utils/model.py,sha256=pPOIjD0qjQirLibiRxxfjOR6efimOcDAd9R-74eus-k,1282
-tico/utils/padding.py,sha256=qKke-dJeeLHiRaePjDS66txrGyiYuipLVQeqLYad8uk,3349
+tico/utils/padding.py,sha256=Yz9D1VaSYdxWVCgPCzkH1Ip1UIQoUSIxPJUf2IEd8pE,3656
 tico/utils/passes.py,sha256=kGmDe__5cPaO6i5EDAoXSVe6yXEoX9hAny4ROb3ZEmQ,2409
 tico/utils/pytree_utils.py,sha256=jrk3N6X6LiUnBCX_gM1K9nywbVAJBVnszlTAgeIeDUc,5219
-tico/utils/record_input.py,sha256=QN-8D71G_WAX3QQQ5CIwbEfFJZTQ3CvL4wCMiVddua4,3894
-tico/utils/register_custom_op.py,sha256=895SKZeXQzolK-mPG38cQC37Be76xUV_Ujw1k1ts9_w,28218
+tico/utils/record_input.py,sha256=qPJ7RljVfPy-y9cb9AwP-8OmIiYLxCzTzbKJWQB5slk,3914
+tico/utils/register_custom_op.py,sha256=NKzHIALhWX0DQqjFAfTPWTnO7I1WQClrGFCPh8J1e9o,30115
 tico/utils/serialize.py,sha256=mEuusEzi82WFsz3AkowgWwxSLeo50JDxyOj6yYDQhEI,1914
 tico/utils/signature.py,sha256=3OOwyVJzfcGcgC0LiVmOcUIzfqSk27qoNHhkoCI7zPY,9530
 tico/utils/torch_compat.py,sha256=oc6PztVsXdHcQ3iaVR90wLLxrGaj6zFHWZ8K9rRS6q8,1795
 tico/utils/trace_decorators.py,sha256=ddLIiKQfSaQrxgF1kNpwjFTQnXENzeSfcr1kuAW4jGI,3221
 tico/utils/utils.py,sha256=aySftYnNTsqVAMcGs_3uX3-hz577a2cj4p1aVV-1XeQ,12747
-tico/utils/validate_args_kwargs.py,sha256=RhBOHShi7DRHpCV_j4UcACk6wz4b1SUTWKj494_6zCQ,27439
+tico/utils/validate_args_kwargs.py,sha256=qq_VUVSepl-Lmt-jdZuFwcqxbXnxu2_CL2TLyj39OSU,29069
 tico/utils/mx/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/utils/mx/elemwise_ops.py,sha256=V6glyAHsVR1joqpsgnNytatCD_ew92xNWZ19UFDoMTA,10281
 tico/utils/mx/formats.py,sha256=uzNWyu-1onUlwQfX5cZ6fZSUfHMRqorper7_T1k3jfk,3404
 tico/utils/mx/mx_ops.py,sha256=RcfUTYVi-wilGB2sC35OeARdwDqnixv7dG5iyZ-fQT8,8555
-tico-0.1.0.dev251106.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
-tico-0.1.0.dev251106.dist-info/METADATA,sha256=cH9ZYH9ysDRREwJcSjMV1VkLFpNjoOSxPBChSzab-A0,9730
-tico-0.1.0.dev251106.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tico-0.1.0.dev251106.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
-tico-0.1.0.dev251106.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
-tico-0.1.0.dev251106.dist-info/RECORD,,
+tico-0.2.0.dev260122.dist-info/licenses/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
+tico-0.2.0.dev260122.dist-info/METADATA,sha256=xP5llEHZwROmzeBKNy2sYOEXpp6OVYM-_tiJwambEO4,24279
+tico-0.2.0.dev260122.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+tico-0.2.0.dev260122.dist-info/entry_points.txt,sha256=y6XCEOgx67ShUIDNeFluVkrQDYx1Sa59pN-lRCSmulk,58
+tico-0.2.0.dev260122.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
+tico-0.2.0.dev260122.dist-info/RECORD,,

{tico-0.1.0.dev251106.dist-info → tico-0.2.0.dev260122.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.1)
+Generator: setuptools (80.10.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{tico-0.1.0.dev251106.dist-info → tico-0.2.0.dev260122.dist-info}/entry_points.txt

@@ -1,3 +1,2 @@
 [console_scripts]
 pt2-to-circle = tico.pt2_to_circle:main
-

tico/quantization/algorithm/pt2e/annotation/annotator.py

@@ -1,208 +0,0 @@
-# Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from __future__ import annotations
-
-import functools
-from typing import Any, Callable, Dict, Optional, TYPE_CHECKING
-
-if TYPE_CHECKING:
-    import torch.fx
-    from torch.ao.quantization.qconfig import _ObserverOrFakeQuantizeConstructor
-import torch
-from torch.ao.quantization.observer import MinMaxObserver, PerChannelMinMaxObserver
-from torch.ao.quantization.quantizer import QuantizationSpec, Quantizer
-from torch.ao.quantization.quantizer.utils import _get_module_name_filter
-
-from tico.quantization.algorithm.pt2e.annotation.op import *
-import tico.quantization.algorithm.pt2e.annotation.spec as annot_spec
-import tico.quantization.algorithm.pt2e.annotation.utils as annot_utils
-import tico.quantization.algorithm.pt2e.utils as quant_utils
-from tico.quantization.algorithm.pt2e.annotation.config import QuantizationConfig
-from tico.quantization.algorithm.pt2e.transformation.convert_scalars_to_attrs import (
-    convert_scalars_to_attrs,
-)
-
-
-class PT2EAnnotator(Quantizer):
-    """
-    The class annotates quantization configurations on each nodes.
-
-    Observers would be attached according to those configurations in
-     'torch.prepare_pt2e'.
-    """
-
-    def __init__(self):
-        super().__init__()
-        self.global_config: Optional[QuantizationConfig] = None
-        self.operator_type_config: Dict[
-            torch._ops.OpOverloadPacket, QuantizationConfig
-        ] = {}
-        self.module_type_config: Dict[Callable, QuantizationConfig] = {}
-        self.module_name_config: Dict[str, QuantizationConfig] = {}
-
-    def set_global(self, quantization_config: QuantizationConfig) -> PT2EAnnotator:
-        """
-        Set quantization config globally.
-        """
-        assert quantization_config is not None
-        self.global_config = quantization_config
-        return self
-
-    def set_operator_type(
-        self,
-        operator_type: torch._ops.OpOverloadPacket,
-        quantization_config: QuantizationConfig,
-    ) -> PT2EAnnotator:
-        """
-        Set quantization config for given operator type.
-        """
-        assert quantization_config is not None
-        self.operator_type_config[operator_type] = quantization_config
-        return self
-
-    def set_module_type(
-        self, module_type: Callable, quantization_config: QuantizationConfig
-    ):
-        """
-        Set quantization config for given module type.
-
-        For example, let's say quantizer.set_module_type(nn.Linear).
-        It will quantize all 'nn.Linear' modules with the `quantization_config`.
-        """
-        assert quantization_config is not None
-        self.module_type_config[module_type] = quantization_config
-        return self
-
-    def set_module_name(
-        self, module_name: str, quantization_config: QuantizationConfig
-    ):
-        """
-        Set quantization config for given module name.
-
-        For example, let's say quantizer.set_module_name("blocks.sub").
-        It will quantize all nodes that come from a module whose name is "blocks.sub"
-         with the `quantization_config`.
-        """
-        assert quantization_config is not None
-        self.module_name_config[module_name] = quantization_config
-        return self
-
-    def transform_for_annotation(
-        self, model: torch.fx.GraphModule
-    ) -> torch.fx.GraphModule:
-        """Allows for user defined transforms to run before annotating the graph."""
-        model = convert_scalars_to_attrs(model)
-        return model
-
-    def annotate(self, model: torch.fx.GraphModule) -> torch.fx.GraphModule:
-        model = self._annotate_for_quantization(model)
-        annot_utils.propagate_annotation_forward(model)
-        return model
-
-    def validate(self, model: torch.fx.GraphModule) -> None:
-        # TODO Consider this method.
-        pass
-
-    def _annotate_by_config_and_filter(
-        self,
-        model: torch.fx.GraphModule,
-        quantization_config: Optional[QuantizationConfig],
-        filter_fn: Optional[Callable[[torch.fx.Node], bool]] = None,
-    ) -> torch.fx.GraphModule:
-        assert quantization_config is not None
-
-        for node in model.graph.nodes:
-            if node.target not in annot_spec.OP_TO_ANNOTATOR:
-                continue
-            annot_spec.OP_TO_ANNOTATOR[node.target](
-                model, node, quantization_config, filter_fn
-            )
-        return model
-
-    def _annotate_for_quantization(
-        self, model: torch.fx.GraphModule
-    ) -> torch.fx.GraphModule:
-        # Annotate according to the given module names.
-        module_name_list = list(self.module_name_config.keys())
-        for module_name, config in self.module_name_config.items():
-            self._annotate_by_config_and_filter(
-                model, config, _get_module_name_filter(module_name)
-            )
-
-        # Annotate according to the given module types.
-        tp_list = list(self.module_type_config.keys())
-        for module_type, config in self.module_type_config.items():
-            self._annotate_by_config_and_filter(
-                model, config, quant_utils.get_module_type_filter(module_type)
-            )
-
-        # TODO Annotate according to the given operator types.
-
-        self._annotate_by_config_and_filter(
-            model,
-            self.global_config,
-            quant_utils.get_not_module_type_or_name_filter(tp_list, module_name_list),
-        )
-        return model
-
-
-@functools.lru_cache
-def get_asymmetric_quantization_config(
-    weight_is_per_channel: bool = True,
-    act_qmin: int = 0,
-    act_qmax: int = 255,
-    weight_qmin: int = 0,
-    weight_qmax: int = 255,
-) -> QuantizationConfig:
-    # activation
-    act_extra_args: Dict[str, Any] = {"eps": 2**-12}
-    act_observer = MinMaxObserver
-    act_qspec = QuantizationSpec(
-        dtype=torch.uint8,
-        quant_min=act_qmin,
-        quant_max=act_qmax,
-        qscheme=torch.per_tensor_affine,
-        is_dynamic=False,
-        observer_or_fake_quant_ctr=act_observer.with_args(
-            **act_extra_args,
-        ),
-    )
-    # weight
-    weight_extra_args: Dict[str, Any] = {"eps": 2**-12}
-    weight_qscheme = (
-        torch.per_channel_affine if weight_is_per_channel else torch.per_tensor_affine
-    )
-    weight_observer: _ObserverOrFakeQuantizeConstructor = (
-        PerChannelMinMaxObserver if weight_is_per_channel else MinMaxObserver
-    )
-    weight_qspec = QuantizationSpec(
-        dtype=torch.uint8,
-        quant_min=weight_qmin,
-        quant_max=weight_qmax,
-        qscheme=weight_qscheme,
-        ch_axis=0,
-        is_dynamic=False,
-        observer_or_fake_quant_ctr=weight_observer.with_args(**weight_extra_args),
-    )
-
-    # Set bias qspec in each annotation functions.
-    bias_qspec = None
-    quantization_config = QuantizationConfig(
-        act_qspec,
-        act_qspec,
-        weight_qspec,
-        bias_qspec,
-    )
-    return quantization_config

tico/quantization/algorithm/pt2e/annotation/config.py

@@ -1,26 +0,0 @@
-# Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from dataclasses import dataclass
-from typing import Optional
-
-from torch.ao.quantization.quantizer import QuantizationSpec
-
-
-@dataclass(eq=True, frozen=True)
-class QuantizationConfig:
-    input_activation: Optional[QuantizationSpec]
-    output_activation: Optional[QuantizationSpec]
-    weight: Optional[QuantizationSpec]
-    bias: Optional[QuantizationSpec]

tico/quantization/algorithm/pt2e/annotation/op/__init__.py

@@ -1,21 +0,0 @@
-# Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import glob
-from os.path import basename, dirname, isfile, join
-
-modules = glob.glob(join(dirname(__file__), "*.py"))
-__all__ = [
-    basename(f)[:-3] for f in modules if isfile(f) and not f.endswith("__init__.py")
-]

tico/quantization/algorithm/pt2e/annotation/op/adaptive_avg_pool2d.py

@@ -1,63 +0,0 @@
-# Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from typing import Callable, Optional, TYPE_CHECKING
-
-if TYPE_CHECKING:
-    import torch.fx
-import torch
-from torch.ao.quantization.quantizer import SharedQuantizationSpec
-
-import tico.quantization.algorithm.pt2e.annotation.spec as annot_spec
-import tico.quantization.algorithm.pt2e.annotation.utils as annot_utils
-import tico.quantization.algorithm.pt2e.utils as quant_utils
-from tico.quantization.algorithm.pt2e.annotation.config import QuantizationConfig
-from tico.utils.validate_args_kwargs import AdaptiveAvgPool2dArgs
-
-
-@annot_spec.register_annotator([torch.ops.aten.adaptive_avg_pool2d.default])
-def _annotate_adaptive_avg_pool2d(
-    gm: torch.fx.GraphModule,
-    node: torch.fx.Node,
-    quantization_config: Optional[QuantizationConfig],
-    filter_fn: Optional[Callable[[torch.fx.Node], bool]] = None,
-):
-    if (
-        node.op != "call_function"
-        or node.target != torch.ops.aten.adaptive_avg_pool2d.default
-    ):
-        return
-    if filter_fn and not filter_fn(node):
-        return
-    if quant_utils.is_annotated(node):
-        return
-
-    args = AdaptiveAvgPool2dArgs(*node.args)  # type: ignore[arg-type]
-    input = args.input
-
-    assert isinstance(input, torch.fx.Node)
-    if (
-        "quantization_annotation" not in input.meta
-        or not input.meta["quantization_annotation"]._annotated
-        or input.meta["quantization_annotation"].output_qspec is None
-    ):
-        input_act_qspec = quant_utils.get_input_act_qspec(quantization_config)
-    else:
-        input_act_qspec = SharedQuantizationSpec(input)
-    annot_utils.annotate_input_qspec_map(node, input, input_act_qspec)
-
-    output_act_qspec = SharedQuantizationSpec((input, node))
-    annot_utils.annotate_output_qspec(node, output_act_qspec)
-
-    annot_utils.mark_nodes_as_annotated(node)

tico/quantization/algorithm/pt2e/annotation/op/add.py

@@ -1,55 +0,0 @@
-# Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from typing import Callable, Optional, TYPE_CHECKING
-
-if TYPE_CHECKING:
-    import torch.fx
-import torch
-
-import tico.quantization.algorithm.pt2e.annotation.spec as annot_spec
-import tico.quantization.algorithm.pt2e.annotation.utils as annot_utils
-import tico.quantization.algorithm.pt2e.utils as quant_utils
-from tico.quantization.algorithm.pt2e.annotation.config import QuantizationConfig
-from tico.utils.validate_args_kwargs import AddTensorArgs
-
-
-@annot_spec.register_annotator([torch.ops.aten.add.Tensor])
-def _annotate_add(
-    gm: torch.fx.GraphModule,
-    node: torch.fx.Node,
-    quantization_config: Optional[QuantizationConfig],
-    filter_fn: Optional[Callable[[torch.fx.Node], bool]] = None,
-):
-    if node.op != "call_function" or node.target != torch.ops.aten.add.Tensor:
-        return
-    if filter_fn and not filter_fn(node):
-        return
-    if quant_utils.is_annotated(node):
-        return
-
-    args = AddTensorArgs(*node.args, **node.kwargs)  # type: ignore[arg-type]
-    input_ = args.input
-    other = args.other
-
-    input_act_qspec = quant_utils.get_input_act_qspec(quantization_config)
-    if isinstance(input_, torch.fx.Node):
-        annot_utils.annotate_input_qspec_map(node, input_, input_act_qspec)
-    if isinstance(other, torch.fx.Node):
-        annot_utils.annotate_input_qspec_map(node, other, input_act_qspec)
-
-    output_act_qspec = quant_utils.get_output_act_qspec(quantization_config)
-    annot_utils.annotate_output_qspec(node, output_act_qspec)
-
-    annot_utils.mark_nodes_as_annotated(node)