diff --git a/drivers/gpu/nvgpu/common/pmu/therm/thrm.c b/drivers/gpu/nvgpu/common/pmu/therm/thrm.c
index 9d3aa725f..26f4c58d5 100644
--- a/drivers/gpu/nvgpu/common/pmu/therm/thrm.c
+++ b/drivers/gpu/nvgpu/common/pmu/therm/thrm.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2016-2018, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2016-2019, NVIDIA CORPORATION.  All rights reserved.
  *
  * Permission is hereby granted, free of charge, to any person obtaining a
  * copy of this software and associated documentation files (the "Software"),
@@ -50,7 +50,7 @@ exit:
 	return status;
 }
 
-u32 therm_domain_pmu_setup(struct gk20a *g)
+int therm_domain_pmu_setup(struct gk20a *g)
 {
 	return therm_send_pmgr_tables_to_pmu(g);
 }
diff --git a/drivers/gpu/nvgpu/include/nvgpu/pmu/therm.h b/drivers/gpu/nvgpu/include/nvgpu/pmu/therm.h
index bd16d6b3f..72ed90df9 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/pmu/therm.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/pmu/therm.h
@@ -27,7 +27,7 @@
 struct gk20a;
 
 int therm_domain_sw_setup(struct gk20a *g);
-u32 therm_domain_pmu_setup(struct gk20a *g);
+int therm_domain_pmu_setup(struct gk20a *g);
 int therm_pmu_init_pmupstate(struct gk20a *g);
 void therm_pmu_free_pmupstate(struct gk20a *g);
 int therm_configure_therm_alert(struct gk20a *g);