1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gm20b/hw_fifo_gm20b.h b/drivers/gpu/nvgpu/gm20b/hw_fifo_gm20b.h
index 2b8bd9ce..09e311ef 100644
--- a/drivers/gpu/nvgpu/gm20b/hw_fifo_gm20b.h
+++ b/drivers/gpu/nvgpu/gm20b/hw_fifo_gm20b.h
@@ -346,10 +346,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;