2 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index 676dd6c9..f0e89f8a 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -2002,6 +2002,7 @@ void gr_gk20a_load_ctxsw_ucode_header(struct gk20a *g, u64 addr_base,
         /* Write out the actual data */
         switch (segments->boot_signature) {
         case FALCON_UCODE_SIG_T18X_GPCCS_WITH_RESERVED:
+        case FALCON_UCODE_SIG_T21X_FECS_WITH_DMEM_SIZE:
         case FALCON_UCODE_SIG_T21X_FECS_WITH_RESERVED:
         case FALCON_UCODE_SIG_T21X_GPCCS_WITH_RESERVED:
         case FALCON_UCODE_SIG_T12X_FECS_WITH_RESERVED:
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.h b/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
index 326b4f96..2c575534 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
@@ -345,6 +345,7 @@ struct gk20a_ctxsw_ucode_segments {
  * so FALCON_UCODE_SIG_T21X_FECS_WITH_RESERVED used
  * for T18X*/
 #define FALCON_UCODE_SIG_T18X_GPCCS_WITH_RESERVED       0x68edab34
+#define FALCON_UCODE_SIG_T21X_FECS_WITH_DMEM_SIZE       0x9121ab5c
 #define FALCON_UCODE_SIG_T21X_FECS_WITH_RESERVED        0x9125ab5c
 #define FALCON_UCODE_SIG_T12X_FECS_WITH_RESERVED        0x8a621f78
 #define FALCON_UCODE_SIG_T12X_FECS_WITHOUT_RESERVED     0x67e5344b