adding async tf strategy for gpu memory (#264)

jperez999 · jperez999 · commit feaf7486be03 · 2022-12-28T14:02:45.000-05:00
diff --git a/ci/test_unit.sh b/ci/test_unit.sh
@@ -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
 
 
