Update op_builder/npu/fused_adam.py

Co-authored-by: Hz, Ji <[email protected]>
microsoft · Oct 14, 2023 · fb77cd0 · fb77cd0
1 parent d808f86
commit fb77cd0
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/op_builder/npu/fused_adam.py b/op_builder/npu/fused_adam.py
@@ -52,7 +52,6 @@ def multi_tensor_adam(chunk_size, noop_flag_buffer, tensor_lists, lr, beta1, bet
                     False,  # use_locking
                     False,  # use_nesterov
                     out=(param_flat.data, m_flat, v_flat))
-                )
 
 
 class FusedAdamBuilder(NPUOpBuilder):