@@ -3,8 +3,14 @@ python -c "import neural_compressor as nc"
33test_case=" run 3x Torch Habana FP8"
44echo " ${test_case} "
55
6+ echo " ##[section]Run import check"
7+ set -e
8+ python -c " import neural_compressor.torch"
9+ python -c " import neural_compressor.common"
10+ echo " ##[section]import check pass"
11+
612# install requirements
7- echo " set up UT env..."
13+ echo " ##[group] set up UT env..."
814export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
915sed -i ' /^intel_extension_for_pytorch/d'
1016sed -i ' /^auto_round/d'
@@ -13,6 +19,7 @@ pip install -r /neural-compressor/test/3x/torch/requirements.txt
1319pip install pytest-cov
1420pip install pytest-html
1521pip install pytest-html-merger
22+ echo " ##[endgroup]"
1623pip list
1724
1825export COVERAGE_RCFILE=/neural-compressor/.azure-pipelines/scripts/ut/3x/coverage.3x_pt_fp8
@@ -28,6 +35,18 @@ pytest --cov="${inc_path}" -vs --disable-warnings --html=report_2.html --self-co
2835pytest --cov=" ${inc_path} " 2>&1 | tee -a ${ut_log_name}
2936pytest --cov=" ${inc_path} " 2>&1 | tee -a ${ut_log_name}
3037
38+ # Below folder contains some special configuration for pytest so we need to enter the path and run it separately
39+ cd /neural-compressor/test/3x/torch/algorithms/fp8_quant
40+ pytest --cov=" ${inc_path} " . 2>&1 | tee -a ${ut_log_name}
41+ cp .coverage ${LOG_DIR} /.coverage.algo_fp8
42+ cd - && mv /neural-compressor/test/3x/torch/algorithms/fp8_quant/* .html .
43+
44+ # Below folder contains some special configuration for pytest so we need to enter the path and run it separately
45+ cd /neural-compressor/test/3x/torch/quantization/fp8_quant
46+ pytest --cov=" ${inc_path} " . 2>&1 | tee -a ${ut_log_name}
47+ cp .coverage ${LOG_DIR} /.coverage.quant_fp8
48+ cd - && mv /neural-compressor/test/3x/torch/quantization/fp8_quant/* .html .
49+
3150mkdir -p report && mv * .html report
3251pytest_html_merger -i ./report -o ./report.html
3352cp report.html ${LOG_DIR} /
4059
4160# if ut pass, collect the coverage file into artifacts
4261cp .coverage ${LOG_DIR} /.coverage
62+ cd ${LOG_DIR}
63+ coverage combine .coverage.*
4364
4465echo " UT finished successfully! "
0 commit comments