mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 00:20:18 +01:00
- Follow-up to #107981, adding testing for metadata copying in placeholder nodes within the `split_by_tags` utility - Validation included in the test from #107248, since both tests are relevant to the same aspect of the utility Pull Request resolved: https://github.com/pytorch/pytorch/pull/108731 Approved by: https://github.com/angelayi
This commit is contained in:
parent
bf20b56e9d
commit
e30d436b01
|
|
@ -34,6 +34,11 @@ class TestFXSplit(TestCase):
|
|||
self.assertIn("name", n.meta)
|
||||
self.assertEqual(n.meta["name"], n.name)
|
||||
|
||||
# Validate that metadata is copied correctly for graph placeholder nodes
|
||||
for node in split_gm.graph.nodes:
|
||||
if node.op == "placeholder":
|
||||
self.assertIn("name", node.meta)
|
||||
self.assertEqual(node.meta["name"], node.name)
|
||||
|
||||
class TestSplitByTags(TestCase):
|
||||
class TestModule(torch.nn.Module):
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user