def METHOD_NAME():
    a1 = torch.rand(128, 128)
    a2 = torch.zeros(128, 128)
    a3 = torch.zeros(128, 128)
    result = a1 + 2
    func_with_part_access_region.tune()
    func_with_part_access_region(a1, a2, a3)
    tvm.testing.assert_allclose(a3.numpy(), result.numpy(), atol=1e-5, rtol=1e-5)