1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/os_ops.c b/drivers/gpu/nvgpu/os/linux/os_ops.c
index 14f92787..5fc5beb4 100644
--- a/drivers/gpu/nvgpu/os/linux/os_ops.c
+++ b/drivers/gpu/nvgpu/os/linux/os_ops.c
@@ -18,6 +18,8 @@
 
 #include "os_ops_gm20b.h"
 #include "os_ops_gp10b.h"
+#include "os_ops_gp106.h"
+#include "os_ops_gv100.h"
 
 int nvgpu_init_os_linux_ops(struct nvgpu_os_linux *l)
 {
@@ -32,6 +34,12 @@ int nvgpu_init_os_linux_ops(struct nvgpu_os_linux *l)
         case NVGPU_GPUID_GP10B:
                 nvgpu_gp10b_init_os_ops(l);
                 break;
+        case NVGPU_GPUID_GP106:
+                nvgpu_gp106_init_os_ops(l);
+                break;
+        case NVGPU_GPUID_GV100:
+                nvgpu_gv100_init_os_ops(l);
+                break;
         default:
                 break;
         }