gpu: nvgpu: Use new kmem API functions (gm206/*)

Use the new kmem API functions in gm206/*.

Bug 1799159
Bug 1823380

Change-Id: I7117c3153d266f0ac72978e2b54f6d4921d919e5
Signed-off-by: Alex Waterman <alexw@nvidia.com>
Reviewed-on: http://git-master/r/1318310
Reviewed-by: svccoveritychecker <svccoveritychecker@nvidia.com>
GVS: Gerrit_Virtual_Submit
Reviewed-by: Terje Bergstrom <tbergstrom@nvidia.com>

1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/gm206/bios_gm206.c b/drivers/gpu/nvgpu/gm206/bios_gm206.c
index 6e20e23a..3993691a 100644
--- a/drivers/gpu/nvgpu/gm206/bios_gm206.c
+++ b/drivers/gpu/nvgpu/gm206/bios_gm206.c
@@ -17,13 +17,13 @@
 #include <linux/pci.h>
 
 #include <nvgpu/bios.h>
+#include <nvgpu/kmem.h>
+#include <nvgpu/nvgpu_common.h>
 
 #include "gk20a/gk20a.h"
 #include "gm20b/fifo_gm20b.h"
 #include "bios_gm206.h"
 
-#include <nvgpu/nvgpu_common.h>
-
 #include <nvgpu/hw/gm206/hw_pwr_gm206.h>
 #include <nvgpu/hw/gm206/hw_mc_gm206.h>
 #include <nvgpu/hw/gm206/hw_top_gm206.h>
@@ -260,7 +260,7 @@ int gm206_bios_init(struct gk20a *g)
         } else {
                 gk20a_dbg_info("reading bios from EEPROM");
                 g->bios.size = BIOS_SIZE;
-                g->bios.data = vmalloc(BIOS_SIZE);
+                g->bios.data = nvgpu_vmalloc(g, BIOS_SIZE);
                 if (!g->bios.data)
                         return -ENOMEM;
                 g->ops.xve.disable_shadow_rom(g);