PivddlZddlmZddZdedefd Zded efd Zded efd Zd ed edefdZdedefdZdS)N)TensorTcTg}||rdndz }d||z zdz }t|D]}t|rd||z|z zdznd||z|z dzzdz}tjdd|}|dd|ddzdz } |d|dz |zz| zz }|r%|d|dz |zz | zz }|dkr|tjdd|dz}|dd|ddzdz } |d|dz |zz| zz }|r%|d|dz |zz | zz }|d|d t |d|zksJ||S) a+ Creates the dynamic quantiztion map. The dynamic data type is made up of a dynamic exponent and fraction. As the exponent increase from 0 to -7 the number of bits available for the fraction shrinks. This is a generalization of the dynamic type where a certain number of the bits and be reserved for the linear quantization region (the fraction). n determines the maximum number of exponent bits. For more details see (8-Bit Approximations for Parallelism in Deep Learning)[https://arxiv.org/abs/1511.04561] g?Ng@ rg?)rangeinttorchlinspacetolistappendlensort) signedmax_exponent_bits total_bitsdata non_sign_bitsadditional_itemsifraction_items boundariesmeanss m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchao/optim/quant_utils.pycreate_dynamic_mapr s3" Dv"4!!15M]->>?!C $ % % O O FA!m#&77 81 < <q=(+<1S8 ",q01A56%?GGIII  O r 1A 56:;1S8 ",q01A56%?GGIII  O r 1A 56:;rUs   0000f$$C$$$$!6!!!!!,!6!!!!!$!V!6!&!!!! 5F5v555555r