Skip to content

GitLab

Commit 33de83c4 authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

Fix stages for onnx run

parent 41c4398d
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
Jenkinsfile
View file @ 33de83c4
...@@ -156,7 +156,7 @@ def onnxnode(name, body) { ...@@ -156,7 +156,7 @@ def onnxnode(name, body) {
} }
rocmtest onnxbuild onnxnode('anygpu') { cmake_build -> rocmtest onnxbuild: onnxnode('anygpu') { cmake_build ->
stage("Onnxruntime Build and Install") { stage("Onnxruntime Build and Install") {
sh ''' sh '''
apt install half apt install half
...@@ -182,15 +182,17 @@ def onnxtestnode(name, body) { ...@@ -182,15 +182,17 @@ def onnxtestnode(name, body) {
} }
} }
rocmtest onnxtests onnxtestnode('anygpu') {cmake_build -> rocmtest onnxtests: onnxtestnode('anygpu') {cmake_build ->
stage("Onnxruntime Unit tests") { stage("Onnxruntime Unit tests") {
sh ''' sh '''
cd /onnxruntime && ./test_onnxrt_unit_tests.sh cd /onnxruntime && ./test_onnxrt_unit_tests.sh
''' '''
} }
stage("Onnxruntime Parity tests") { }, onnx_parity: onnxtestnode('anygpu') {cmake_build ->
stage("Parity tests Stable Pytorch") {
sh ''' sh '''
cd /onnxruntime && ./test_onnxrt_parity_tests.sh cd /onnxruntime && ./test_onnxrt_parity_tests.sh
''' '''
} }
} }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment