Skip to content

Commit

Permalink
fix __all__ in pytorch/permutation.py
Browse files Browse the repository at this point in the history
  • Loading branch information
hxbai committed Dec 13, 2024
1 parent 7a1d963 commit 6160104
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions transformer_engine/pytorch/permutation.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
__all__ = [
"moe_permute",
"moe_unpermute",
"moe_sort_chunks_by_indices",
]


Expand Down

0 comments on commit 6160104

Please sign in to comment.