codestype format

Author: lxd-cumt

transformer_engine/plugin/core/backends/flagos/impl/softmax.py

@@ -60,4 +60,4 @@ def scaled_masked_softmax_backward_fl(
     # grad_softmax = softmax_output_f32 * (output_grad_f32 - sum_term)
     grad_softmax = flag_gems.mul(softmax_output_f32, flag_gems.sub(output_grad_f32, sum_term))
     grad_scaled = flag_gems.mul(grad_softmax, scale_factor)
-    return flag_gems.to_copy(grad_scaled, dtype=orig_dtype)
+    return flag_gems.to_copy(grad_scaled, dtype=orig_dtype)