Add job to run NVTabular unit tests on GPU

NVIDIA-Merlin · Apr 17, 2023 · e24f2ba · e24f2ba
1 parent bc65f6e
commit e24f2ba
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/.github/workflows/merlin.yml b/.github/workflows/merlin.yml
@@ -32,4 +32,11 @@ jobs:
     name: "NVTabular (CPU)"
     uses: ./.github/workflows/tox.yml
     with:
-      env: test-nvtabular-cpu
+      env: test-nvtabular
+
+  nvtabular-gpu:
+    runs-on: 1GPU
+    name: "NVTabular (GPU)"
+    uses: ./.github/workflows/tox.yml
+    with:
+      env: test-nvtabular
diff --git a/tox.ini b/tox.ini
@@ -52,8 +52,13 @@ commands =
     python -m pytest --cov-report term --cov merlin -rxs tests/unit
 
 
-[testenv:test-nvtabular-cpu]
-passenv=GIT_COMMIT
+[testenv:test-nvtabular]
+setenv =
+    TF_GPU_ALLOCATOR=cuda_malloc_async
+passenv =
+    GIT_COMMIT
+    OPAL_PREFIX
+    CUDA_VISIBLE_DEVICES
 sitepackages=true
 allowlist_externals = git
 deps =