From 2d0f1845b612468ceef085e7e07b6ac7f627a0b7 Mon Sep 17 00:00:00 2001 From: Jacob Hinkle Date: Thu, 21 Mar 2024 17:41:13 +0000 Subject: [PATCH] Fix sloppy merged Fixes the failing test_cse_rematerialization --- thunder/tests/test_nvfuser.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/thunder/tests/test_nvfuser.py b/thunder/tests/test_nvfuser.py index cdeac92a29..3cc325ec30 100644 --- a/thunder/tests/test_nvfuser.py +++ b/thunder/tests/test_nvfuser.py @@ -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