From 513c4afaa15ec8cc00c4c97321408d21459afcb5 Mon Sep 17 00:00:00 2001
From: samou <samou@google.com>
Date: Fri, 16 Apr 2021 11:17:18 +0000
Subject: [PATCH] powerstats: Add TPU EnergyConsumer power attribution

Bug: 185564760
Change-Id: I6550ed839599c979ee09492fb8eaabd50f04ebe2
---
 powerstats/Gs101CommonDataProviders.cpp | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/powerstats/Gs101CommonDataProviders.cpp b/powerstats/Gs101CommonDataProviders.cpp
index b62262a0..5a14488e 100644
--- a/powerstats/Gs101CommonDataProviders.cpp
+++ b/powerstats/Gs101CommonDataProviders.cpp
@@ -651,6 +651,21 @@ void addDevfreq(std::shared_ptr<PowerStats> p) {
             "INT", "/sys/devices/platform/17000020.devfreq_int/devfreq/17000020.devfreq_int"));
 }
 
+void addTPU(std::shared_ptr<PowerStats> p) {
+    std::map<std::string, int32_t> stateCoeffs;
+
+    stateCoeffs = {
+        {"500000",  10},
+        {"800000",  20},
+        {"1066000", 30},
+        {"1230000", 40}};
+
+    p->addEnergyConsumer(PowerStatsEnergyConsumer::createMeterAndAttrConsumer(p,
+            EnergyConsumerType::OTHER, "TPU", {"S10M_VDD_TPU"},
+            {{UID_TIME_IN_STATE, "/sys/class/edgetpu/abrolhos/device/tpu_usage"}},
+            stateCoeffs));
+}
+
 /**
  * Unlike other data providers, which source power entity state residency data from the kernel,
  * this data provider acts as a general-purpose channel for state residency data providers
@@ -684,6 +699,7 @@ void addGs101CommonDataProviders(std::shared_ptr<PowerStats> p) {
     addUfs(p);
     addPowerDomains(p);
     addDevfreq(p);
+    addTPU(p);
 
     // TODO (b/181070764) (b/182941084):
     // Remove this when Wifi/BT energy consumption models are available or revert before ship