From e24f2ba7fb40306672768f403b271652630970d8 Mon Sep 17 00:00:00 2001
From: Oliver Holworthy <oholworthy@nvidia.com>
Date: Mon, 17 Apr 2023 13:11:45 +0100
Subject: [PATCH] Add job to run NVTabular unit tests on GPU

---
 .github/workflows/merlin.yml | 9 ++++++++-
 tox.ini                      | 9 +++++++--
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/merlin.yml b/.github/workflows/merlin.yml
index cbb68af28..534d65773 100644
--- 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
index 91ce69f1c..2ae67e1ee 100644
--- 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 =