@@ -0,0 +1,501 @@
+import taichi_lang as ti%0Afrom pytest import approx%0A%0Ati.cfg.use_llvm = True%0A%0Adef test_arg_load():%0A  ti.reset()%0A  x = ti.var(ti.i32)%0A  y = ti.var(ti.f32)%0A%0A  @ti.layout%0A  def layout():%0A    ti.root.place(x, y)%0A%0A  @ti.kernel%0A  def set_i32(v : ti.i32):%0A    x%5BNone%5D = v%0A%0A  @ti.kernel%0A  def set_f32(v : ti.f32):%0A    y%5BNone%5D = v%0A%0A  set_i32(123)%0A  assert x%5BNone%5D == 123%0A%0A  set_i32(456)%0A  assert x%5BNone%5D == 456%0A%0A  set_f32(0.125)%0A  assert y%5BNone%5D == 0.125%0A%0A  set_f32(1.5)%0A  assert y%5BNone%5D == 1.5%0A%0Atest_arg_load()%0A

test llvm arg load