File tree 1 file changed +10
-5
lines changed
frontend/server/src/main/java/org/pytorch/serve/device
1 file changed +10
-5
lines changed Original file line number Diff line number Diff line change @@ -71,11 +71,16 @@ private IAcceleratorUtility createAcceleratorUtility() {
71
71
private void populateAccelerators () {
72
72
if (this .acceleratorUtil != null ) {
73
73
String envVarName = this .acceleratorUtil .getGpuEnvVariableName ();
74
- String requestedAcceleratorIds = System .getenv (envVarName );
75
- LinkedHashSet <Integer > availableAcceleratorIds =
76
- IAcceleratorUtility .parseVisibleDevicesEnv (requestedAcceleratorIds );
77
- this .accelerators =
78
- this .acceleratorUtil .getAvailableAccelerators (availableAcceleratorIds );
74
+ if (envVarName != null ) {
75
+ String requestedAcceleratorIds = System .getenv (envVarName );
76
+ LinkedHashSet <Integer > availableAcceleratorIds =
77
+ IAcceleratorUtility .parseVisibleDevicesEnv (requestedAcceleratorIds );
78
+ this .accelerators =
79
+ this .acceleratorUtil .getAvailableAccelerators (availableAcceleratorIds );
80
+ } else {
81
+ // Handle the case where envVarName is null
82
+ this .accelerators = this .acceleratorUtil .getAvailableAccelerators (new LinkedHashSet <>());
83
+ }
79
84
} else {
80
85
this .accelerators = new ArrayList <>();
81
86
}
You can’t perform that action at this time.
0 commit comments