Skip to content

Commit

Permalink
Add Jenkinsfile (#167)
Browse files Browse the repository at this point in the history
* add jenkinsfile

* add build step

* rename jenkinsfile

* add timeout

Co-authored-by: Karl Higley <kmhigley@gmail.com>
  • Loading branch information
AyodeAwe and karlhigley authored Nov 22, 2022
1 parent 57ab7e0 commit 72f967a
Showing 1 changed file with 44 additions and 0 deletions.
44 changes: 44 additions & 0 deletions ci/pr.gpu.Jenkinsfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
pipeline {
agent {
docker {
image 'nvcr.io/nvstaging/merlin/merlin-ci-runner-wrapper'
label 'merlin_gpu'
registryCredentialsId 'jawe-nvcr-io'
registryUrl 'https://nvcr.io'
args "--runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all"
}
}

options {
buildDiscarder(logRotator(numToKeepStr: '10'))
ansiColor('xterm')
disableConcurrentBuilds(abortPrevious: true)
}

stages {
stage("test-gpu") {
options {
timeout(time: 60, unit: 'MINUTES', activity: true)
}
steps {
sh """#!/bin/bash
#set -e
printenv

rm -rf $HOME/.cudf/
export TF_MEMORY_ALLOCATION="0.1"
export CUDA_VISIBLE_DEVICES=0,1
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION='python'
export MKL_SERVICE_FORCE_INTEL=1
export NR_USER=true

tox -re test-gpu

rm -rf "nvtabular-$GIT_COMMIT"
rm -rf "models-$GIT_COMMIT"
rm -rf "systems-$GIT_COMMIT"
"""
}
}
}
}

0 comments on commit 72f967a

Please sign in to comment.