1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
index 2ebe0011..8d7ccfa8 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
@@ -138,6 +138,8 @@ enum gk20a_cbc_op {
 
 #define nvgpu_get_litter_value(g, v) (g)->ops.get_litter_value((g), v)
 
+#define MAX_TPC_PG_CONFIGS      3
+
 enum nvgpu_unit;
 
 enum nvgpu_flush_op;
@@ -1592,7 +1594,7 @@ struct gk20a {
         u32 tpc_pg_mask;
         bool can_tpc_powergate;
 
-        u32 valid_tpc_mask;
+        u32 valid_tpc_mask[MAX_TPC_PG_CONFIGS];
 
         struct nvgpu_bios bios;
         bool bios_is_init;