Skip to content

Commit

Permalink
adding async tf strategy for gpu memory (#264)
Browse files Browse the repository at this point in the history
  • Loading branch information
@jperez999
jperez999 committed Dec 28, 2022
1 parent 3421fc0 commit feaf748
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ci/test_unit.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,6 @@
set -e

# Run tests
pytest -rsx tests/unit tests/notebook
TF_GPU_ALLOCATOR=cuda_malloc_async pytest -rsx tests/unit tests/notebook


0 comments on commit feaf748

Please sign in to comment.