projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[pytorch.git]
/
pscan.py
diff --git
a/pscan.py
b/pscan.py
index
3526c31
..
c547823
100755
(executable)
--- a/
pscan.py
+++ b/
pscan.py
@@
-114,3
+114,8
@@
if __name__ == "__main__":
print((gA - gA_ref).norm())
print((gX - gX_ref).norm())
print((gY_init - gY_init_ref).norm())
print((gA - gA_ref).norm())
print((gX - gX_ref).norm())
print((gY_init - gY_init_ref).norm())
+
+ Y1 = pscan(A[:, : T // 2], X[:, : T // 2], Y_init)
+ Y2 = pscan(A[:, T // 2 :], X[:, T // 2 :], Y1[:, -1])
+
+ print((Y - torch.cat([Y1, Y2], dim=1)).norm())