documentation fixes

Signed-off-by: Alp Dener <[email protected]>
NVIDIA · Dec 5, 2024 · b07bb2d · b07bb2d
1 parent 2acb92f
commit b07bb2d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/transformer_engine/jax/cpp_extensions/gemm.py b/transformer_engine/jax/cpp_extensions/gemm.py
@@ -516,7 +516,6 @@ def infer_sharding_from_operands(
         )
 
         # Modify operand specs:
-        # - FSDP axes are all-gathered
         # - LHS operand outer dimension is all-gathered if RHS operand outer dimension is sharded
         # - LHS operand contracting dimension sharding is forced to match RHS contracting dimension
         lhs_spec_new = [spec for spec in lhs_spec]
@@ -584,7 +583,6 @@ def partition(
         )
 
         # Modify operand specs:
-        # - FSDP axes are all-gathered
         # - LHS operand outer dimension is all-gathered if RHS operand outer dimension is sharded
         # - LHS operand contracting dimension sharding is forced to match RHS contracting dimension
         lhs_spec_new = [spec for spec in lhs_spec]