mirror of https://github.com/llvm/torch-mlir
Fix pythonpath for e2etesting
parent
8069d8da89
commit
c4e41d6742
|
@ -287,7 +287,7 @@ function test_in_tree() {
|
||||||
cmake --build /main_checkout/torch-mlir/build --target check-torch-mlir-all
|
cmake --build /main_checkout/torch-mlir/build --target check-torch-mlir-all
|
||||||
|
|
||||||
cd /main_checkout/torch-mlir/
|
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
|
case $torch_version in
|
||||||
nightly)
|
nightly)
|
||||||
|
|
|
@ -359,6 +359,7 @@ Torch-MLIR has two types of tests:
|
||||||
Alternatively, you can run the tests via Python directly:
|
Alternatively, you can run the tests via Python directly:
|
||||||
|
|
||||||
```shell
|
```shell
|
||||||
|
cd projects/pt1
|
||||||
python -m e2e_testing.main -f 'AtenEmbeddingBag'
|
python -m e2e_testing.main -f 'AtenEmbeddingBag'
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue