nit

pytorch · Dec 12, 2023 · cb3bad6 · cb3bad6
1 parent f03ab86
commit cb3bad6
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/test/spmd/test_xla_sharding.py b/test/spmd/test_xla_sharding.py
@@ -1052,7 +1052,6 @@ def test_backward_optimization_barrier(self):
     # The first layer won't have gradients in the hook. Not sure why.
     xs.xla_sharding.apply_backward_optimization_barrier(model.fc2)
 
-    # optimizer.zero_grad()
     x = torch.randn(2, 128).to(xm.xla_device())
     y = model(x)
     loss = y.sum()