This commit is contained in:
2026-06-17 16:22:13 -07:00
parent 0b5796e3c8
commit 009b484dc0
+1 -1
View File
@@ -568,7 +568,7 @@ class TestFunctionGrad(unittest.TestCase):
GlobalCounters.reset()
loss.realize(w1.grad, w2.grad, w3.grad)
print(GlobalCounters.global_ops, GlobalCounters.global_mem)
self.assertLessEqual(GlobalCounters.global_ops, 4739344)
self.assertLessEqual(GlobalCounters.global_ops, 5000000)
def test_function_grad_ops_precompile(self): self.test_function_grad_ops(precompile=True)
def test_function_grad_ops_precompile_backward(self):
self.test_function_grad_ops(precompile=True, precompile_backward=True)