Skip to content

Commit 40759cd

Browse files
[CI] use new docker with uv binary to fix shim/uv didn't exist (#1724)
* [CI] use new docker without pip uv * fix ipex error
1 parent 9fba8bc commit 40759cd

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

.github/workflows/unit_tests.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ jobs:
205205
- list-test-files
206206
if: needs.list-test-files.outputs.torch-files != '[]' || needs.list-test-files.outputs.transformers-files != '[]'
207207
container:
208-
image: ${{ needs.check-vm.outputs.ip }}:5000/nvidia/cuda:${{ needs.check-vm.outputs.cuda_version }}-ubuntu22.04
208+
image: ${{ needs.check-vm.outputs.ip }}:5000/nvidia/cuda:${{ needs.check-vm.outputs.cuda_version }}-ubuntu22.04_0822
209209
options: --device /dev/dri --ipc=host --runtime=nvidia --gpus all
210210
volumes:
211211
- /dev/dri/by-path:/dev/dri/by-path
@@ -327,7 +327,7 @@ jobs:
327327
- check-vm
328328
runs-on: [ self-hosted, xeon5 ]
329329
container:
330-
image: ${{ needs.check-vm.outputs.ip }}:5000/nvidia/cuda:${{ needs.check-vm.outputs.cuda_version }}-ubuntu22.04
330+
image: ${{ needs.check-vm.outputs.ip }}:5000/nvidia/cuda:${{ needs.check-vm.outputs.cuda_version }}-ubuntu22.04_0822
331331
options: --device /dev/dri --ipc=host --runtime=nvidia --gpus all
332332
volumes:
333333
- /dev/dri/by-path:/dev/dri/by-path

tests/test_packable.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from gptqmodel.nn_modules.qlinear.exllama import ExllamaQuantLinear # noqa: E402
99
from gptqmodel.nn_modules.qlinear.exllama_eora import ExllamaEoraQuantLinear
1010
from gptqmodel.nn_modules.qlinear.exllamav2 import ExllamaV2QuantLinear # noqa: E402
11-
from gptqmodel.nn_modules.qlinear.ipex import IPEXQuantLinear # noqa: E402
11+
from gptqmodel.nn_modules.qlinear.ipex import IPEXQuantLinear, HAS_IPEX # noqa: E402
1212
from gptqmodel.nn_modules.qlinear.marlin import MarlinQuantLinear # noqa: E402
1313
from gptqmodel.nn_modules.qlinear.torch import TorchQuantLinear # noqa: E402
1414
from gptqmodel.nn_modules.qlinear.tritonv2 import TritonV2QuantLinear # noqa: E402
@@ -59,6 +59,8 @@ def setUpClass(cls):
5959
]
6060
)
6161
def test_post_init(self, backend: BACKEND, equal: Dict[str, bool]):
62+
if backend == BACKEND.IPEX and not HAS_IPEX:
63+
self.skipTest("IPEX is not available")
6264
model = GPTQModel.load(self.model_id, backend=backend, device_map="auto")
6365
model = convert_gptq_v2_to_v1_format(model, model.quantize_config, self.QLINEAR_DICT[backend])
6466

0 commit comments

Comments
 (0)