def test_native_unroll(): """ Tests that unrolling functionality works. """ a = 0 for i in dace.unroll(range(2, 4)): a += i * i assert a == 13
def sometest(A): for i in dace.unroll(range(n)): something[i](A)
def tounroll(A: dace.float64[1]): for val in dace.unroll(mygenerator()): A += val
def tounroll(A: dace.float64[1]): for i in dace.unroll(range(1, 4)): A[0] += i * i if i in (2, 3): break
def program(A): for i in dace.unroll(range(n)): called(A)