()
| 68 | |
| 69 | |
| 70 | def test_buffer_access_ptr_extent(): |
| 71 | m = tvm.tirx.SizeVar("m", "int32") |
| 72 | n = tvm.tirx.SizeVar("n", "int32") |
| 73 | Ab = tvm.tirx.decl_buffer((m, n), "float32") |
| 74 | aptr = Ab.access_ptr("rw") |
| 75 | tvm.ir.assert_structural_equal(aptr.args[3], m * n) |
| 76 | aptr = Ab.access_ptr("rw", offset=100) |
| 77 | tvm.ir.assert_structural_equal(aptr.args[3], m * n - 100) |
| 78 | Ab = tvm.tirx.decl_buffer((m, n), "float32", strides=[n + 1, 1]) |
| 79 | aptr = Ab.access_ptr("rw", offset=100) |
| 80 | tvm.ir.assert_structural_equal(aptr.args[3], Ab.strides[0] * m - 100) |
| 81 | |
| 82 | # Test extent from input params |
| 83 | aptr = Ab.access_ptr("rw", extent=200) |
| 84 | tvm.ir.assert_structural_equal(aptr.args[3], T.int32(200)) |
| 85 | aptr = Ab.access_ptr("rw", offset=100, extent=100) |
| 86 | tvm.ir.assert_structural_equal(aptr.args[3], T.int32(100)) |
| 87 | |
| 88 | |
| 89 | def test_buffer_vload(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…