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
|
||||
|
||||
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)
|
||||
|
|
|
@ -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'
|
||||
```
|
||||
|
||||
|
|
Loading…
Reference in New Issue