Skip to content

Commit

Permalink
Fix sloppy merged
Browse files Browse the repository at this point in the history
Fixes the failing test_cse_rematerialization
  • Loading branch information
jacobhinkle committed Mar 21, 2024
1 parent a93b35e commit 2d0f184
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions thunder/tests/test_nvfuser.py
Original file line number Diff line number Diff line change
Expand Up @@ -356,11 +356,11 @@ def test_cse_rematerialization(executor, device, _):
assert len(fusion_bsyms) == 11
# fusion groups 1 and 6 correspond with the apply_rotary_emb function
# Nvfuser with recomputation should use precomputed cos and sin values.
assert len(fusion_bsyms[1].args) == len(fusion_bsyms[7].args)
assert len(fusion_bsyms[1].args) == len(fusion_bsyms[6].args)
assert fusion_bsyms[1].subsymbols[0].output.name == "freqs_cos"
assert fusion_bsyms[1].subsymbols[1].output.name == "freqs_sin"
assert fusion_bsyms[7].subsymbols[0].output.name == "freqs_cos"
assert fusion_bsyms[7].subsymbols[1].output.name == "freqs_sin"
assert fusion_bsyms[6].subsymbols[0].output.name == "freqs_cos"
assert fusion_bsyms[6].subsymbols[1].output.name == "freqs_sin"


# Tests that two separated nvFuser regions can be merged when they don't depend
Expand Down

0 comments on commit 2d0f184

Please sign in to comment.