def test_field_builder_place_grad():
@ti.kernel
def mul(arr: ti.template(), out: ti.template()):
for i in arr:
out[i] = arr[i] * 2.0
@ti.kernel
def calc_loss(arr: ti.template(), loss: ti.template()):
for i in arr:
loss[None] += arr[i]
arr = ti.field(ti.f32, needs_grad=True)
fb0 = ti.FieldsBuilder()
fb0.dense(ti.i, 10).place(arr, arr.grad)
snode0 = fb0.finalize()
out = ti.field(ti.f32)
fb1 = ti.FieldsBuilder()
fb1.dense(ti.i, 10).place(out, out.grad)
snode1 = fb1.finalize()
loss = ti.field(ti.f32)
fb2 = ti.FieldsBuilder()
fb2.place(loss, loss.grad)
snode2 = fb2.finalize()
arr.fill(1.0)
mul(arr, out)
calc_loss(out, loss)
loss.grad[None] = 1.0
calc_loss.grad(out, loss)
mul.grad(arr, out)
for i in range(10):
assert arr.grad[i] == 2.0
def test_fields_builder_pointer():
n = 5
fb1 = ti.FieldsBuilder()
x = ti.field(ti.f32)
fb1.pointer(ti.i, n).place(x)
fb1.finalize()
@ti.kernel
def func1():
for i in range(n):
x[i] = i * 3
func1()
for i in range(n):
assert x[i] == i * 3
fb2 = ti.FieldsBuilder()
y = ti.field(ti.f32)
fb2.pointer(ti.i, n).place(y)
z = ti.field(ti.f32)
fb2.pointer(ti.i, n).place(z)
fb2.finalize()
# test range-for
@ti.kernel
def func2():
for i in range(n):
x[i] = i * 2
for i in range(n):
y[i] = i + 5
for i in range(n):
z[i] = i + 10
func2()
for i in range(n):
assert x[i] == i * 2
assert y[i] == i + 5
assert z[i] == i + 10
# test struct-for
@ti.kernel
def func3():
for i in y:
y[i] += 5
for i in z:
z[i] -= 5
func3()
for i in range(n):
assert y[i] == i + 10
assert z[i] == i + 5
func1()
for i in range(n):
assert x[i] == i * 3
def test_fields_builder_pointer():
shape = 5
fb1 = ti.FieldsBuilder()
x = ti.field(ti.f32)
fb1.pointer(ti.i, shape).place(x)
fb1.finalize()
@ti.kernel
def assign_field_single():
for i in range(shape):
x[i] = i * 3
assign_field_single()
for i in range(shape):
assert x[i] == i * 3
fb2 = ti.FieldsBuilder()
y = ti.field(ti.f32)
fb2.pointer(ti.i, shape).place(y)
z = ti.field(ti.f32)
fb2.pointer(ti.i, shape).place(z)
fb2.finalize()
@ti.kernel
def assign_field_multiple_range_for():
for i in range(shape):
x[i] = i * 2
for i in range(shape):
y[i] = i + 5
for i in range(shape):
z[i] = i + 10
assign_field_multiple_range_for()
for i in range(shape):
assert x[i] == i * 2
assert y[i] == i + 5
assert z[i] == i + 10
@ti.kernel
def assign_field_multiple_struct_for():
for i in y:
y[i] += 5
for i in z:
z[i] -= 5
assign_field_multiple_struct_for()
for i in range(shape):
assert y[i] == i + 10
assert z[i] == i + 5
assign_field_single()
for i in range(shape):
assert x[i] == i * 3
def test_field_initialize_zero():
fb0 = ti.FieldsBuilder()
a = ti.field(ti.i32)
fb0.dense(ti.i, 1).place(a)
c = fb0.finalize()
a[0] = 5
c.destroy()
fb1 = ti.FieldsBuilder()
b = ti.field(ti.i32)
fb1.dense(ti.i, 1).place(b)
d = fb1.finalize()
assert b[0] == 0
def B(i):
n = i * 10**3
fb = ti.FieldsBuilder()
a = ti.field(ti.f64)
fb.dense(ti.i, n).place(a)
c = fb.finalize()
ni = i * 10**3
fbi = ti.FieldsBuilder()
ai = ti.field(ti.f64)
fbi.dense(ti.i, n).place(ai)
ci = fbi.finalize()
c.destroy()
ci.destroy()
def test_for_multi_destroy_multi_fields():
fb0 = ti.FieldsBuilder()
fb1 = ti.FieldsBuilder()
for create_field_idx in range(10):
field0 = ti.field(field_type)
field1 = ti.field(field_type)
fb0.dense(ti.i, test_1d_size).place(field0)
fb1.pointer(ti.i, test_1d_size).place(field1)
fb0_snode_tree = fb0.finalize()
fb1_snode_tree = fb1.finalize()
fb0_snode_tree.destroy()
fb1_snode_tree.destroy()
def test_for_single_destroy_multi_fields():
fb = ti.FieldsBuilder()
for create_field_idx in range(10):
field = ti.field(field_type)
fb.dense(ti.i, test_1d_size).place(field)
fb_snode_tree = fb.finalize()
fb_snode_tree.destroy()
def A(i):
n = i * 10**3
fb = ti.FieldsBuilder()
a = ti.field(ti.f64)
fb.dense(ti.i, n).place(a)
c = fb.finalize()
c.destroy()
def test_for_raise_destroy_twice():
fb = ti.FieldsBuilder()
a = ti.field(ti.f32)
fb.dense(ti.i, test_1d_size).place(a)
c = fb.finalize()
with pytest.raises(TaichiRuntimeError):
c.destroy()
c.destroy()
def test_fields_builder_dense():
n = 5
fb1 = ti.FieldsBuilder()
x = ti.field(ti.f32)
fb1.dense(ti.i, n).place(x)
fb1.finalize()
@ti.kernel
def func1():
for i in range(n):
x[i] = i * 3
func1()
for i in range(n):
assert x[i] == i * 3
fb2 = ti.FieldsBuilder()
y = ti.field(ti.f32)
fb2.dense(ti.i, n).place(y)
z = ti.field(ti.f32)
fb2.dense(ti.i, n).place(z)
fb2.finalize()
@ti.kernel
def func2():
for i in range(n):
x[i] = i * 2
for i in range(n):
y[i] = i + 5
for i in range(n):
z[i] = i + 10
func2()
for i in range(n):
assert x[i] == i * 2
assert y[i] == i + 5
assert z[i] == i + 10
func1()
for i in range(n):
assert x[i] == i * 3
def test_fields_builder_dense():
shape = 5
fb1 = ti.FieldsBuilder()
x = ti.field(ti.f32)
fb1.dense(ti.i, shape).place(x)
fb1.finalize()
@ti.kernel
def assign_field_single():
for i in range(shape):
x[i] = i * 3
assign_field_single()
for i in range(shape):
assert x[i] == i * 3
fb2 = ti.FieldsBuilder()
y = ti.field(ti.f32)
fb2.dense(ti.i, shape).place(y)
z = ti.field(ti.f32)
fb2.dense(ti.i, shape).place(z)
fb2.finalize()
@ti.kernel
def assign_field_multiple():
for i in range(shape):
x[i] = i * 2
for i in range(shape):
y[i] = i + 5
for i in range(shape):
z[i] = i + 10
assign_field_multiple()
for i in range(shape):
assert x[i] == i * 2
assert y[i] == i + 5
assert z[i] == i + 10
assign_field_single()
for i in range(shape):
assert x[i] == i * 3
