Resolve review: small tensor for tests

Author: shino16

thunder/tests/test_dynamo.py

@@ -857,7 +857,7 @@ def checkpoint_fn(x):
 
     initial_mem = torch.cuda.memory_allocated()
 
-    x = torch.randn((1024 // 4, 1024, 1024), device="cuda", requires_grad=True)
+    x = torch.randn((128, 128), device="cuda", requires_grad=True)
     jfn = thunderfx(checkpoint_fn)
     y = jfn(x)