MCPcopy
hub / github.com/tinygrad/tinygrad / _has_metadata

Method _has_metadata

test/null/test_tensor_metadata.py:96–102  ·  view source on GitHub ↗
(self, h, name)

Source from the content-addressed store, hash-verified

94 self.assertEqual(si.arg.metadata, ())
95
96 def _has_metadata(self, h, name):
97 linears = []
98 capturing.append(type("", (), {"add_linear": lambda _, linear, var_vals: linears.append(linear)})())
99 try: h.realize()
100 finally: capturing.clear()
101 calls = [call for linear in linears for call in linear.src]
102 return any(m.name == name for call in calls for m in call.arg.metadata)
103
104 def test_metadata_survives_realize_pending_assign(self):
105 shared = Tensor.rand(4)

Calls 3

appendMethod · 0.80
realizeMethod · 0.80
clearMethod · 0.80

Tested by

no test coverage detected