File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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+ }
You can’t perform that action at this time.
0 commit comments