2 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
index a5863567..5a50b843 100644
--- a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
@@ -24,7 +24,6 @@
 #ifdef CONFIG_TEGRA_GK20A_NVHOST
 #include <linux/nvhost.h>
 #endif
-#include <linux/sort.h>
 
 #include <nvgpu/dma.h>
 #include <nvgpu/timers.h>
@@ -33,6 +32,7 @@
 #include <nvgpu/log.h>
 #include <nvgpu/soc.h>
 #include <nvgpu/atomic.h>
+#include <nvgpu/sort.h>
 
 #include "gk20a.h"
 #include "debug_gk20a.h"
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index 96ec7f6e..8489816e 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -25,7 +25,6 @@
 #include <linux/dma-mapping.h>
 #include <linux/firmware.h>
 #include <linux/nvhost.h>
-#include <linux/sort.h>
 #include <trace/events/gk20a.h>
 
 #include <nvgpu/dma.h>
@@ -34,6 +33,7 @@
 #include <nvgpu/nvgpu_common.h>
 #include <nvgpu/log.h>
 #include <nvgpu/bsearch.h>
+#include <nvgpu/sort.h>
 
 #include "gk20a.h"
 #include "kind_gk20a.h"