1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/hw_fifo_gk20a.h b/drivers/gpu/nvgpu/gk20a/hw_fifo_gk20a.h
index 83e7d776..aa77d026 100644
--- a/drivers/gpu/nvgpu/gk20a/hw_fifo_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/hw_fifo_gk20a.h
@@ -398,10 +398,18 @@ static inline u32 fifo_preempt_type_channel_f(void)
 {
         return 0x0;
 }
+static inline u32 fifo_preempt_type_tsg_f(void)
+{
+        return 0x1000000;
+}
 static inline u32 fifo_preempt_chid_f(u32 v)
 {
         return (v & 0xfff) << 0;
 }
+static inline u32 fifo_preempt_id_f(u32 v)
+{
+        return (v & 0xfff) << 0;
+}
 static inline u32 fifo_trigger_mmu_fault_r(u32 i)
 {
         return 0x00002a30 + i*4;