[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
NVIDIA · Feb 2, 2025 · cf8b2fe · cf8b2fe
1 parent c385529
commit cf8b2fe
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/transformer_engine/common/util/cast_gated_kernels.cuh b/transformer_engine/common/util/cast_gated_kernels.cuh
@@ -955,9 +955,7 @@ void quantize_gated(const Tensor &grad, const Tensor &gated_input, Tensor *outpu
     NVTE_CHECK(output->flat_last_dim() == output_cols, "Wrong dimension of the output.");
   }
 
-  const bool use_tma_kernels = is_fp8_rowwise_output &&
-                               is_fp8_colwise_output &&
-                               cols % 16 == 0;
+  const bool use_tma_kernels = is_fp8_rowwise_output && is_fp8_colwise_output && cols % 16 == 0;
 
   if (is_delayed_tensor_scaling(output->scaling_mode)) {
     if (use_tma_kernels) {