test

Author: matthewdouglas

.github/workflows/tests.yml

@@ -231,12 +231,14 @@ jobs:
         torch_version: ["2.6.0"]
     runs-on:
       group: bandb-itac-bmemr-gaudi3-1gaudi
-    # container:
-    #   image: vault.habana.ai/gaudi-docker/1.20.1/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
-    #   options: --runtime=habana --shm-size=64G --env HABANA_VISIBLE_DEVICES --env HABANA_VISIBLE_MODULES
-    #   env:
-    #     OMPI_MCA_btl_vader_single_copy_mechanism: none
-    #     BNB_TEST_DEVICE: hpu
+    env:
+      BNB_TEST_DEVICE: hpu
+    container:
+      image: vault.habana.ai/gaudi-docker/1.20.1/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
+      options: --runtime=habana --shm-size=64G --env HABANA_VISIBLE_DEVICES --env HABANA_VISIBLE_MODULES
+      env:
+        OMPI_MCA_btl_vader_single_copy_mechanism: none
+        BNB_TEST_DEVICE: hpu
     steps:
       - name: Show system information
         run: |
@@ -248,6 +250,34 @@ jobs:
         run: |
           hl-smi
 
+      - uses: actions/checkout@v4
+
+      - name: Download build artifact
+        uses: actions/download-artifact@v4
+        with:
+          name: lib_cpu_ubuntu-22.04_x86_64
+          path: bitsandbytes/
+          merge-multiple: true
+
+      - name: Show installed packages
+        run: pip list
+
+      - name: Install dependencies
+        run: |
+          pip install -e ".[test]"
+          pip install pytest-cov
+
+      - name: Show installed packages
+        run: pip list
+
+      - name: Show environment information
+        run: |
+          python -m torch.utils.collect_env
+          python -m bitsandbytes
+
+      - name: Run tests
+        run: pytest --durations=100
+
   test-xpu:
     if: github.repository == 'bitsandbytes-foundation/bitsandbytes'
     needs: build-cpu