[PyTorch] Fix typo (#1495)

Fix typo Signed-off-by: Tim Moon <[email protected]> Co-authored-by: Kirthi Shankar Sivamani <[email protected]>
NVIDIA · Feb 19, 2025 · 56c0c07 · 56c0c07
1 parent 978f1d7
commit 56c0c07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transformer_engine/pytorch/distributed.py b/transformer_engine/pytorch/distributed.py
@@ -874,7 +874,7 @@ def _all_gather_fp8(
             dtype = input_.dtype
             device = input_.device
         out = quantizer.make_empty(out_shape, dtype=dtype, device=device)
-    elif isinstance(input, Float8Tensor):
+    elif isinstance(input_, Float8Tensor):
         out = input_.make_like(input_, shape=out_shape)
         out._data = torch.empty_like(
             out_shape,