File tree Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ case $torch_version in
1515 ;;
1616 stable)
1717 echo " ::group::installing stable torch"
18- python3 -m pip install --no-cache-dir torch==2.8 .0 torchvision --index-url https://download.pytorch.org/whl/cpu
18+ python3 -m pip install --no-cache-dir torch==2.9 .0 torchvision --index-url https://download.pytorch.org/whl/cpu
1919 python3 -m pip install --no-cache-dir -r $repo_root /build-requirements.txt
2020 echo " ::endgroup::"
2121 ;;
Original file line number Diff line number Diff line change @@ -4,10 +4,17 @@ set -eu -o errtrace
44
55this_dir=" $( cd $( dirname $0 ) && pwd) "
66repo_root=" $( cd $this_dir /../.. && pwd) "
7- torch_version=" ${1:- unknown} "
87
98export PYTHONPATH=" $repo_root /build/tools/torch-mlir/python_packages/torch_mlir:$repo_root /projects/pt1"
109
1110echo " ::group::Run TOSA LINALG e2e integration tests"
1211python3 -m e2e_testing.main --config=fx_importer_tosa_linalg -v
1312echo " ::endgroup::"
13+
14+ echo " ::group::Run FxImporter e2e integration tests"
15+ python3 -m e2e_testing.main --config=fx_importer -s
16+ echo " ::endgroup::"
17+
18+ echo " ::group::Run FxImporter TOSA e2e integration tests"
19+ python3 -m e2e_testing.main --config=fx_importer_tosa -v
20+ echo " ::endgroup::"
You can’t perform that action at this time.
0 commit comments