code: def testLargeTensor(self):
      t = torch.zeros(2359332864, dtype=torch.half, device="cuda")
      t2 = torch.zeros(2359332864, dtype=torch.half, device="cuda")
      grad = torch.randn_like(t)
      t.grad = grad
      t2.grad = grad
      params = [t]
      params2 = [t2]
      optimizer = apex.optimizers.FusedAdam(params, lr=self.lr)
      optimizer.step()
      optimizer2 = torch.optim.Adam(params2, lr=self.lr)
      torch.testing.assert_close(t, t2)
      torch.cuda.synchronize()
  
file_path: apex/tests/L0/run_optimizers/test_adam.py
imports: import copy
  import math
  import random
  import unittest
  import torch
  import torch.nn.functional
  from torch import nn
label_desc: Use image node_id 6 for calling the AdamTest obj's underlying member method code with example usage: obj.testLargeTensor() without return types