Method test_graph_offset_bufs

test/backend/test_graph.py:211–233

(self)

      for buf in b0 + b1: np.testing.assert_equal(expected[buf], np.frombuffer(buf.as_memoryview(), np.int32))

  def test_graph_offset_bufs(self):
    self.skip_if_not_multigraph()
    d0 = Device.DEFAULT
    if not hasattr(Device[d0].allocator, "_offset"): self.skipTest("device does not support _offset")

    b0 = make_buffer(d0, fill=True)
    b1 = make_view(b0, 0, b0.size)
    b2 = make_view(b0, 0, b0.size)
    c: dict[Buffer,UOp] = {}

    calls = [
      UOp(Ops.COPY).call(get_buf_uop(b0,c), get_buf_uop(b2,c), metadata=()),
      get_ast(d0, 2).call(get_buf_uop(b1,c), get_buf_uop(b0,c), get_buf_uop(b2,c), metadata=()),
    ]

    zero_bufs([b0])
    run_schedule(calls)
    expected = np.frombuffer(b0.as_memoryview(), np.int32).copy()

    for _ in range(RUN_CNT):
      zero_bufs([b0])
      make_graph(Device[d0].graph, calls)([], {})
      np.testing.assert_equal(expected, np.frombuffer(b0.as_memoryview(), np.int32))

  def test_partial_write_preserves_write_dep(self):
    self.skip_if_not_multigraph()