Fix github action failed test

Autodesk · Jul 7, 2022 · 988dead · 988dead
1 parent 8aba020
commit 988dead
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/libNeonDomain/tests/unit/domainUt_swap/src/Swap.cu b/libNeonDomain/tests/unit/domainUt_swap/src/Swap.cu
@@ -104,11 +104,15 @@ void SwapContainerRun(TestData<G, T, C>& data)
 namespace {
 int getNGpus()
 {
-    int maxGPUs = Neon::set::DevSet::maxSet().setCardinality();
-    if (maxGPUs > 1) {
-        return maxGPUs;
+    if (Neon::sys::globalSpace::gpuSysObjStorage.numDevs() > 0) {
+        int maxGPUs = Neon::set::DevSet::maxSet().setCardinality();
+        if (maxGPUs > 1) {
+            return maxGPUs;
+        } else {
+            return 3;
+        }
     } else {
-        return 3;
+        return 0;
     }
 }
 }  // namespace