re-enabled SD pipeline test

Author: AleksKnezevic

.github/workflows/run-op-by-op-model-tests.yml

@@ -104,9 +104,13 @@ jobs:
               "
           },
           {
-            runs-on: wormhole_b0, name: "stable-diffusion", tests: "
+            runs-on: wormhole_b0, name: "stable-diffusion-pipe", tests: "
               tests/models/stable_diffusion/test_stable_diffusion.py::test_stable_diffusion
-              tests/models/stable_diffusion/test_stable_diffusion_v2.py::test_stable_diffusion_v2
+              "
+          },
+          {
+            runs-on: wormhole_b0, name: "stable-diffusion-unet", tests: "
+              tests/models/stable_diffusion/test_stable_diffusion_unet.py::test_stable_diffusion_unet
               "
           },
           {

tests/models/stable_diffusion/test_stable_diffusion.py

@@ -6,7 +6,7 @@
 import torch
 import pytest
 from tests.utils import ModelTester
-from tt_torch.tools.utils import OpByOpBackend
+from tt_torch.tools.utils import CompilerConfig, CompileDepth, OpByOpBackend
 
 
 class ThisTester(ModelTester):
@@ -16,24 +16,37 @@ def _load_model(self):
         return pipe
 
     def _load_inputs(self):
-        prompt = "a photo of an astronaut riding a horse on mars"
+        prompt = [
+            "a photo of an astronaut riding a horse on mars",
+        ]
         return prompt
 
 
 @pytest.mark.parametrize(
     "mode",
     ["eval"],
 )
-@pytest.mark.skip(reason="Dynamo cannot support pipeline.")
 @pytest.mark.parametrize(
     "op_by_op",
     [OpByOpBackend.STABLEHLO, OpByOpBackend.TORCH, None],
     ids=["op_by_op_stablehlo", "op_by_op_torch", "full"],
 )
+@pytest.mark.xfail(
+    reason="Fails due to pt2 compile issue when finishing generation, but we can still generate a graph"
+)
 def test_stable_diffusion(record_property, mode, op_by_op):
     model_name = "Stable Diffusion"
 
-    tester = ThisTester(model_name, mode, record_property_handle=record_property)
+    cc = CompilerConfig()
+    cc.enable_consteval = True
+    cc.consteval_parameters = True
+    if op_by_op:
+        cc.compile_depth = CompileDepth.EXECUTE_OP_BY_OP
+        if op_by_op == OpByOpBackend.STABLEHLO:
+            cc.op_by_op_backend = OpByOpBackend.STABLEHLO
+    tester = ThisTester(
+        model_name, mode, compiler_config=cc, record_property_handle=record_property
+    )
     results = tester.test_model()
     if mode == "eval":
         image = results.images[0]

tests/models/stable_diffusion/test_stable_diffusion_v2.py renamed to tests/models/stable_diffusion/test_stable_diffusion_unet.py

@@ -69,8 +69,8 @@ def _load_inputs(self):
     [OpByOpBackend.STABLEHLO, OpByOpBackend.TORCH, None],
     ids=["op_by_op_stablehlo", "op_by_op_torch", "full"],
 )
-def test_stable_diffusion_v2(record_property, mode, op_by_op):
-    model_name = "Stable Diffusion V2"
+def test_stable_diffusion_unet(record_property, mode, op_by_op):
+    model_name = "Stable Diffusion UNET"
 
     cc = CompilerConfig()
     cc.enable_consteval = True