Skip to content

Commit 16c7afd

Browse files
authored
Update test_dynamic_cache_exportability_multiple_run (failing on torch 2.10 nightly) (#42212)
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
1 parent 309180f commit 16c7afd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/utils/test_cache_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -628,7 +628,7 @@ def test_dynamic_cache_exportability_multiple_run(self):
628628
past_key_values = res.past_key_values
629629

630630
shapes = torch.export.ShapesCollection()
631-
dyn = torch.export.Dim("seq", max=512)
631+
dyn = torch.export.Dim.DYNAMIC(max=512)
632632

633633
for ix in range(len(past_key_values)):
634634
shapes[past_key_values.layers[ix].keys] = (None, None, dyn, None)

0 commit comments

Comments
 (0)