o Ơi@sjddlZddlZddlZddlmZddlmZmZeej eej ddZ eej j j dd Z dS) N)tensorrt_converter) get_dyn_rangemark_as_int8_layercCsht|dkr |\}}n |d|d}}tdd||fDs#td|||tjj}||_|dS)Nrinputothercs|] }t|tjjVqdSN isinstancetrttensorrtITensor).0argrT/home/ubuntu/.local/lib/python3.10/site-packages/torch_tensorrt/fx/converters/mul.py zmul..z@mul() received an input that is not part of the TensorRT region!) lenall RuntimeErroradd_elementwiser ElementWiseOperationPRODname get_output)networktargetargskwargs layer_namelhs_valrhs_vallayerrrrmul s   r&c Cst|dksJ|d|d}}tdd||fDs td|||tjj}||_t|d|dt j }t ||| dS) Nrqaqbcsr r r )rirrrr&rz quantized_mul..zHQuantized mul received an input that is not part of the TensorRT region!scale zero_point) rrrrr rrrrtorchquint8rr) rrr r!r"r#r$r% dyn_rangerrr quantized_mul!s  r/) operatorrr r,converter_registryrconverter_utilsrrr&ops quantizedr/rrrrs