Fix null pointer bug with populateAccelerators trying to get null App…

…leUtil GPU env value
pytorch · Dec 17, 2024 · 29895d9 · 29895d9
1 parent bc96fa7
commit 29895d9
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/frontend/server/src/main/java/org/pytorch/serve/device/SystemInfo.java b/frontend/server/src/main/java/org/pytorch/serve/device/SystemInfo.java
@@ -71,11 +71,16 @@ private IAcceleratorUtility createAcceleratorUtility() {
     private void populateAccelerators() {
         if (this.acceleratorUtil != null) {
             String envVarName = this.acceleratorUtil.getGpuEnvVariableName();
-            String requestedAcceleratorIds = System.getenv(envVarName);
-            LinkedHashSet<Integer> availableAcceleratorIds =
-                    IAcceleratorUtility.parseVisibleDevicesEnv(requestedAcceleratorIds);
-            this.accelerators =
-                    this.acceleratorUtil.getAvailableAccelerators(availableAcceleratorIds);
+            if (envVarName != null) {
+                String requestedAcceleratorIds = System.getenv(envVarName);
+                LinkedHashSet<Integer> availableAcceleratorIds =
+                        IAcceleratorUtility.parseVisibleDevicesEnv(requestedAcceleratorIds);
+                this.accelerators =
+                        this.acceleratorUtil.getAvailableAccelerators(availableAcceleratorIds);
+            } else {
+                // Handle the case where envVarName is null
+                this.accelerators = this.acceleratorUtil.getAvailableAccelerators(new LinkedHashSet<>());
+            }
         } else {
             this.accelerators = new ArrayList<>();
         }