def METHOD_NAME(self):
    @jit.rawkernel()
    def test_thread_block(x):
        g = jit.cg.this_thread_block()
        if g.thread_rank() == 0:
            x[0] = g.num_threads()
        if g.thread_rank() == 1:
            d_th = g.dim_threads()
            x[1], x[2], x[3] = d_th.x, d_th.y, d_th.z
        g.sync()
    x = cupy.empty((16,), dtype=cupy.int64)
    x[:] = -1
    test_thread_block[1, 32](x)
    assert x[0] == 32
    assert (x[1], x[2], x[3]) == (32, 1, 1)
    assert (x[4:] == -1).all()