Update hpu-tests.yml

Lightning-AI · Oct 17, 2024 · 7ecbfd8 · 7ecbfd8
1 parent bb8f4e8
commit 7ecbfd8
Showing 1 changed file with 15 additions and 15 deletions.
diff --git a/.azure/hpu-tests.yml b/.azure/hpu-tests.yml
@@ -135,14 +135,14 @@ jobs:
         PT_HPU_LAZY_MODE: 0
       displayName: 'HPU torch compile tests'
 
-    # - bash: |
-    #     python -m pytest -sv tests/test_fabric/test_fsdp.py \
-    #       --hpus 1 --junitxml=hpu_test_fsdp-fabric-results.xml
-    #     python -m pytest -sv tests/test_pytorch/strategies/test_fsdp.py \
-    #       --hpus 1 --junitxml=hpu_test_fsdp-results.xml
-    #   env:
-    #     PT_HPU_LAZY_MODE: 0
-    #   displayName: 'HPU FSDP tests'
+    - bash: |
+        python -m pytest -sv tests/test_fabric/test_fsdp.py \
+          --hpus 1 --junitxml=hpu_test_fsdp-fabric-results.xml
+        python -m pytest -sv tests/test_pytorch/strategies/test_fsdp.py \
+          --hpus 1 --junitxml=hpu_test_fsdp-results.xml
+      env:
+        PT_HPU_LAZY_MODE: 0
+      displayName: 'HPU FSDP tests'
 
     - bash: |
         python -m pytest -sv tests/test_pytorch/strategies/test_deepspeed.py \
@@ -172,13 +172,13 @@ jobs:
           tests/test_pytorch/test_profiler.py
       displayName: 'Multi card(2) HPU test'
 
-    # - bash: |
-    #     bash tests/run_standalone_tests.sh --hpus 2 -f \
-    #       tests/test_fabric/test_fsdp.py
-    #   env:
-    #     PT_HPU_LAZY_MODE: 0
-    #   condition: or(eq(variables['HABANA_VISIBLE_MODULES'], '4,5'), eq(variables['HABANA_VISIBLE_MODULES'], '6,7'))
-    #   displayName: 'FSDP Fabric Multi card(2) HPU test'
+    - bash: |
+        bash tests/run_standalone_tests.sh --hpus 2 -f \
+          tests/test_fabric/test_fsdp.py
+      env:
+        PT_HPU_LAZY_MODE: 0
+      condition: or(eq(variables['HABANA_VISIBLE_MODULES'], '4,5'), eq(variables['HABANA_VISIBLE_MODULES'], '6,7'))
+      displayName: 'FSDP Fabric Multi card(2) HPU test'
 
     - bash: pip install ".[examples]"
       condition: or(eq(variables['HABANA_VISIBLE_MODULES'], '4,5'), eq(variables['HABANA_VISIBLE_MODULES'], '6,7'))