gpu: nvgpu: Add new CE class for gp10x

Added new CE class(PASCAL_DMA_COPY_B) for gp106 and gp104.

JIRA DNVGPU-25

Change-Id: I3c85e3ffdedf7594d41bf5c2fbebbf44addd1720
Signed-off-by: Lakshmanan M <lm@nvidia.com>
Reviewed-on: http://git-master/r/1166709
Reviewed-by: Konsta Holtta <kholtta@nvidia.com>
GVS: Gerrit_Virtual_Submit
Reviewed-by: Terje Bergstrom <tbergstrom@nvidia.com>

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