Skip to content

Commit

Permalink
Change CI test_posix.sh config to run fx_importer linalg/tosa/stablehlo
Browse files Browse the repository at this point in the history
- fix bug: main.py: error: argument -c/--config: invalid choice: 'linalg'
 (choose from 'native_torch', 'torchscript', 'lazy_tensor_core', 'onnx',
  'onnx_tosa', 'fx_importer', 'fx_importer_stablehlo', 'fx_importer_tosa')
  • Loading branch information
AmosLewis committed Aug 21, 2024
1 parent 995be98 commit 54499ff
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 16 deletions.
20 changes: 6 additions & 14 deletions build_tools/ci/test_posix.sh
Original file line number Diff line number Diff line change
Expand Up @@ -8,24 +8,16 @@ torch_version="${1:-unknown}"

export PYTHONPATH="$repo_root/build/tools/torch-mlir/python_packages/torch_mlir:$repo_root/projects/pt1"

echo "::group::Run Linalg e2e integration tests"
python -m e2e_testing.main --config=linalg -v
echo "::group::Run fx_importer Linalg e2e integration tests"
python -m e2e_testing.main --config=fx_importer -v
echo "::endgroup::"

echo "::group::Run make_fx + TOSA e2e integration tests"
python -m e2e_testing.main --config=make_fx_tosa -v
echo "::group::Run fx_importer Tosa e2e integration tests"
python -m e2e_testing.main --config=fx_importer_tosa -v
echo "::endgroup::"

echo "::group::Run TOSA e2e integration tests"
python -m e2e_testing.main --config=tosa -v
echo "::endgroup::"

echo "::group::Run Stablehlo e2e integration tests"
python -m e2e_testing.main --config=stablehlo -v
echo "::endgroup::"

echo "::group::Run ONNX e2e integration tests"
python -m e2e_testing.main --config=onnx -v
echo "::group::Run fx_importer Stablehlo e2e integration tests"
python -m e2e_testing.main --config=fx_importer_stablehlo -v
echo "::endgroup::"

case $torch_version in
Expand Down
2 changes: 0 additions & 2 deletions projects/pt1/e2e_testing/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,8 +68,6 @@ def _get_argparse():
"native_torch",
"torchscript",
"lazy_tensor_core",
"onnx",
"onnx_tosa",
"fx_importer",
"fx_importer_stablehlo",
"fx_importer_tosa",
Expand Down

0 comments on commit 54499ff

Please sign in to comment.