diff --git a/drivers/gpu/nvgpu/gp106/gr_gp106.c b/drivers/gpu/nvgpu/gp106/gr_gp106.c
index e4768e0db..d54a20893 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 e08a7dc55..5338789f0 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