comment

NVIDIA · Dec 9, 2024 · 0d1dfb6 · 0d1dfb6
1 parent 6a84d85
commit 0d1dfb6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/python/test_transformer_engine.py b/tests/python/test_transformer_engine.py
@@ -23,7 +23,9 @@ class ComputeType(Enum):
 
 
 class Parallelism(Enum):
+    # https://docs.nvidia.com/nemo-framework/user-guide/latest/nemotoolkit/features/parallelisms.html#tensor-parallelism
     TENSOR_PARALLEL = auto()
+    # https://docs.nvidia.com/nemo-framework/user-guide/latest/nemotoolkit/features/parallelisms.html#sequence-parallelism
     SEQUENCE_PARALLEL = auto()