Skip to content

Commit 3718095

Browse files
committed
extend test
1 parent 8cad654 commit 3718095

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

_unittests/ut_torch_export_patches/test_patch_module.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -568,6 +568,13 @@ def loop_body_1(z, iv, x, y):
568568
ep = torch.export.export(Model(), (x, y), dynamic_shapes=ds, strict=False)
569569
self.assertEqualAny(expected, ep.module()(x, y))
570570

571+
"""
572+
position_encodings = torch.cat(
573+
[weight[:, :required_pos_encodings_columns]
574+
for weight in broadcasted_weights], dim=-1
575+
)
576+
"""
577+
571578

572579
if __name__ == "__main__":
573580
unittest.main(verbosity=2)

0 commit comments

Comments
 (0)