transformer_engine/pytorch/triton/permutation.py · 12c3e323773d639319f917d07c4561e2424010fc · OpenDAS / TransformerEngine

Fix import error on CPU only devices (#1578) · 12c3e323

hx authored Mar 14, 2025



fix cpu device import error
Signed-off-by: Hongxiao Bai <hongxiaob@nvidia.com>
Co-authored-by: Tim Moon <4406448+timmoon10@users.noreply.github.com>

12c3e323

permutation.py 23.3 KB

Replace permutation.py