2 files changed, 3 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 64716ac4..ffce1c5e 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -1448,6 +1448,8 @@ int gk20a_wait_for_idle(struct device *dev);
         GK20A_GPUID(NVGPU_GPU_ARCH_GM200, NVGPU_GPU_IMPL_GM20B)
 #define NVGPU_GPUID_GP10B \
         GK20A_GPUID(NVGPU_GPU_ARCH_GP100, NVGPU_GPU_IMPL_GP10B)
+#define GK20A_GPUID_GM20B_B \
+        GK20A_GPUID(NVGPU_GPU_ARCH_GM200, NVGPU_GPU_IMPL_GM20B_B)
 #define NVGPU_GPUID_GP104 \
         GK20A_GPUID(NVGPU_GPU_ARCH_GP100, NVGPU_GPU_IMPL_GP104)
 #define NVGPU_GPUID_GP106 \
diff --git a/drivers/gpu/nvgpu/gk20a/hal.c b/drivers/gpu/nvgpu/gk20a/hal.c
index ce99aa7a..8fadc199 100644
--- a/drivers/gpu/nvgpu/gk20a/hal.c
+++ b/drivers/gpu/nvgpu/gk20a/hal.c
@@ -35,6 +35,7 @@ int gpu_init_hal(struct gk20a *g)
                 gk20a_init_hal(g);
                 break;
         case GK20A_GPUID_GM20B:
+        case GK20A_GPUID_GM20B_B:
                 gk20a_dbg_info("gm20b detected");
                 if (gm20b_init_hal(g))
                         return -ENODEV;