code: def test_grad_values_conserved():
      with torch.no_grad():  # remove a warning
          param = torch.rand((2, 3), requires_grad=True)
          param.grad = torch.rand(2, 3)
  
          bucket = GradBucket(10, param.dtype, param.device, -1)
          param_ = param.clone()
  
          bucket.add_grad(param_)
          torch.allclose(param.grad, param_.grad)
  
imports: import pytest
  import torch
  from fairscale.nn.misc import GradBucket
label_desc: Use image node_id 1 for calling a global function with example usage: test_grad_values_conserved() without return types