transformer_engine/pytorch/triton/permutation.py · e05f87e193b9aa9cc385c958a42e9498ea0c68cd · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[PyTorch] Change order of args in another permutation triton kernel (#2488) · e05f87e1
Teddy Do authored Dec 09, 2025
```
change order
Signed-off-by: tdophung <tdophung@nvidia.com>
```
e05f87e1

permutation.py 13.3 KB