[LLM] Enable UT workflow logics for LLM (#8243)
* check push connection * enable UT workflow logics for LLM * test fix * add licenses * test fix according to suggestions * test fix * update changes
This commit is contained in:
parent
657ea0ee50
commit
c48d5f7cff
5 changed files with 83 additions and 20 deletions
|
|
@ -27,7 +27,7 @@ jobs:
|
|||
fail-fast: false
|
||||
matrix:
|
||||
os: ["windows-latest"]
|
||||
python-version: ["3.8"]
|
||||
python-version: ["3.9"]
|
||||
steps:
|
||||
- uses: actions/checkout@v2
|
||||
- name: Set up Python ${{ matrix.python-version }}
|
||||
|
|
@ -39,6 +39,7 @@ jobs:
|
|||
python -m pip install --upgrade pip
|
||||
python -m pip install --upgrade setuptools==58.0.4
|
||||
python -m pip install --upgrade wheel
|
||||
|
||||
- name: Run LLM-init test
|
||||
shell: bash
|
||||
run: |
|
||||
|
|
@ -46,3 +47,17 @@ jobs:
|
|||
pip install -i https://pypi.python.org/simple python/llm/dist/bigdl_llm*.whl
|
||||
env:
|
||||
ANALYTICS_ZOO_ROOT: ${{ github.workspace }}
|
||||
|
||||
- name: Run LLM naive installation test
|
||||
shell: bash
|
||||
run: |
|
||||
bash python/llm/dev/release_default_windows.sh default false
|
||||
pip install -i https://pypi.python.org/simple python/llm/dist/bigdl_llm*.whl
|
||||
pip install pytest
|
||||
pip install numpy
|
||||
pip install torch
|
||||
pip install transformers
|
||||
pip install sentencepiece
|
||||
bash python/llm/test/run-llm-basic-tests.sh
|
||||
env:
|
||||
ANALYTICS_ZOO_ROOT: ${{ github.workspace }}
|
||||
|
|
|
|||
|
|
@ -43,10 +43,8 @@ echo "The effective version is: ${bigdl_version}"
|
|||
|
||||
if [ "$platform" == "linux" ]; then
|
||||
verbose_pname="manylinux2010_x86_64"
|
||||
setup_file="setup-linux.py"
|
||||
elif [ "$platform" == "windows" ]; then
|
||||
verbose_pname="win_amd64"
|
||||
setup_file="setup.py"
|
||||
else
|
||||
echo "Unsupported platform"
|
||||
fi
|
||||
|
|
@ -55,36 +53,21 @@ if [ -d "${BIGDL_DIR}/python/llm/dist" ]; then
|
|||
rm -r ${BIGDL_DIR}/python/llm/dist
|
||||
fi
|
||||
|
||||
if [ -d "${BIGDL_DIR}/python/llm/_skbuild" ]; then
|
||||
rm -r ${BIGDL_DIR}/python/llm/_skbuild
|
||||
fi
|
||||
|
||||
if [ -d "${BIGDL_DIR}/python/llm/build" ]; then
|
||||
rm -r ${BIGDL_DIR}/python/llm/build
|
||||
fi
|
||||
|
||||
if [ -d "${BIGDL_DIR}/python/llm/bigdl_llm.egg-info" ]; then
|
||||
rm -r ${BIGDL_DIR}/python/llm/bigdl_llm.egg-info
|
||||
fi
|
||||
|
||||
cd $BIGDL_PYTHON_DIR
|
||||
|
||||
wheel_command="python ${setup_file} bdist_wheel --plat-name ${verbose_pname} --python-tag py3"
|
||||
wheel_command="python setup.py clean --all bdist_wheel --plat-name ${verbose_pname} --python-tag py3"
|
||||
|
||||
echo "Packing python distribution: $wheel_command"
|
||||
${wheel_command}
|
||||
|
||||
if [ -d "${BIGDL_DIR}/python/llm/_skbuild" ]; then
|
||||
rm -r ${BIGDL_DIR}/python/llm/_skbuild
|
||||
fi
|
||||
|
||||
if [ -d "${BIGDL_DIR}/python/llm/build" ]; then
|
||||
rm -r ${BIGDL_DIR}/python/llm/build
|
||||
fi
|
||||
|
||||
if [ -d "${BIGDL_DIR}/python/llm/bigdl_llm.egg-info" ]; then
|
||||
rm -r ${BIGDL_DIR}/python/llm/bigdl_llm.egg-info
|
||||
fi
|
||||
|
||||
if [ ${upload} == true ]; then
|
||||
upload_command="twine upload dist/bigdl_llm-${bigdl_version}-*-${verbose_pname}.whl"
|
||||
echo "Please manually upload with this command: $upload_command"
|
||||
|
|
|
|||
15
python/llm/test/__init__.py
Normal file
15
python/llm/test/__init__.py
Normal file
|
|
@ -0,0 +1,15 @@
|
|||
#
|
||||
# Copyright 2016 The BigDL Authors.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
32
python/llm/test/packaging/test_install.py
Normal file
32
python/llm/test/packaging/test_install.py
Normal file
|
|
@ -0,0 +1,32 @@
|
|||
#
|
||||
# Copyright 2016 The BigDL Authors.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
|
||||
import bigdl.llm
|
||||
import pytest
|
||||
from unittest import TestCase
|
||||
|
||||
|
||||
class Test_LLM_Basics(TestCase):
|
||||
|
||||
def test_naive(self):
|
||||
from bigdl.llm.ggml import quantize
|
||||
from bigdl.llm.utils.common import invalidInputError
|
||||
pass
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
pytest.main([__file__])
|
||||
18
python/llm/test/run-llm-basic-tests.sh
Normal file
18
python/llm/test/run-llm-basic-tests.sh
Normal file
|
|
@ -0,0 +1,18 @@
|
|||
#!/bin/bash
|
||||
|
||||
export ANALYTICS_ZOO_ROOT=${ANALYTICS_ZOO_ROOT}
|
||||
export LLM_HOME=${ANALYTICS_ZOO_ROOT}/python/llm/src
|
||||
export LLM_BASIC_TEST_DIR=${ANALYTICS_ZOO_ROOT}/python/llm/test/packaging
|
||||
|
||||
set -e
|
||||
|
||||
# ipex is not installed here. Any tests needs ipex should be moved to next pytest command.
|
||||
echo "# Start testing"
|
||||
start=$(date "+%s")
|
||||
python -m pytest -s ${LLM_BASIC_TEST_DIR}
|
||||
|
||||
now=$(date "+%s")
|
||||
time=$((now-start))
|
||||
|
||||
echo "Bigdl-llm tests finished"
|
||||
echo "Time used:$time seconds"
|
||||
Loading…
Reference in a new issue