Skip to content

Commit e24f2ba

Browse files
Add job to run NVTabular unit tests on GPU
1 parent bc65f6e commit e24f2ba

2 files changed

Lines changed: 15 additions & 3 deletions

File tree

.github/workflows/merlin.yml

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,4 +32,11 @@ jobs:
3232
name: "NVTabular (CPU)"
3333
uses: ./.github/workflows/tox.yml
3434
with:
35-
env: test-nvtabular-cpu
35+
env: test-nvtabular
36+
37+
nvtabular-gpu:
38+
runs-on: 1GPU
39+
name: "NVTabular (GPU)"
40+
uses: ./.github/workflows/tox.yml
41+
with:
42+
env: test-nvtabular

tox.ini

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,13 @@ commands =
5252
python -m pytest --cov-report term --cov merlin -rxs tests/unit
5353

5454

55-
[testenv:test-nvtabular-cpu]
56-
passenv=GIT_COMMIT
55+
[testenv:test-nvtabular]
56+
setenv =
57+
TF_GPU_ALLOCATOR=cuda_malloc_async
58+
passenv =
59+
GIT_COMMIT
60+
OPAL_PREFIX
61+
CUDA_VISIBLE_DEVICES
5762
sitepackages=true
5863
allowlist_externals = git
5964
deps =

0 commit comments

Comments
 (0)