We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bc65f6e commit e24f2baCopy full SHA for e24f2ba
2 files changed
.github/workflows/merlin.yml
@@ -32,4 +32,11 @@ jobs:
32
name: "NVTabular (CPU)"
33
uses: ./.github/workflows/tox.yml
34
with:
35
- env: test-nvtabular-cpu
+ env: test-nvtabular
36
+
37
+ nvtabular-gpu:
38
+ runs-on: 1GPU
39
+ name: "NVTabular (GPU)"
40
+ uses: ./.github/workflows/tox.yml
41
+ with:
42
tox.ini
@@ -52,8 +52,13 @@ commands =
52
python -m pytest --cov-report term --cov merlin -rxs tests/unit
53
54
55
-[testenv:test-nvtabular-cpu]
56
-passenv=GIT_COMMIT
+[testenv:test-nvtabular]
+setenv =
57
+ TF_GPU_ALLOCATOR=cuda_malloc_async
58
+passenv =
59
+ GIT_COMMIT
60
+ OPAL_PREFIX
61
+ CUDA_VISIBLE_DEVICES
62
sitepackages=true
63
allowlist_externals = git
64
deps =
0 commit comments