1 files changed, 17 insertions, 6 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gr_pri_gk20a.h b/drivers/gpu/nvgpu/gk20a/gr_pri_gk20a.h
index 248fa291..62e276de 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_pri_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gr_pri_gk20a.h
@@ -1,7 +1,7 @@
 /*
  * GK20A Graphics Context Pri Register Addressing
  *
- * Copyright (c) 2014, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2014-2016, NVIDIA CORPORATION.  All rights reserved.
  *
  * This program is free software; you can redistribute it and/or modify it
  * under the terms and conditions of the GNU General Public License,
@@ -167,12 +167,21 @@ static inline u32 pri_ppc_addr(struct gk20a *g, u32 addr, u32 gpc, u32 ppc)
                 ppc_in_gpc_base + (ppc * ppc_in_gpc_stride) + addr;
 }
 
+/*
+ * LTC pri addressing
+ */
+static inline bool pri_is_ltc_addr(u32 addr)
+{
+        return ((addr >= ltc_pltcg_base_v()) && (addr < ltc_pltcg_extent_v()));
+}
+
 enum ctxsw_addr_type {
-        CTXSW_ADDR_TYPE_SYS = 0,
-        CTXSW_ADDR_TYPE_GPC = 1,
-        CTXSW_ADDR_TYPE_TPC = 2,
-        CTXSW_ADDR_TYPE_BE  = 3,
-        CTXSW_ADDR_TYPE_PPC = 4
+        CTXSW_ADDR_TYPE_SYS  = 0,
+        CTXSW_ADDR_TYPE_GPC  = 1,
+        CTXSW_ADDR_TYPE_TPC  = 2,
+        CTXSW_ADDR_TYPE_BE   = 3,
+        CTXSW_ADDR_TYPE_PPC  = 4,
+        CTXSW_ADDR_TYPE_LTCS = 5
 };
 
 #define PRI_BROADCAST_FLAGS_NONE  0
@@ -180,5 +189,7 @@ enum ctxsw_addr_type {
 #define PRI_BROADCAST_FLAGS_TPC   BIT(1)
 #define PRI_BROADCAST_FLAGS_BE    BIT(2)
 #define PRI_BROADCAST_FLAGS_PPC   BIT(3)
+#define PRI_BROADCAST_FLAGS_LTCS  BIT(4)
+#define PRI_BROADCAST_FLAGS_LTSS  BIT(5)
 
 #endif /* GR_PRI_GK20A_H */