Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
42 changes: 34 additions & 8 deletions .github/workflows/unit_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,7 @@ env:
MAX_JOBS: 4
RUNNER: 10.0.13.31
XEON5: 10.0.14.249
UV_INDEX_URL: http://10.0.14.249/simple
CUDA_VERSION: 128
TORCH_VERSION: 2.8.0
PYTHON_VERSION: 312 # use 312, no need to recompile libs like flash_attn...
Expand Down Expand Up @@ -235,6 +236,10 @@ jobs:

- name: Print Env
run: |
echo "::group::pyenv versions"
pyenv versions
echo "::endgroup::"

python_version=${{ env.PYTHON_VERSION }}
if [[ "$python_version" != *"."* ]]; then
python_version="${python_version/3/3.}"
Expand All @@ -258,16 +263,22 @@ jobs:
pyenv local $env_name
pyenv activate $env_name

echo "== pyenv =="
echo "::group::pyenv versions"
pyenv versions
echo "::endgroup::"

echo "== python =="
python --version

echo "== nvcc =="
nvcc --version

echo "== torch =="
pip show torch || true
echo "##### pip list #####"
pip show torch

echo "::group::pip list"
pip list
echo "::endgroup::"

- name: Compress dir
run: |
Expand All @@ -292,7 +303,16 @@ jobs:
- name: Compile
if: github.event.inputs.artifact_id == '' && !cancelled()
timeout-minutes: 45
run: python setup.py bdist_wheel
run: |
echo "::group::pyenv version"
pyenv version
which python
python --version
echo "::endgroup::"

echo "::group::compile logs"
python setup.py bdist_wheel
echo "::endgroup::"

# - name: Test install
# if: github.event.inputs.artifact_id == '' && !cancelled()
Expand Down Expand Up @@ -388,16 +408,22 @@ jobs:
pyenv local $env_name
pyenv activate $env_name

echo "== pyenv =="
echo "::group::pyenv versions"
pyenv versions
echo "::endgroup::"

echo "== python =="
python --version

echo "== nvcc =="
nvcc --version

echo "== torch =="
pip show torch
echo "== pip list =="

echo "::group::pip list"
pip list
echo "::endgroup::"

# - name: Install requirements
# run: |
Expand Down Expand Up @@ -454,7 +480,7 @@ jobs:

- name: Install wheel
run: |
uv pip install -U transformers
uv pip install -U transformers tabulate
if [ "${{ matrix.test_script }}" == "test_perplexity" ] || \
[ "${{ matrix.test_script }}" == "test_inference_speed" ] || \
[ "${{ matrix.test_script }}" == "test_q4_bitblas" ] || \
Expand Down Expand Up @@ -607,7 +633,7 @@ jobs:
needs:
- check-vm
- list-test-files
if: (github.event.inputs.test_names == '' || contains(github.event.inputs.test_names, 'apple') || contains(github.event.inputs.test_names, 'mlx') ) && (needs.list-test-files.outputs.m4-files != '' && needs.list-test-files.outputs.m4-files != '[]') && !cancelled()
if: false && (github.event.inputs.test_names == '' || contains(github.event.inputs.test_names, 'apple') || contains(github.event.inputs.test_names, 'mlx') ) && (needs.list-test-files.outputs.m4-files != '' && needs.list-test-files.outputs.m4-files != '[]') && !cancelled()
strategy:
fail-fast: false
matrix:
Expand Down