2 files changed, 22 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 9e0dd64a..b955a84a 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -593,6 +593,7 @@ struct gpu_ops {
                 int (*get_internal_sensor_curr_temp)(struct gk20a *g, u32 *temp_f24_8);
                 void (*get_internal_sensor_limits)(s32 *max_24_8,
                                                         s32 *min_24_8);
+                u32 (*configure_therm_alert)(struct gk20a *g, s32 curr_warn_temp);
         } therm;
         struct {
                 bool (*is_pmu_supported)(struct gk20a *g);
diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
index 56ebda1a..58dd6f43 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
@@ -4029,6 +4029,24 @@ static int pmu_handle_perfmon_event(struct pmu_gk20a *pmu,
 }
 
 
+static int pmu_handle_therm_event(struct pmu_gk20a *pmu,
+                        struct nv_pmu_therm_msg *msg)
+{
+        gk20a_dbg_fn("");
+
+        switch (msg->msg_type) {
+        case NV_PMU_THERM_MSG_ID_EVENT_HW_SLOWDOWN_NOTIFICATION:
+                gk20a_dbg_pmu("received asserted event mask %d",
+                        msg->hw_slct_msg.mask);
+                break;
+        default:
+                gk20a_dbg_pmu("unkown therm event received %d", msg->msg_type);
+                break;
+        }
+
+        return 0;
+}
+
 static int pmu_handle_event(struct pmu_gk20a *pmu, struct pmu_msg *msg)
 {
         int err = 0;
@@ -4048,6 +4066,9 @@ static int pmu_handle_event(struct pmu_gk20a *pmu, struct pmu_msg *msg)
                         WARN_ON(1);
                 }
                 break;
+        case PMU_UNIT_THERM:
+                err = pmu_handle_therm_event(pmu, &msg->msg.therm);
+                break;
         default:
                 break;
         }