@ti.kernel
def splat(t: ti.i32):
    for i in range(n_particles):
        pos = ti.cast((x[(t, i)] * visualize_resolution), ti.i32)
        screen[(pos[0], pos[1])][0] += 0.1
@ti.kernel
def splat(t: ti.i32):
    for i in range(n_particles):
        pos = ti.cast((x[(t, i)] * visualize_resolution), ti.i32)
        screen[(pos[0], pos[1])][0] += 0.1