[Compile] Only test compiling model forward pass (#35658)

* rename test to only compile forward! * style emu
2025-01-13 13:43:29 +01:00
parent 84a6789145
commit e6f9b03464
7 changed files with 9 additions and 34 deletions
--- a/tests/models/dbrx/test_modeling_dbrx.py
+++ b/tests/models/dbrx/test_modeling_dbrx.py
@@ -369,7 +369,7 @@ class DbrxModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin
        pass

    @unittest.skip("Dbrx does not support `torch.compile` with `fullgraph=True`.")
-    def test_generate_compile_fullgraph(self):
+    def test_generate_compile_model_forward(self):
        pass