update grad value

apache · Jul 24, 2020 · 7ec9749 · 7ec9749
1 parent 303e5ab
commit 7ec9749
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
@@ -1236,17 +1236,17 @@ def run_trsm(inp):
     grad, out = run_det(A)
     assert(out[0] == 1)
     out.backward()
-    check_diag(grad, 0)
+    check_diag(grad, 1)
 
     grad, out = run_inverse(A)
     check_diag(out, 1)
     out.backward()
-    check_diag(grad, 0)
+    check_diag(grad, -1)
 
     grad, out = run_trmm(A)
     check_diag(out, 1)
     out.backward()
-    check_diag(grad, 0)
+    check_diag(grad, 2)
 
     grad, out = run_trsm(A)
     check_diag(out, 1)
@@ -1258,12 +1258,12 @@ def run_trsm(inp):
     grad, out = run_inverse(B)
     check_batch_diag(out, 1)
     out.backward()
-    check_batch_diag(grad, 0)
+    check_batch_diag(grad, -1)
 
     grad, out = run_trmm(B)
     check_batch_diag(out, 1)
     out.backward()
-    check_batch_diag(grad, 0)
+    check_batch_diag(grad, 2)
 
     grad, out = run_trsm(B)
     check_batch_diag(out, 1)