Update gpuScheduler.ts (#1043)

34988d9b · demianzhang · chicm-ms · 9974c602 · 34988d9b
Commit 34988d9b authored May 14, 2019 by demianzhang Committed by chicm-ms May 14, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 2 deletions

src/nni_manager/training_service/local/gpuScheduler.ts src/nni_manager/training_service/local/gpuScheduler.ts +7 -2

No files found.
--- a/src/nni_manager/training_service/local/gpuScheduler.ts
+++ b/src/nni_manager/training_service/local/gpuScheduler.ts
@@ -73,8 +73,13 @@ class GPUScheduler {

    public getAvailableGPUIndices(): number[] {
        if (this.gpuSummary !== undefined) {
-            return this.gpuSummary.gpuInfos.filter((info: GPUInfo) => info.activeProcessNum === 0)
-                .map((info: GPUInfo) => info.index);
+            if(process.platform === 'win32') {
+                return this.gpuSummary.gpuInfos.map((info: GPUInfo) => info.index);
+            }
+            else{
+                return this.gpuSummary.gpuInfos.filter((info: GPUInfo) => info.activeProcessNum === 0)
+                    .map((info: GPUInfo) => info.index);
+            }
        }

        return [];