()
| 50 | |
| 51 | |
| 52 | def test_buffer_access_ptr_offset(): |
| 53 | m = tvm.tirx.SizeVar("m", "int32") |
| 54 | n = tvm.tirx.SizeVar("n", "int32") |
| 55 | Ab = tvm.tirx.decl_buffer((m, n), "float32") |
| 56 | aptr = Ab.access_ptr("rw", offset=100) |
| 57 | tvm.testing.assert_prim_expr_equal(aptr.args[2], 100) |
| 58 | assert aptr.args[4].value == Buffer.READ | Buffer.WRITE |
| 59 | v = tvm.tirx.SizeVar("int32", "int32") |
| 60 | aptr = Ab.access_ptr("rw", offset=100 + 100 + v) |
| 61 | tvm.testing.assert_prim_expr_equal(aptr.args[2], 200 + v) |
| 62 | assert aptr.args[4].value == Buffer.READ | Buffer.WRITE |
| 63 | aptr = Ab.access_ptr("rw", offset=tvm.tirx.call_extern("int32", "test_call", 100 + 100 + v)) |
| 64 | tvm.testing.assert_prim_expr_equal( |
| 65 | aptr.args[2], tvm.tirx.call_extern("int32", "test_call", 200 + v) |
| 66 | ) |
| 67 | assert aptr.args[4].value == Buffer.READ | Buffer.WRITE |
| 68 | |
| 69 | |
| 70 | def test_buffer_access_ptr_extent(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…