Add test_compile stage to CI

Author: qubvel

.github/workflows/tests.yml

@@ -51,7 +51,7 @@ jobs:
       run: uv pip list
 
     - name: Test with PyTest
-      run: uv run pytest -v -rsx -n 2 --cov=segmentation_models_pytorch --cov-report=xml --cov-config=pyproject.toml -k "not logits_match"
+      run: uv run pytest -v -rsx -n 2 --cov=segmentation_models_pytorch --cov-report=xml --cov-config=pyproject.toml --non-marked-only
 
     - name: Upload coverage reports to Codecov
       uses: codecov/codecov-action@v5
@@ -73,7 +73,22 @@ jobs:
     - name: Show installed packages
       run: uv pip list
     - name: Test with PyTest
-      run: RUN_SLOW=1 uv run pytest -v -rsx -n 2 -k "logits_match"
+      run: RUN_SLOW=1 uv run pytest -v -rsx -n 2 -m "logits_match"
+
+  test_torch_compile:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python
+      uses: astral-sh/setup-uv@v5
+      with:
+        python-version: "3.10"
+    - name: Install dependencies
+      run: uv pip install -r requirements/required.txt -r requirements/test.txt
+    - name: Show installed packages
+      run: uv pip list
+    - name: Test with PyTest
+      run: uv run pytest -v -rsx -n 2 -m "compile"
 
   minimum:
     runs-on: ubuntu-latest
@@ -88,4 +103,4 @@ jobs:
     - name: Show installed packages
       run: uv pip list
     - name: Test with pytest
-      run: uv run pytest -v -rsx -n 2 -k "not logits_match"
+      run: uv run pytest -v -rsx -n 2 --non-marked-only