2 files changed, 2 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp106/gr_gp106.c b/drivers/gpu/nvgpu/gp106/gr_gp106.c
index e4768e0d..d54a2089 100644
--- a/drivers/gpu/nvgpu/gp106/gr_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/gr_gp106.c
@@ -32,6 +32,7 @@ static bool gr_gp106_is_valid_class(struct gk20a *g, u32 class_num)
         case PASCAL_A:
         case PASCAL_B:
         case PASCAL_DMA_COPY_A:
+        case PASCAL_DMA_COPY_B:
                 valid = true;
                 break;
 
diff --git a/drivers/gpu/nvgpu/gp10b/gr_gp10b.h b/drivers/gpu/nvgpu/gp10b/gr_gp10b.h
index e08a7dc5..5338789f 100644
--- a/drivers/gpu/nvgpu/gp10b/gr_gp10b.h
+++ b/drivers/gpu/nvgpu/gp10b/gr_gp10b.h
@@ -25,6 +25,7 @@ enum {
         PASCAL_A                 = 0xC097,
         PASCAL_COMPUTE_A         = 0xC0C0,
         PASCAL_DMA_COPY_A        = 0xC0B5,
+        PASCAL_DMA_COPY_B        = 0xC1B5,
 };
 
 #define NVC097_SET_GO_IDLE_TIMEOUT              0x022c