future

Lightning-AI · Jan 8, 2025 · 165d38a · 165d38a
1 parent 5e9009b
commit 165d38a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/.azure/gpu-tests-fabric.yml b/.azure/gpu-tests-fabric.yml
@@ -117,10 +117,9 @@ jobs:
         displayName: "Install package & dependencies"
 
       - bash: |
-          # get pytorch version
-          PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
           # FixMe: uninstall bitsandbytes for pytorch 2.6 as it is not compatible with `triton.ops`
-          if [[ "${PYTORCH_VERSION}" == "2.6.0" ]]; then
+          # if the job name include key word future, then uninstall bitsandbytes
+          if [[ "$(Agent.JobName)" == *"future"* ]]; then
             pip uninstall -y bitsandbytes
           else
             python -c "import bitsandbytes"

diff --git a/.azure/gpu-tests-pytorch.yml b/.azure/gpu-tests-pytorch.yml
@@ -131,10 +131,9 @@ jobs:
         displayName: "Drop PL for LAI"
 
       - bash: |
-          # get pytorch version
-          PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
           # FixMe: uninstall bitsandbytes for pytorch 2.6 as it is not compatible with `triton.ops`
-          if [[ "${PYTORCH_VERSION}" == "2.6.0" ]]; then
+          # if the job name include key word future, then uninstall bitsandbytes
+          if [[ "$(Agent.JobName)" == *"future"* ]]; then
             pip uninstall -y bitsandbytes
           else
             python -c "import bitsandbytes"