Enable ipex xpu op

.github/workflows/package.yml

@@ -53,42 +53,42 @@ jobs:
           # - "3.13"
     steps:
       - uses: actions/checkout@v4
-      
+
       - name: Checkout bitsandbytes
         uses: actions/checkout@v4
         with:
           repository: bitsandbytes-foundation/bitsandbytes
           path: bitsandbytes
-      
+
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
           cache: pip
-      
+
       - name: Install build tools
         run: |
           sudo apt-get update
           sudo apt-get install -y build-essential cmake
-      
+
       - name: Compile bitsandbytes with CPU backend
         run: |
           cd bitsandbytes
           cmake -DCOMPUTE_BACKEND=cpu -S . && make
           cd ..
-          
+
       - name: Download build artifacts
         uses: actions/download-artifact@v4
         with:
           name: dist
           path: dist/
-          
+
       - name: Install dependencies and built package
         run: |
           python -m pip install --upgrade pip
           pip install ./bitsandbytes
           pip install dist/*.whl
-          
+
       - name: Test import works
         run: |
           python -c "
@@ -98,7 +98,7 @@ jobs:
           print('✅ bitsandbytes_intel import successful')
           print('✅ All imports successful - no XPU operations tested, as for that we would need to configure the XPU runner..')
           "
-      
+
       # - name: Test with pytest
       #   run: pytest
 

docker/Dockerfile

@@ -35,14 +35,13 @@ RUN pip install --no-cache-dir \
     ruff
 
 # Then install PyTorch-dependent packages with constraint to use existing torch
-RUN pip install --no-cache-dir \
-    --extra-index-url https://download.pytorch.org/whl/xpu \
-    -C torch==2.6.0+xpu \
-    transformers \
-    accelerate \
-    bitsandbytes
+RUN pip install transformers accelerate bitsandbytes
+
+# Copy the bitsandbytes-intel repository into /workspace/src/bnb and install it.
+COPY .. ${WORKSPACE}/src/bnb
+RUN cd ${WORKSPACE}/src/bnb && pip install .
 
 COPY --chmod=755 docker/entrypoint.sh /entrypoint.sh
 ENTRYPOINT ["/entrypoint.sh"]
 
-CMD ["sleep", "infinity"]
+CMD ["sleep", "infinity"]

docker/entrypoint.sh

@@ -4,4 +4,4 @@ set -euo pipefail
 pip install --no-deps -e /workspace/src/bnb
 pip install --no-deps -e /workspace/src/bnb_intel
 
-exec "$@"
+exec "$@"