Fix lint.

Signed-off-by: Dennis Liu <[email protected]>
NVIDIA · Mar 15, 2024 · 8dc92d1 · 8dc92d1
1 parent 23c24d8
commit 8dc92d1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/transformer_engine/pytorch/cpp_extensions/cast.py b/transformer_engine/pytorch/cpp_extensions/cast.py
@@ -42,8 +42,7 @@ def cast_to_fp8(
             fp8_tensor,
             otype,
         )
-    else:
-        return inp
+    return inp
 
 def cast_from_fp8(
     inp: torch.Tensor,

diff --git a/transformer_engine/pytorch/module/linear.py b/transformer_engine/pytorch/module/linear.py
@@ -903,7 +903,6 @@ def forward(
                 weight_tensor = self.weight_tensor
                 bias_tensor = self.bias_tensor
 
-
             # Fetch the fp8 weights placeholders (for linear/gemm)
             weight1_fp8, weight1_t_fp8 = self.get_fp8_weights_scratchpad(
                 is_first_microbatch