1 files changed, 8 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
index e140b47a..4654491f 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
@@ -4036,8 +4036,14 @@ static int pmu_handle_therm_event(struct pmu_gk20a *pmu,
 
         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);
+#ifdef CONFIG_ARCH_TEGRA_18x_SOC
+                if (msg->hw_slct_msg.mask == BIT(NV_PMU_THERM_EVENT_THERMAL_1))
+                        nvgpu_clk_arb_schedule_alarm(gk20a_from_pmu(pmu),
+                                (0x1UL << NVGPU_GPU_EVENT_ALARM_THERMAL_ABOVE_THRESHOLD));
+                else
+#endif
+                        gk20a_dbg_pmu("Unwanted/Unregistered thermal event received %d",
+                                msg->hw_slct_msg.mask);
                 break;
         default:
                 gk20a_dbg_pmu("unkown therm event received %d", msg->msg_type);