MCPcopy
hub / github.com/tinygrad/tinygrad / test_subbuffer_is_allocated

Method test_subbuffer_is_allocated

test/backend/test_subbuffer.py:69–107  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

67 assert tst == [13, 14]
68
69 def test_subbuffer_is_allocated(self):
70 buf = self.buf_unalloc
71 sub_buf = buf.view(3, dtypes.uint8, offset=4)
72 self.assertFalse(buf.is_allocated())
73 self.assertFalse(buf.is_initialized())
74 self.assertFalse(sub_buf.is_allocated())
75 self.assertFalse(sub_buf.is_initialized())
76
77 # base buffer alloc
78 buf.allocate()
79 self.assertTrue(buf.is_allocated())
80 self.assertTrue(buf.is_initialized())
81 self.assertTrue(sub_buf.is_allocated())
82 self.assertFalse(sub_buf.is_initialized())
83
84 # sub buffer alloc
85 sub_buf.allocate()
86 self.assertTrue(sub_buf.is_initialized())
87
88 # sub buffer dealloc
89 sub_buf.deallocate()
90 self.assertTrue(buf.is_allocated())
91 self.assertTrue(buf.is_initialized())
92 self.assertTrue(sub_buf.is_allocated())
93 self.assertFalse(sub_buf.is_initialized())
94
95 # base buffer dealloc
96 buf.deallocate()
97 self.assertFalse(buf.is_allocated())
98 self.assertFalse(buf.is_initialized())
99 self.assertFalse(sub_buf.is_allocated())
100 self.assertFalse(sub_buf.is_initialized())
101
102 # sub buffer alloc
103 sub_buf.ensure_allocated()
104 self.assertTrue(buf.is_allocated())
105 self.assertTrue(buf.is_initialized())
106 self.assertTrue(sub_buf.is_allocated())
107 self.assertTrue(sub_buf.is_initialized())
108
109 def test_subbuffer_copy_in_out(self):
110 sub_buf = self.buf.view(3, dtypes.uint8, offset=3).ensure_allocated() # [3:6]

Callers

nothing calls this directly

Calls 6

allocateMethod · 0.80
deallocateMethod · 0.80
ensure_allocatedMethod · 0.80
viewMethod · 0.45
is_allocatedMethod · 0.45
is_initializedMethod · 0.45

Tested by

no test coverage detected