run unit tests with multiple device streams

ValeevGroup · Sep 26, 2023 · b4b5997 · b4b5997
1 parent 4033081
commit b4b5997
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
@@ -165,10 +165,9 @@ if(ENABLE_MPI)
         $<TARGET_FILE:${executable}> --log_level=unit_scope ${${executable}_np_${p}_args}
         ${MPIEXEC_POSTFLAGS}
       )
-    # N.B. some CUDA unit tests require TA_CUDA_NUM_STREAMS=1 for now
     set_tests_properties(tiledarray/unit/run-np-${p}
             PROPERTIES FIXTURES_REQUIRED TA_UNIT_TESTS_EXEC
-            ENVIRONMENT "MAD_NUM_THREADS=2;TA_CUDA_NUM_STREAMS=1"
+            ENVIRONMENT "MAD_NUM_THREADS=2"
             )
 
     if (p GREATER 1)
@@ -178,9 +177,8 @@ if(ENABLE_MPI)
 else()
   add_test(NAME tiledarray/unit/run-np-1
            COMMAND ${executable})
-  # N.B. some CUDA unit tests require TA_CUDA_NUM_STREAMS=1 for now
   set_tests_properties(tiledarray/unit/run-np-1
           PROPERTIES FIXTURES_REQUIRED TA_UNIT_TESTS_EXEC
-          ENVIRONMENT "MAD_NUM_THREADS=2;TA_CUDA_NUM_STREAMS=1"
+          ENVIRONMENT "MAD_NUM_THREADS=2"
           )
 endif()