@@ -83,7 +83,6 @@ tvm_multilib_tsim = 'build/libvta_tsim.so, ' +
8383
8484// command to start a docker container
8585docker_run = ' docker/bash.sh'
86- pytest_wrapper = ' ./tests/scripts/pytest_wrapper.py'
8786// timeout in minutes
8887max_time = 240
8988
@@ -256,14 +255,14 @@ def ci_setup(image) {
256255
257256def python_unittest (image ) {
258257 sh (
259- script : " ${ docker_run} ${ image} ${ pytest_wrapper } ./tests/scripts/task_python_unittest.sh" ,
258+ script : " ${ docker_run} ${ image} ./tests/scripts/task_python_unittest.sh" ,
260259 label : ' Run Python unit tests' ,
261260 )
262261}
263262
264263def fsim_test (image ) {
265264 sh (
266- script : " ${ docker_run} ${ image} ${ pytest_wrapper } ./tests/scripts/task_python_vta_fsim.sh" ,
265+ script : " ${ docker_run} ${ image} ./tests/scripts/task_python_vta_fsim.sh" ,
267266 label : ' Run VTA tests in FSIM' ,
268267 )
269268}
@@ -460,11 +459,11 @@ stage('Test') {
460459 label : ' Run Java unit tests' ,
461460 )
462461 sh (
463- script : " ${ docker_run} ${ ci_gpu} ${ pytest_wrapper } ./tests/scripts/task_python_unittest_gpuonly.sh" ,
462+ script : " ${ docker_run} ${ ci_gpu} ./tests/scripts/task_python_unittest_gpuonly.sh" ,
464463 label : ' Run Python GPU unit tests' ,
465464 )
466465 sh (
467- script : " ${ docker_run} ${ ci_gpu} ${ pytest_wrapper } ./tests/scripts/task_python_integration_gpuonly.sh" ,
466+ script : " ${ docker_run} ${ ci_gpu} ./tests/scripts/task_python_integration_gpuonly.sh" ,
468467 label : ' Run Python GPU integration tests' ,
469468 )
470469 }
@@ -487,7 +486,7 @@ stage('Test') {
487486 timeout(time : max_time, unit : ' MINUTES' ) {
488487 ci_setup(ci_cpu)
489488 sh (
490- script : " ${ docker_run} ${ ci_cpu} ${ pytest_wrapper } ./tests/scripts/task_python_integration.sh" ,
489+ script : " ${ docker_run} ${ ci_cpu} ./tests/scripts/task_python_integration.sh" ,
491490 label : ' Run CPU integration tests' ,
492491 )
493492 }
@@ -512,7 +511,7 @@ stage('Test') {
512511 python_unittest(ci_cpu)
513512 fsim_test(ci_cpu)
514513 sh (
515- script : " ${ docker_run} ${ ci_cpu} ${ pytest_wrapper } ./tests/scripts/task_python_vta_tsim.sh" ,
514+ script : " ${ docker_run} ${ ci_cpu} ./tests/scripts/task_python_vta_tsim.sh" ,
516515 label : ' Run VTA tests in TSIM' ,
517516 )
518517 }
@@ -536,7 +535,7 @@ stage('Test') {
536535 ci_setup(ci_i386)
537536 python_unittest(ci_i386)
538537 sh (
539- script : " ${ docker_run} ${ ci_i386} ${ pytest_wrapper } ./tests/scripts/task_python_integration_i386only.sh" ,
538+ script : " ${ docker_run} ${ ci_i386} ./tests/scripts/task_python_integration_i386only.sh" ,
540539 label : ' Run i386 integration tests' ,
541540 )
542541 fsim_test(ci_i386)
@@ -585,7 +584,7 @@ stage('Test') {
585584 timeout(time : max_time, unit : ' MINUTES' ) {
586585 ci_setup(ci_gpu)
587586 sh (
588- script : " ${ docker_run} ${ ci_gpu} ${ pytest_wrapper } ./tests/scripts/task_python_topi.sh" ,
587+ script : " ${ docker_run} ${ ci_gpu} ./tests/scripts/task_python_topi.sh" ,
589588 label : ' Run TOPI tests' ,
590589 )
591590 }
@@ -608,7 +607,7 @@ stage('Test') {
608607 timeout(time : max_time, unit : ' MINUTES' ) {
609608 ci_setup(ci_gpu)
610609 sh (
611- script : " ${ docker_run} ${ ci_gpu} ${ pytest_wrapper } ./tests/scripts/task_python_frontend.sh 1" ,
610+ script : " ${ docker_run} ${ ci_gpu} ./tests/scripts/task_python_frontend.sh 1" ,
612611 label : ' Run Python frontend tests (shard 1)' ,
613612 )
614613 }
@@ -631,7 +630,7 @@ stage('Test') {
631630 timeout(time : max_time, unit : ' MINUTES' ) {
632631 ci_setup(ci_gpu)
633632 sh (
634- script : " ${ docker_run} ${ ci_gpu} ${ pytest_wrapper } ./tests/scripts/task_python_frontend.sh 2" ,
633+ script : " ${ docker_run} ${ ci_gpu} ./tests/scripts/task_python_frontend.sh 2" ,
635634 label : ' Run Python frontend tests (shard 2)' ,
636635 )
637636 }
@@ -654,7 +653,7 @@ stage('Test') {
654653 timeout(time : max_time, unit : ' MINUTES' ) {
655654 ci_setup(ci_cpu)
656655 sh (
657- script : " ${ docker_run} ${ ci_cpu} ${ pytest_wrapper } ./tests/scripts/task_python_frontend_cpu.sh" ,
656+ script : " ${ docker_run} ${ ci_cpu} ./tests/scripts/task_python_frontend_cpu.sh" ,
658657 label : ' Run Python frontend tests' ,
659658 )
660659 }
0 commit comments