Add GPUUtilization as expect metric

pytorch · Dec 20, 2024 · e40e6c4 · e40e6c4
1 parent 1deabff
commit e40e6c4
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/frontend/server/src/test/java/org/pytorch/serve/ModelServerTest.java b/frontend/server/src/test/java/org/pytorch/serve/ModelServerTest.java
@@ -1351,8 +1351,7 @@ public void testMetricManager() throws JsonParseException, InterruptedException
         Map<String, Map<String, String>> expectedMetrics = new HashMap<>();
         expectedMetrics.put("GPUMemoryUtilization", Map.of(UNIT, "Percent", LEVEL, HOST));
         expectedMetrics.put("GPUMemoryUsed", Map.of(UNIT, "Megabytes", LEVEL, HOST));
-        // torch.mps does not allow to calculate GPUUtilization, see ts/metrics/system_metrics.py
-        // expectedMetrics.put("GPUUtilization", Map.of(UNIT, "Percent", LEVEL, HOST));
+        expectedMetrics.put("GPUUtilization", Map.of(UNIT, "Percent", LEVEL, HOST));
         expectedMetrics.put("CPUUtilization", Map.of(UNIT, "Percent", LEVEL, HOST));
         expectedMetrics.put("MemoryUsed", Map.of(UNIT, "Megabytes", LEVEL, HOST));
         expectedMetrics.put("MemoryAvailable", Map.of(UNIT, "Megabytes", LEVEL, HOST));