mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-27 03:11:28 +00:00
19 lines
718 B
Bash
Executable file
19 lines
718 B
Bash
Executable file
#!/bin/bash
|
|
|
|
cur_dir=$(basename `pwd`)
|
|
|
|
if [[ ${cur_dir} != "RelWithDebInfo" ]]
|
|
then
|
|
echo "Going to build folder (aka build/Linux/RelWithDebInfo)"
|
|
cd build/Linux/RelWithDebInfo
|
|
fi
|
|
|
|
echo "Exporting PYTHONPATH to use build dir as onnxruntime package"
|
|
export PYTHONPATH=$(pwd)
|
|
|
|
echo "Copying PyTorch frontend source-code to build folder"
|
|
cp -Rf ../../../orttraining/orttraining/python/training/* ../../../build/Linux/RelWithDebInfo/onnxruntime/training/
|
|
|
|
echo "Running Flexible API (ORTModule)"
|
|
python ../../../orttraining/orttraining/test/python/orttraining_test_ortmodule_torch_lightning_basic.py --help
|
|
python ../../../orttraining/orttraining/test/python/orttraining_test_ortmodule_torch_lightning_basic.py $@
|