@@ -28,9 +28,7 @@ commands =
28
28
; Runs in: Github Actions
29
29
; Runs GPU-based tests.
30
30
allowlist_externals =
31
- sh
32
31
horovodrun
33
- examples/usecases/multi-gpu/install_distributed_embeddings.sh
34
32
deps =
35
33
-rrequirements/test.txt
36
34
passenv =
@@ -43,12 +41,12 @@ commands =
43
41
python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/core.git@{posargs:main}
44
42
python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/dataloader.git@{posargs:main}
45
43
python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/nvtabular.git@{posargs:main}
46
- # Install distributed embeddings and check build
47
- sh examples/usecases/multi-gpu/install_distributed_embeddings.sh
48
44
# Run multi-gpu tests marked with `horovod` marker
49
45
horovodrun -np 2 sh examples/usecases/multi-gpu/hvd_wrapper.sh pytest -m horovod -rxs tests/unit
50
46
51
47
[testenv:py38-horovod-cpu]
48
+ allowlist_externals =
49
+ bash
52
50
setenv =
53
51
HOROVOD_WITH_MPI =1
54
52
HOROVOD_WITH_TENSORFLOW =1
@@ -61,7 +59,7 @@ commands =
61
59
{envdir}/env/bin/python -m pip install horovod --no-cache-dir
62
60
{envdir}/env/bin/horovodrun --check-build
63
61
# Install distributed embeddings and check build
64
- sh examples/usecases/multi-gpu/install_distributed_embeddings.sh
62
+ bash examples/usecases/multi-gpu/install_distributed_embeddings.sh {envtmpdir}
65
63
# Install Merlin packages
66
64
{envdir}/env/bin/python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/core.git
67
65
{envdir}/env/bin/python -m pip install --upgrade git+https://github.com/NVIDIA-Merlin/dataloader.git
0 commit comments