Merge branch 'main' into patch-1

kit1980 · Jan 28, 2025 · b589483 · b589483
2 parents 8576a7a + 2fce82b
commit b589483
Show file tree

Hide file tree

Showing 6 changed files with 1,625 additions and 50 deletions.
diff --git a/docs/api/pytorch.rst b/docs/api/pytorch.rst
@@ -52,6 +52,8 @@ pyTorch
 
 .. autoapifunction:: transformer_engine.pytorch.moe_unpermute
 
+.. autoapifunction:: transformer_engine.pytorch.moe_sort_chunks_by_index
+
 .. autoapifunction:: transformer_engine.pytorch.initialize_ub
 
 .. autoapifunction:: transformer_engine.pytorch.destroy_ub