Skip to content

Commit

Permalink
Add reason for test skip
Browse files Browse the repository at this point in the history
  • Loading branch information
tbennun committed Nov 10, 2024
1 parent b95afb4 commit 1333054
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions tests/python_frontend/augassign_wcr_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,11 +44,11 @@ def augassign_wcr3(A: dace.int32[10, 10, 10], B: dace.int32[10], W: dace.bool_[1

@dace.program
def augassign_wcr4():
a = np.zeros((10,))
a = np.zeros((10, ))
for i in dace.map[1:9]:
a[i-1] += 1
a[i - 1] += 1
a[i] += 2
a[i+1] += 3
a[i + 1] += 3
return a


Expand Down Expand Up @@ -121,6 +121,7 @@ def test_augassign_wcr3():


def test_augassign_no_wcr():

@dace.program
def no_wcr(A: dace.int32[5, 5, 5]):
A[2, 3, :] += A[3, 2, :]
Expand All @@ -139,6 +140,7 @@ def no_wcr(A: dace.int32[5, 5, 5]):


def test_augassign_no_wcr2():

@dace.program
def no_wcr(A: dace.int32[5, 5, 5]):
A[2, 3, 1:4] += A[2:5, 1, 4]
Expand All @@ -156,8 +158,9 @@ def no_wcr(A: dace.int32[5, 5, 5]):
assert (np.allclose(A, ref))


@pytest.mark.skip('Atomic reduction is generated as non-atomic')
def test_augassign_wcr4():

with dace.config.set_temporary('frontend', 'avoid_wcr', value=False):
val = augassign_wcr4()
ref = augassign_wcr4.f()
Expand Down

0 comments on commit 1333054

Please sign in to comment.