Skip to content

Commit 1bd971a

Browse files
authored
Merge branch 'main' into qiming/enale_eplb
2 parents 0110dff + 80bd4c7 commit 1bd971a

File tree

3 files changed

+28
-4
lines changed

3 files changed

+28
-4
lines changed

.github/workflows/ut.yaml

Lines changed: 18 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ jobs:
6464
timeout-minutes: 40
6565
container:
6666
image: localhost:5000/xpu-kernel-ci-image:latest
67-
options: --device /dev/dri -v /dev/dri/by-path:/dev/dri/by-path --privileged
67+
options: --device /dev/dri -v /dev/dri/by-path:/dev/dri/by-path --privileged -v ccache:/root/.ccache -e CCACHE_DIR=/root/.ccache
6868
steps:
6969
- name: Checkout
7070
uses: actions/checkout@8e8c483db84b4bee98b60c0593521ed34d9990e8 # v6.0.1
@@ -74,8 +74,15 @@ jobs:
7474
- name: build & install wheel
7575
run: |
7676
git config --global --add safe.directory "${GITHUB_WORKSPACE}"
77+
export CCACHE_DIR=/root/.ccache
78+
export CCACHE_BASEDIR="${GITHUB_WORKSPACE}"
79+
export CCACHE_NOHASHDIR=1
80+
export CCACHE_COMPILERCHECK=content
81+
ccache -s || true
82+
ccache -p || true
7783
uv pip install -r requirements.txt
78-
MAX_JOB=128 uv pip install -e . -v
84+
MAX_JOBS=128 uv pip install --no-build-isolation -e . -v
85+
ccache -s || true
7986
8087
- name: test
8188
run: |
@@ -98,7 +105,7 @@ jobs:
98105
timeout-minutes: 40
99106
container:
100107
image: localhost:5000/xpu-kernel-ci-image:latest
101-
options: --device /dev/dri -v /dev/dri/by-path:/dev/dri/by-path --privileged
108+
options: --device /dev/dri -v /dev/dri/by-path:/dev/dri/by-path --privileged -v ccache:/root/.ccache -e CCACHE_DIR=/root/.ccache
102109
steps:
103110
- name: Checkout
104111
uses: actions/checkout@8e8c483db84b4bee98b60c0593521ed34d9990e8 # v6.0.1
@@ -108,8 +115,15 @@ jobs:
108115
- name: build & install wheel
109116
run: |
110117
git config --global --add safe.directory "${GITHUB_WORKSPACE}"
118+
export CCACHE_DIR=/root/.ccache
119+
export CCACHE_BASEDIR="${GITHUB_WORKSPACE}"
120+
export CCACHE_NOHASHDIR=1
121+
export CCACHE_COMPILERCHECK=content
122+
ccache -s || true
123+
ccache -p || true
111124
uv pip install -r requirements.txt
112-
MAX_JOBS=128 uv pip install -e . -v
125+
MAX_JOBS=128 uv pip install --no-build-isolation -e . -v
126+
ccache -s || true
113127
114128
- name: test
115129
run: |

Dockerfile.xpu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ RUN apt clean && apt-get update -y && \
1414
curl \
1515
ffmpeg \
1616
git \
17+
ccache \
1718
libsndfile1 \
1819
libsm6 \
1920
libxext6 \

tests/test_fp8_quant.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,15 @@ def seed_everything(seed):
218218

219219
# override pytest parameters when enable mini pytest
220220
MINI_PYTEST_PARAMS = {
221+
"test_per_block_fp8_quant": {
222+
"num_tokens_block_quant": [1],
223+
"hidden_size_block_quant": [64],
224+
"group_size": [32],
225+
},
226+
"test_per_block_mxfp8_quant": {
227+
"num_tokens_block_quant": [1],
228+
"hidden_size_block_quant": [64],
229+
},
221230
"default": {
222231
"num_tokens": [1],
223232
"hidden_size": [1],

0 commit comments

Comments
 (0)