(self, h, name)
| 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) |
no test coverage detected