Skip to content

Commit 152b625

Browse files
AyodeAwekarlhigley
andauthored
Add Jenkinsfile (#49)
* add jenkinsfile * add gcp label --------- Co-authored-by: Karl Higley <kmhigley@gmail.com>
1 parent 7b10d37 commit 152b625

1 file changed

Lines changed: 44 additions & 0 deletions

File tree

ci/pr.gpu.Jenkinsfile

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
1+
pipeline {
2+
agent {
3+
docker {
4+
image 'nvcr.io/nvstaging/merlin/merlin-ci-runner-wrapper'
5+
label 'merlin_gpu_gcp || merlin_gpu'
6+
registryCredentialsId 'jawe-nvcr-io'
7+
registryUrl 'https://nvcr.io'
8+
args "--runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all"
9+
}
10+
}
11+
12+
options {
13+
buildDiscarder(logRotator(numToKeepStr: '10'))
14+
ansiColor('xterm')
15+
disableConcurrentBuilds(abortPrevious: true)
16+
}
17+
18+
stages {
19+
stage("test-gpu") {
20+
options {
21+
timeout(time: 60, unit: 'MINUTES', activity: true)
22+
}
23+
steps {
24+
sh """#!/bin/bash
25+
set -e
26+
printenv
27+
28+
rm -rf $HOME/.cudf/
29+
export TF_MEMORY_ALLOCATION="0.1"
30+
export CUDA_VISIBLE_DEVICES="2,3"
31+
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION='python'
32+
export MKL_SERVICE_FORCE_INTEL=1
33+
export NR_USER=true
34+
35+
tox -re test-gpu
36+
37+
rm -rf "nvtabular-$GIT_COMMIT"
38+
rm -rf "models-$GIT_COMMIT"
39+
rm -rf "systems-$GIT_COMMIT"
40+
"""
41+
}
42+
}
43+
}
44+
}

0 commit comments

Comments
 (0)