import taichi as ti
ti.init(arch=ti.cpu,default_fp=ti.f32,default_ip=ti.i32)
@ti.kernel
def cycle():
M=ti.Matrix.identity(float,8)
for i0,j0 in ti.static(ti.ndrange(2, 2)):
for i1,j1 in ti.static(ti.ndrange(2, 2)):
for i2,j2 in ti.static(ti.ndrange(2, 2)):
if i1==j1==1 and i0 != j0 :
m,n=2*i0 + 6*j0 + i2, 4 + j2
print(m,n,'d1')
# M[2*i0 + 6*j0 + i2,4 + j2]=1
cycle()
import taichi as ti
ti.init(arch=ti.cpu,default_fp=ti.f32,default_ip=ti.i32)
@ti.kernel
def cycle():
M=ti.Matrix.identity(float,8)
for i0,j0 in ti.static(ti.ndrange(2, 2)):
for i1,j1 in ti.static(ti.ndrange(2, 2)):
for i2,j2 in ti.static(ti.ndrange(2, 2)):
if ti.static(i1==j1==1 and i0 != j0):
m,n=2*i0 + 6*j0 + i2, 4 + j2
print(m,n,'d1')
M[2*i0 + 6*j0 + i2,4 + j2]=1
cycle()