Skip to content

Commit e40e6c4

Browse files
committed
Add GPUUtilization as expect metric
1 parent 1deabff commit e40e6c4

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

frontend/server/src/test/java/org/pytorch/serve/ModelServerTest.java

+1-2
Original file line numberDiff line numberDiff line change
@@ -1351,8 +1351,7 @@ public void testMetricManager() throws JsonParseException, InterruptedException
13511351
Map<String, Map<String, String>> expectedMetrics = new HashMap<>();
13521352
expectedMetrics.put("GPUMemoryUtilization", Map.of(UNIT, "Percent", LEVEL, HOST));
13531353
expectedMetrics.put("GPUMemoryUsed", Map.of(UNIT, "Megabytes", LEVEL, HOST));
1354-
// torch.mps does not allow to calculate GPUUtilization, see ts/metrics/system_metrics.py
1355-
// expectedMetrics.put("GPUUtilization", Map.of(UNIT, "Percent", LEVEL, HOST));
1354+
expectedMetrics.put("GPUUtilization", Map.of(UNIT, "Percent", LEVEL, HOST));
13561355
expectedMetrics.put("CPUUtilization", Map.of(UNIT, "Percent", LEVEL, HOST));
13571356
expectedMetrics.put("MemoryUsed", Map.of(UNIT, "Megabytes", LEVEL, HOST));
13581357
expectedMetrics.put("MemoryAvailable", Map.of(UNIT, "Megabytes", LEVEL, HOST));

0 commit comments

Comments
 (0)