Skip to content

Commit

Permalink
polish fuser
Browse files Browse the repository at this point in the history
  • Loading branch information
cjld committed Apr 22, 2022
1 parent cac2b83 commit 3d48612
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 1 deletion.
2 changes: 1 addition & 1 deletion python/jittor/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# file 'LICENSE.txt', which is part of this source code package.
# ***************************************************************

__version__ = '1.3.3.1'
__version__ = '1.3.3.3'
from jittor_utils import lock
with lock.lock_scope():
ori_int = int
Expand Down
10 changes: 10 additions & 0 deletions python/jittor/test/test_fuser.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,16 @@ def test_wrong_fuse2(self):
np.testing.assert_allclose(b.data, bb.data)
np.testing.assert_allclose(c.data, cc.data)

def test_for_fuse(self):
arr = []
x = 0
for i in range(100):
arr.append(jt.array(1))
x += arr[-1]
x.sync()
for i in range(100):
# print(arr[i].debug_msg())
assert ",0)" not in arr[i].debug_msg()



Expand Down
Binary file modified python/jittor/utils/data.gz
Binary file not shown.

0 comments on commit 3d48612

Please sign in to comment.