2 files changed, 12 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/dbg_gpu_gk20a.h b/drivers/gpu/nvgpu/gk20a/dbg_gpu_gk20a.h
index 50002557..4d3c4d74 100644
--- a/drivers/gpu/nvgpu/gk20a/dbg_gpu_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/dbg_gpu_gk20a.h
@@ -72,6 +72,12 @@ struct dbg_session_gk20a {
         bool broadcast_stop_trigger;
 
         struct nvgpu_mutex ioctl_lock;
+
+        /*
+         * sm set exception type mask flag, to check whether
+         * exception type mask is requested or not.
+         */
+        bool is_sm_exception_type_mask_set;
 };
 
 struct dbg_session_data {
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.h b/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
index 0c6be57b..804e0e25 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
@@ -437,6 +437,12 @@ struct gr_gk20a {
         u32 no_of_sm;
         struct sm_info *sm_to_cluster;
         struct nvgpu_gr_sm_error_state *sm_error_states;
+
+#define NVGPU_SM_EXCEPTION_TYPE_MASK_NONE               (0x0U)
+#define NVGPU_SM_EXCEPTION_TYPE_MASK_FATAL              (0x1U << 0)
+        u32 sm_exception_mask_type;
+        u32 sm_exception_mask_refcount;
+
 #if defined(CONFIG_GK20A_CYCLE_STATS)
         struct nvgpu_mutex                      cs_lock;
         struct gk20a_cs_snapshot        *cs_data;