Fix pythonpath for e2etesting

pt2_phase1
Stella Laurenzo 2023-11-02 16:16:34 -07:00
parent 8069d8da89
commit c4e41d6742
2 changed files with 2 additions and 1 deletions

View File

@ -287,7 +287,7 @@ function test_in_tree() {
cmake --build /main_checkout/torch-mlir/build --target check-torch-mlir-all
cd /main_checkout/torch-mlir/
export PYTHONPATH="/main_checkout/torch-mlir/build/tools/torch-mlir/python_packages/torch_mlir"
export PYTHONPATH="/main_checkout/torch-mlir/build/tools/torch-mlir/python_packages/torch_mlir:/main_checkout/torch-mlir/projects/pt1"
case $torch_version in
nightly)

View File

@ -359,6 +359,7 @@ Torch-MLIR has two types of tests:
Alternatively, you can run the tests via Python directly:
```shell
cd projects/pt1
python -m e2e_testing.main -f 'AtenEmbeddingBag'
```