o Þ)i3 ã @s„UddlmZmZddlZdgZeƒZeeejejfe d<d dej dej de dej fd d„Z dej dej de dej fd d „Z dS) é)ÚSetÚTupleNÚ safe_permuteÚ_EXPERIMENTAL_DTYPESÚvalueÚpermÚdimÚreturncCsh|j|jf}|tvrt|||ƒSz|ttdƒg||gƒWSty3t |¡t|||ƒYSw)a> Perform out-of-place permutation without using torch.Tensor.index_put_, whose implementation is missing for datatypes such as `torch.float8_e4m3fn` :param value: tensor to permute :param perm: permutation map :param dim: dimension along which to apply permutation :return: permuted value N)ÚdtypeÚdevicerÚ_fallback_permuteÚtupleÚsliceÚ RuntimeErrorÚadd)rrrZ dtype_tuple©rúl/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/compressed_tensors/utils/permute.pyrs    ýcCsf| ¡}tdƒg|d}tdƒg|d}t|ƒD]\}}|||<|||<|t|ƒ|t|ƒ<q|S)zÛ Fallback permutation method for experimental dtypes. :param value: tensor to permute :param perm: permutation map :param dim: dimension along which to apply permutation :return: permuted value Né)ÚclonerÚ enumerater )rrrZ value_retZ orig_slicesZ perm_slicesÚindexZ perm_indexrrrr 2s r )r)ÚtypingrrZtorchÚ__all__Úsetrr r Ú__annotations__ZTensorÚintrr rrrrÚs"ÿÿÿþ