torchrec
torchrec copied to clipboard
Gpu smoketest
until we have 8 gpus, GPU unit tests are mostly skipped - Use smoke test from above to get some GPU signal.
# conda run -n build_binary
# python -m pip install pytest
# conda run -n build_binary
# python -m pytest torchrec -v -s -W ignore::pytest.PytestCollectionWarning --continue-on-collection-errors