mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
af0750ba1b
Modify gradle build so artifactID has _gpu for GPU builds. Pass USE_CUDA flag on CUDA build Adjust publishing pipelines to extract POM from a correct path. Co-Authored-By: @Craigacp
36 lines
968 B
Bash
Executable file
36 lines
968 B
Bash
Executable file
#!/bin/bash
|
|
|
|
# This is for testing GPU final jar on Linux
|
|
set -e -o -x
|
|
|
|
while getopts r:v: parameter_Option
|
|
do case "${parameter_Option}"
|
|
in
|
|
r) BINARY_DIR=${OPTARG};;
|
|
v) VERSION_NUMBER=${OPTARG};;
|
|
esac
|
|
done
|
|
|
|
EXIT_CODE=1
|
|
|
|
uname -a
|
|
|
|
cd $BINARY_DIR/final-jar
|
|
|
|
mkdir test
|
|
|
|
echo "Directories created"
|
|
echo "Library path:" $LD_LIBRARY_PATH
|
|
|
|
pushd test
|
|
jar xf $BINARY_DIR/final-jar/testing.jar
|
|
popd
|
|
|
|
wget https://oss.sonatype.org/service/local/repositories/releases/content/org/junit/platform/junit-platform-console-standalone/1.6.2/junit-platform-console-standalone-1.6.2.jar -P ./
|
|
wget https://oss.sonatype.org/service/local/repositories/google-releases/content/com/google/protobuf/protobuf-java/3.9.2/protobuf-java-3.9.2.jar -P ./
|
|
java -DUSE_CUDA=1 -jar ./junit-platform-console-standalone-1.6.2.jar -cp .:./test:./protobuf-java-3.9.2.jar:./onnxruntime_gpu-${VERSION_NUMBER}.jar --scan-class-path --fail-if-no-tests --disable-banner
|
|
|
|
EXIT_CODE=$?
|
|
|
|
set -e
|
|
exit $EXIT_CODE
|