Productive, portable, and performant GPU programming in Python.
import numpy as np
import taichi as ti
from tests import test_utils
@test_utils.test(require=ti.extension.quant, debug=True)
def test_1D_bit_array():
qu1 = ti.types.quant.int(1, False)
x = ti.field(dtype=qu1)
N = 32
ti.root.bit_array(ti.i, N, num_bits=32).place(x)
@ti.kernel
def set_val():
for i in range(N):
x[i] = i % 2
def verify_val():
assert x[i] == i % 2
set_val()
verify_val()
def test_2D_bit_array():
qi1 = ti.types.quant.int(1, False)
x = ti.field(dtype=qi1)
M, N = 4, 8
ti.root.bit_array(ti.ij, (M, N), num_bits=32).place(x)
for i in range(M):
for j in range(N):
x[i, j] = (i * N + j) % 2
assert x[i, j] == (i * N + j) % 2
