Skip to content

Commit e3b9078

Browse files
committed
fix unit test
Signed-off-by: Anna Shors <ashors@nvidia.com>
1 parent 0cebe0e commit e3b9078

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/unit/algorithms/test_sequence_packing_gradients.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -344,6 +344,7 @@ def __exit__(self, exc_type, exc_val, exc_tb):
344344
},
345345
seq_length_key="input_lengths",
346346
pad_individual_seqs_to_multiple_of=pad_to_multiple,
347+
pad_packed_seq_to_multiple_of=1,
347348
pad_full_seq_to=max_seq_len * batch_size if cp_size > 1 else None,
348349
),
349350
model=MockModel(),

0 commit comments

Comments
 (0)