Skip to content

Commit

Permalink
Merge pull request #1317 from bharathappali/gpu-support-pr-3
Browse files Browse the repository at this point in the history
Add accelerator related fields in existing structures
  • Loading branch information
dinogun authored Oct 4, 2024
2 parents a1af211 + e699630 commit 62c42b3
Show file tree
Hide file tree
Showing 3 changed files with 49 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
package com.autotune.common.data.metrics;

import com.autotune.common.data.system.info.device.accelerator.AcceleratorDeviceData;

public class AcceleratorMetricResult {
private AcceleratorDeviceData acceleratorDeviceData;
private MetricResults metricResults;

public AcceleratorMetricResult(AcceleratorDeviceData acceleratorDeviceData, MetricResults metricResults) {
this.acceleratorDeviceData = acceleratorDeviceData;
this.metricResults = metricResults;
}

public AcceleratorDeviceData getAcceleratorDeviceData() {
return acceleratorDeviceData;
}

public void setAcceleratorDeviceData(AcceleratorDeviceData acceleratorDeviceData) {
this.acceleratorDeviceData = acceleratorDeviceData;
}

public MetricResults getMetricResults() {
return metricResults;
}

public void setMetricResults(MetricResults metricResults) {
this.metricResults = metricResults;
}
}
10 changes: 10 additions & 0 deletions src/main/java/com/autotune/common/data/result/ContainerData.java
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
import com.autotune.analyzer.recommendations.ContainerRecommendations;
import com.autotune.analyzer.utils.AnalyzerConstants;
import com.autotune.common.data.metrics.Metric;
import com.autotune.common.data.system.info.device.ContainerDeviceList;
import com.autotune.utils.KruizeConstants;
import com.google.gson.annotations.SerializedName;

Expand All @@ -29,6 +30,7 @@ public class ContainerData {
private String container_name;
//key is intervalEndTime
private HashMap<Timestamp, IntervalResults> results;
private ContainerDeviceList containerDeviceList;
@SerializedName(KruizeConstants.JSONKeys.RECOMMENDATIONS)
private ContainerRecommendations containerRecommendations;
private HashMap<AnalyzerConstants.MetricName, Metric> metrics;
Expand Down Expand Up @@ -85,6 +87,14 @@ public HashMap<AnalyzerConstants.MetricName, Metric> getMetrics() {
public void setMetrics(HashMap<AnalyzerConstants.MetricName, Metric> metrics) {
this.metrics = metrics;
}

public ContainerDeviceList getContainerDeviceList() {
return containerDeviceList;
}

public void setContainerDeviceList(ContainerDeviceList containerDeviceList) {
this.containerDeviceList = containerDeviceList;
}
@Override
public String toString() {
return "ContainerData{" +
Expand Down
10 changes: 10 additions & 0 deletions src/main/java/com/autotune/common/data/result/IntervalResults.java
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
package com.autotune.common.data.result;

import com.autotune.analyzer.utils.AnalyzerConstants;
import com.autotune.common.data.metrics.AcceleratorMetricResult;
import com.autotune.common.data.metrics.MetricResults;
import com.google.gson.annotations.SerializedName;

Expand All @@ -32,6 +33,7 @@
public class IntervalResults {
@SerializedName(METRICS)
HashMap<AnalyzerConstants.MetricName, MetricResults> metricResultsMap;
HashMap<AnalyzerConstants.MetricName, AcceleratorMetricResult> acceleratorMetricResultHashMap;
@SerializedName(INTERVAL_START_TIME)
private Timestamp intervalStartTime;
@SerializedName(INTERVAL_END_TIME)
Expand Down Expand Up @@ -85,6 +87,14 @@ public void setDurationInMinutes(Double durationInMinutes) {
this.durationInMinutes = durationInMinutes;
}

public HashMap<AnalyzerConstants.MetricName, AcceleratorMetricResult> getAcceleratorMetricResultHashMap() {
return acceleratorMetricResultHashMap;
}

public void setAcceleratorMetricResultHashMap(HashMap<AnalyzerConstants.MetricName, AcceleratorMetricResult> acceleratorMetricResultHashMap) {
this.acceleratorMetricResultHashMap = acceleratorMetricResultHashMap;
}

@Override
public String toString() {
return "IntervalResults{" +
Expand Down

0 comments on commit 62c42b3

Please sign in to comment.