diff --git a/drivers/gpu/nvgpu/common/ce/ce.c b/drivers/gpu/nvgpu/common/ce/ce.c
index af126920f..a02f75a14 100644
--- a/drivers/gpu/nvgpu/common/ce/ce.c
+++ b/drivers/gpu/nvgpu/common/ce/ce.c
@@ -595,8 +595,7 @@ u32 nvgpu_ce_create_context(struct gk20a *g,
 
 	/* -1 means default channel runlist level */
 	if (runlist_level != -1) {
-		err = gk20a_tsg_set_runlist_interleave(ce_ctx->tsg,
-						       runlist_level);
+		err = nvgpu_tsg_set_interleave(ce_ctx->tsg, runlist_level);
 		if (err != 0) {
 			nvgpu_err(g, "ce: set runlist interleave failed");
 			goto end;
diff --git a/drivers/gpu/nvgpu/common/fifo/tsg.c b/drivers/gpu/nvgpu/common/fifo/tsg.c
index 176ee7f10..750246d41 100644
--- a/drivers/gpu/nvgpu/common/fifo/tsg.c
+++ b/drivers/gpu/nvgpu/common/fifo/tsg.c
@@ -507,32 +507,34 @@ bool nvgpu_tsg_check_ctxsw_timeout(struct tsg_gk20a *tsg,
 	return recover;
 }
 
-int gk20a_tsg_set_runlist_interleave(struct tsg_gk20a *tsg, u32 level)
+int nvgpu_tsg_set_interleave(struct tsg_gk20a *tsg, u32 level)
 {
 	struct gk20a *g = tsg->g;
 	int ret;
 
-	nvgpu_log(g, gpu_dbg_sched, "tsgid=%u interleave=%u", tsg->tsgid, level);
+	nvgpu_log(g, gpu_dbg_sched,
+			"tsgid=%u interleave=%u", tsg->tsgid, level);
 
 	nvgpu_speculation_barrier();
-	switch (level) {
-	case NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_LOW:
-	case NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_MEDIUM:
-	case NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_HIGH:
-		ret = g->ops.runlist.set_interleave(g, tsg->tsgid,
-							0, level);
-		if (ret == 0) {
-			tsg->interleave_level = level;
-			ret = g->ops.runlist.reload(g, tsg->runlist_id,
-					true, true);
-		}
-		break;
-	default:
-		ret = -EINVAL;
-		break;
+
+	if ((level != NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_LOW) &&
+	    (level != NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_MEDIUM) &&
+	    (level != NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_HIGH)) {
+		return -EINVAL;
 	}
 
-	return ret;
+	if (g->ops.tsg.set_interleave != NULL) {
+		ret = g->ops.tsg.set_interleave(tsg, level);
+		if (ret != 0) {
+			nvgpu_err(g,
+				"set interleave failed tsgid=%u", tsg->tsgid);
+			return ret;
+		}
+	}
+
+	tsg->interleave_level = level;
+
+	return g->ops.runlist.reload(g, tsg->runlist_id, true, true);
 }
 
 int nvgpu_tsg_set_timeslice(struct tsg_gk20a *tsg, u32 timeslice_us)
diff --git a/drivers/gpu/nvgpu/common/vgpu/fifo/fifo_vgpu.h b/drivers/gpu/nvgpu/common/vgpu/fifo/fifo_vgpu.h
index fec8a12ff..0893a558c 100644
--- a/drivers/gpu/nvgpu/common/vgpu/fifo/fifo_vgpu.h
+++ b/drivers/gpu/nvgpu/common/vgpu/fifo/fifo_vgpu.h
@@ -55,6 +55,7 @@ void vgpu_tsg_release(struct tsg_gk20a *tsg);
 int vgpu_tsg_bind_channel(struct tsg_gk20a *tsg, struct channel_gk20a *ch);
 int vgpu_tsg_unbind_channel(struct tsg_gk20a *tsg, struct channel_gk20a *ch);
 int vgpu_tsg_set_timeslice(struct tsg_gk20a *tsg, u32 timeslice);
+int vgpu_tsg_set_interleave(struct tsg_gk20a *tsg, u32 level);
 void vgpu_tsg_enable(struct tsg_gk20a *tsg);
 int vgpu_set_sm_exception_type_mask(struct channel_gk20a *ch, u32 mask);
 void vgpu_channel_free_ctx_header(struct channel_gk20a *c);
diff --git a/drivers/gpu/nvgpu/common/vgpu/fifo/runlist_vgpu.c b/drivers/gpu/nvgpu/common/vgpu/fifo/runlist_vgpu.c
index e6ad47af3..efeca45ba 100644
--- a/drivers/gpu/nvgpu/common/vgpu/fifo/runlist_vgpu.c
+++ b/drivers/gpu/nvgpu/common/vgpu/fifo/runlist_vgpu.c
@@ -199,27 +199,6 @@ int vgpu_runlist_reload(struct gk20a *g, u32 runlist_id,
 	return vgpu_runlist_update(g, runlist_id, NULL, add, wait_for_finish);
 }
 
-int vgpu_runlist_set_interleave(struct gk20a *g,
-					u32 id,
-					u32 runlist_id,
-					u32 new_level)
-{
-	struct tegra_vgpu_cmd_msg msg = {0};
-	struct tegra_vgpu_tsg_runlist_interleave_params *p =
-			&msg.params.tsg_interleave;
-	int err;
-
-	nvgpu_log_fn(g, " ");
-
-	msg.cmd = TEGRA_VGPU_CMD_TSG_SET_RUNLIST_INTERLEAVE;
-	msg.handle = vgpu_get_handle(g);
-	p->tsg_id = id;
-	p->level = new_level;
-	err = vgpu_comm_sendrecv(&msg, sizeof(msg), sizeof(msg));
-	WARN_ON(err || msg.ret);
-	return err ? err : msg.ret;
-}
-
 u32 vgpu_runlist_length_max(struct gk20a *g)
 {
 	struct vgpu_priv_data *priv = vgpu_get_priv_data(g);
diff --git a/drivers/gpu/nvgpu/common/vgpu/fifo/runlist_vgpu.h b/drivers/gpu/nvgpu/common/vgpu/fifo/runlist_vgpu.h
index 2a3ceaf00..e20eb8944 100644
--- a/drivers/gpu/nvgpu/common/vgpu/fifo/runlist_vgpu.h
+++ b/drivers/gpu/nvgpu/common/vgpu/fifo/runlist_vgpu.h
@@ -30,9 +30,5 @@ int vgpu_runlist_update_for_channel(struct gk20a *g, u32 runlist_id,
 					bool add, bool wait_for_finish);
 int vgpu_runlist_reload(struct gk20a *g, u32 runlist_id,
 				bool add, bool wait_for_finish);
-int vgpu_runlist_set_interleave(struct gk20a *g,
-					u32 id,
-					u32 runlist_id,
-					u32 new_level);
 u32 vgpu_runlist_length_max(struct gk20a *g);
 u32 vgpu_runlist_entry_size(struct gk20a *g);
diff --git a/drivers/gpu/nvgpu/common/vgpu/gp10b/vgpu_hal_gp10b.c b/drivers/gpu/nvgpu/common/vgpu/gp10b/vgpu_hal_gp10b.c
index a93cd425c..fb832e05c 100644
--- a/drivers/gpu/nvgpu/common/vgpu/gp10b/vgpu_hal_gp10b.c
+++ b/drivers/gpu/nvgpu/common/vgpu/gp10b/vgpu_hal_gp10b.c
@@ -508,7 +508,6 @@ static const struct gpu_ops vgpu_gp10b_ops = {
 		.reschedule = NULL,
 		.update_for_channel = vgpu_runlist_update_for_channel,
 		.reload = vgpu_runlist_reload,
-		.set_interleave = vgpu_runlist_set_interleave,
 		.count_max = gk20a_runlist_count_max,
 		.entry_size = vgpu_runlist_entry_size,
 		.length_max = vgpu_runlist_length_max,
@@ -562,6 +561,7 @@ static const struct gpu_ops vgpu_gp10b_ops = {
 		.post_event_id = nvgpu_tsg_post_event_id,
 		.set_timeslice = vgpu_tsg_set_timeslice,
 		.default_timeslice_us = vgpu_tsg_default_timeslice_us,
+		.set_interleave = vgpu_tsg_set_interleave,
 	},
 	.netlist = {
 		.get_netlist_name = gp10b_netlist_get_name,
diff --git a/drivers/gpu/nvgpu/common/vgpu/gv11b/vgpu_hal_gv11b.c b/drivers/gpu/nvgpu/common/vgpu/gv11b/vgpu_hal_gv11b.c
index 80ac08219..adc2dca54 100644
--- a/drivers/gpu/nvgpu/common/vgpu/gv11b/vgpu_hal_gv11b.c
+++ b/drivers/gpu/nvgpu/common/vgpu/gv11b/vgpu_hal_gv11b.c
@@ -595,7 +595,6 @@ static const struct gpu_ops vgpu_gv11b_ops = {
 		.reschedule = NULL,
 		.update_for_channel = vgpu_runlist_update_for_channel,
 		.reload = vgpu_runlist_reload,
-		.set_interleave = vgpu_runlist_set_interleave,
 		.count_max = gv11b_runlist_count_max,
 		.entry_size = vgpu_runlist_entry_size,
 		.length_max = vgpu_runlist_length_max,
@@ -649,6 +648,7 @@ static const struct gpu_ops vgpu_gv11b_ops = {
 		.post_event_id = nvgpu_tsg_post_event_id,
 		.set_timeslice = vgpu_tsg_set_timeslice,
 		.default_timeslice_us = vgpu_tsg_default_timeslice_us,
+		.set_interleave = vgpu_tsg_set_interleave,
 	},
 	.usermode = {
 		.setup_hw = NULL,
diff --git a/drivers/gpu/nvgpu/common/vgpu/tsg_vgpu.c b/drivers/gpu/nvgpu/common/vgpu/tsg_vgpu.c
index 621ed911b..d849e8695 100644
--- a/drivers/gpu/nvgpu/common/vgpu/tsg_vgpu.c
+++ b/drivers/gpu/nvgpu/common/vgpu/tsg_vgpu.c
@@ -178,3 +178,21 @@ int vgpu_set_sm_exception_type_mask(struct channel_gk20a *ch,
 	return err;
 }
 
+int vgpu_tsg_set_interleave(struct tsg_gk20a *tsg, u32 new_level)
+{
+	struct tegra_vgpu_cmd_msg msg = {0};
+	struct tegra_vgpu_tsg_runlist_interleave_params *p =
+			&msg.params.tsg_interleave;
+	int err;
+	struct gk20a *g = tsg->g;
+
+	nvgpu_log_fn(g, " ");
+
+	msg.cmd = TEGRA_VGPU_CMD_TSG_SET_RUNLIST_INTERLEAVE;
+	msg.handle = vgpu_get_handle(g);
+	p->tsg_id = tsg->tsgid;
+	p->level = new_level;
+	err = vgpu_comm_sendrecv(&msg, sizeof(msg), sizeof(msg));
+	WARN_ON(err || msg.ret);
+	return err ? err : msg.ret;
+}
diff --git a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.h b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.h
index a2109260c..58daeaf73 100644
--- a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.h
@@ -33,11 +33,6 @@ struct nvgpu_semaphore;
 struct channel_gk20a;
 struct tsg_gk20a;
 
-#define NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_LOW     0U
-#define NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_MEDIUM  1U
-#define NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_HIGH    2U
-#define NVGPU_FIFO_RUNLIST_INTERLEAVE_NUM_LEVELS    3U
-
 #define MAX_RUNLIST_BUFFERS		2U
 
 #define FIFO_INVAL_ENGINE_ID		(~U32(0U))
diff --git a/drivers/gpu/nvgpu/hal/fifo/runlist_gk20a.c b/drivers/gpu/nvgpu/hal/fifo/runlist_gk20a.c
index 5c5a198ee..338648f2f 100644
--- a/drivers/gpu/nvgpu/hal/fifo/runlist_gk20a.c
+++ b/drivers/gpu/nvgpu/hal/fifo/runlist_gk20a.c
@@ -118,18 +118,6 @@ int gk20a_fifo_reschedule_preempt_next(struct channel_gk20a *ch,
 	return ret;
 }
 
-int gk20a_runlist_set_interleave(struct gk20a *g,
-				u32 id,
-				u32 runlist_id,
-				u32 new_level)
-{
-	nvgpu_log_fn(g, " ");
-
-	g->fifo.tsg[id].interleave_level = new_level;
-
-	return 0;
-}
-
 u32 gk20a_runlist_count_max(void)
 {
 	return fifo_eng_runlist_base__size_1_v();
@@ -160,7 +148,7 @@ void gk20a_runlist_get_tsg_entry(struct tsg_gk20a *tsg,
 	}
 
 	if (scale > RL_MAX_TIMESLICE_SCALE) {
-		nvgpu_err(g, "requested timeslice value is clamped\n");
+		nvgpu_err(g, "requested timeslice value is clamped");
 		timeout = RL_MAX_TIMESLICE_TIMEOUT;
 		scale = RL_MAX_TIMESLICE_SCALE;
 	}
diff --git a/drivers/gpu/nvgpu/hal/fifo/runlist_gk20a.h b/drivers/gpu/nvgpu/hal/fifo/runlist_gk20a.h
index ca23da572..5a032a62f 100644
--- a/drivers/gpu/nvgpu/hal/fifo/runlist_gk20a.h
+++ b/drivers/gpu/nvgpu/hal/fifo/runlist_gk20a.h
@@ -31,10 +31,6 @@ struct gk20a;
 int gk20a_runlist_reschedule(struct channel_gk20a *ch, bool preempt_next);
 int gk20a_fifo_reschedule_preempt_next(struct channel_gk20a *ch,
 		bool wait_preempt);
-int gk20a_runlist_set_interleave(struct gk20a *g,
-		u32 id,
-		u32 runlist_id,
-		u32 new_level);
 u32 gk20a_runlist_count_max(void);
 u32 gk20a_runlist_entry_size(struct gk20a *g);
 u32 gk20a_runlist_length_max(struct gk20a *g);
diff --git a/drivers/gpu/nvgpu/hal/fifo/runlist_gv11b.c b/drivers/gpu/nvgpu/hal/fifo/runlist_gv11b.c
index 7c3c74c10..f84bd65e5 100644
--- a/drivers/gpu/nvgpu/hal/fifo/runlist_gv11b.c
+++ b/drivers/gpu/nvgpu/hal/fifo/runlist_gv11b.c
@@ -64,7 +64,7 @@ void gv11b_runlist_get_tsg_entry(struct tsg_gk20a *tsg,
 	}
 
 	if (scale > RL_MAX_TIMESLICE_SCALE) {
-		nvgpu_err(g, "requested timeslice value is clamped\n");
+		nvgpu_err(g, "requested timeslice value is clamped");
 		timeout = RL_MAX_TIMESLICE_TIMEOUT;
 		scale = RL_MAX_TIMESLICE_SCALE;
 	}
@@ -76,7 +76,7 @@ void gv11b_runlist_get_tsg_entry(struct tsg_gk20a *tsg,
 	runlist[2] = ram_rl_entry_tsg_tsgid_f(tsg->tsgid);
 	runlist[3] = 0;
 
-	nvgpu_log_info(g, "gv11b tsg runlist [0] %x [1]  %x [2] %x [3] %x\n",
+	nvgpu_log_info(g, "gv11b tsg runlist [0] %x [1]  %x [2] %x [3] %x",
 		runlist[0], runlist[1], runlist[2], runlist[3]);
 
 }
@@ -116,7 +116,7 @@ void gv11b_runlist_get_ch_entry(struct channel_gk20a *ch, u32 *runlist)
 				ram_rl_entry_chid_f(ch->chid);
 	runlist[3] = ram_rl_entry_chan_inst_ptr_hi_f(addr_hi);
 
-	nvgpu_log_info(g, "gv11b channel runlist [0] %x [1]  %x [2] %x [3] %x\n",
+	nvgpu_log_info(g, "gv11b channel runlist [0] %x [1]  %x [2] %x [3] %x",
 			runlist[0], runlist[1], runlist[2], runlist[3]);
 }
 
diff --git a/drivers/gpu/nvgpu/hal/init/hal_gm20b.c b/drivers/gpu/nvgpu/hal/init/hal_gm20b.c
index 474ba89ca..5f07e63a7 100644
--- a/drivers/gpu/nvgpu/hal/init/hal_gm20b.c
+++ b/drivers/gpu/nvgpu/hal/init/hal_gm20b.c
@@ -788,7 +788,6 @@ static const struct gpu_ops gm20b_ops = {
 	.runlist = {
 		.update_for_channel = gk20a_runlist_update_for_channel,
 		.reload = gk20a_runlist_reload,
-		.set_interleave = gk20a_runlist_set_interleave,
 		.count_max = gk20a_runlist_count_max,
 		.entry_size = gk20a_runlist_entry_size,
 		.length_max = gk20a_runlist_length_max,
diff --git a/drivers/gpu/nvgpu/hal/init/hal_gp10b.c b/drivers/gpu/nvgpu/hal/init/hal_gp10b.c
index d5006fdb5..f7f12c1da 100644
--- a/drivers/gpu/nvgpu/hal/init/hal_gp10b.c
+++ b/drivers/gpu/nvgpu/hal/init/hal_gp10b.c
@@ -867,7 +867,6 @@ static const struct gpu_ops gp10b_ops = {
 		.reschedule_preempt_next_locked = gk20a_fifo_reschedule_preempt_next,
 		.update_for_channel = gk20a_runlist_update_for_channel,
 		.reload = gk20a_runlist_reload,
-		.set_interleave = gk20a_runlist_set_interleave,
 		.count_max = gk20a_runlist_count_max,
 		.entry_size = gk20a_runlist_entry_size,
 		.length_max = gk20a_runlist_length_max,
diff --git a/drivers/gpu/nvgpu/hal/init/hal_gv100.c b/drivers/gpu/nvgpu/hal/init/hal_gv100.c
index 95f7b75dd..7463ce090 100644
--- a/drivers/gpu/nvgpu/hal/init/hal_gv100.c
+++ b/drivers/gpu/nvgpu/hal/init/hal_gv100.c
@@ -1043,7 +1043,6 @@ static const struct gpu_ops gv100_ops = {
 	.runlist = {
 		.update_for_channel = gk20a_runlist_update_for_channel,
 		.reload = gk20a_runlist_reload,
-		.set_interleave = gk20a_runlist_set_interleave,
 		.count_max = gv100_runlist_count_max,
 		.entry_size = gv11b_runlist_entry_size,
 		.length_max = gk20a_runlist_length_max,
diff --git a/drivers/gpu/nvgpu/hal/init/hal_gv11b.c b/drivers/gpu/nvgpu/hal/init/hal_gv11b.c
index c466ac024..e1e033110 100644
--- a/drivers/gpu/nvgpu/hal/init/hal_gv11b.c
+++ b/drivers/gpu/nvgpu/hal/init/hal_gv11b.c
@@ -1018,7 +1018,6 @@ static const struct gpu_ops gv11b_ops = {
 		.reschedule_preempt_next_locked = gk20a_fifo_reschedule_preempt_next,
 		.update_for_channel = gk20a_runlist_update_for_channel,
 		.reload = gk20a_runlist_reload,
-		.set_interleave = gk20a_runlist_set_interleave,
 		.count_max = gv11b_runlist_count_max,
 		.entry_size = gv11b_runlist_entry_size,
 		.length_max = gk20a_runlist_length_max,
diff --git a/drivers/gpu/nvgpu/hal/init/hal_tu104.c b/drivers/gpu/nvgpu/hal/init/hal_tu104.c
index 836dc7496..83d53b1f3 100644
--- a/drivers/gpu/nvgpu/hal/init/hal_tu104.c
+++ b/drivers/gpu/nvgpu/hal/init/hal_tu104.c
@@ -1079,7 +1079,6 @@ static const struct gpu_ops tu104_ops = {
 	.runlist = {
 		.update_for_channel = gk20a_runlist_update_for_channel,
 		.reload = gk20a_runlist_reload,
-		.set_interleave = gk20a_runlist_set_interleave,
 		.count_max = tu104_runlist_count_max,
 		.entry_size = tu104_runlist_entry_size,
 		.length_max = gk20a_runlist_length_max,
diff --git a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
index c18b698d3..8ff1cff16 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/gk20a.h
@@ -1076,8 +1076,6 @@ struct gpu_ops {
 				bool wait_for_finish);
 		int (*reload)(struct gk20a *g, u32 runlist_id,
 				bool add, bool wait_for_finish);
-		int (*set_interleave)(struct gk20a *g, u32 id,
-					u32 runlist_id, u32 new_level);
 		u32 (*count_max)(void);
 		u32 (*entry_size)(struct gk20a *g);
 		u32 (*length_max)(struct gk20a *g);
@@ -1241,6 +1239,7 @@ struct gpu_ops {
 				      enum nvgpu_event_id_type event_id);
 		int (*set_timeslice)(struct tsg_gk20a *tsg, u32 timeslice_us);
 		u32 (*default_timeslice_us)(struct gk20a *g);
+		int (*set_interleave)(struct tsg_gk20a *tsg, u32 new_level);
 	} tsg;
 	struct {
 		void (*setup_hw)(struct gk20a *g);
diff --git a/drivers/gpu/nvgpu/include/nvgpu/runlist.h b/drivers/gpu/nvgpu/include/nvgpu/runlist.h
index 1a9421385..2ef34fe25 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/runlist.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/runlist.h
@@ -31,8 +31,13 @@ struct tsg_gk20a;
 struct fifo_gk20a;
 struct channel_gk20a;
 
-#define RUNLIST_APPEND_FAILURE	U32_MAX
-#define RUNLIST_INVALID_ID	U32_MAX
+#define NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_LOW     0U
+#define NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_MEDIUM  1U
+#define NVGPU_FIFO_RUNLIST_INTERLEAVE_LEVEL_HIGH    2U
+#define NVGPU_FIFO_RUNLIST_INTERLEAVE_NUM_LEVELS    3U
+
+#define RUNLIST_APPEND_FAILURE U32_MAX
+#define RUNLIST_INVALID_ID U32_MAX
 
 u32 nvgpu_runlist_construct_locked(struct fifo_gk20a *f,
 				struct fifo_runlist_info_gk20a *runlist,
diff --git a/drivers/gpu/nvgpu/include/nvgpu/tsg.h b/drivers/gpu/nvgpu/include/nvgpu/tsg.h
index 182907404..10ac55b91 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/tsg.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/tsg.h
@@ -129,6 +129,7 @@ u32 nvgpu_tsg_get_timeslice(struct tsg_gk20a *tsg);
 u32 nvgpu_tsg_default_timeslice_us(struct gk20a *g);
 void nvgpu_tsg_enable_sched(struct gk20a *g, struct tsg_gk20a *tsg);
 void nvgpu_tsg_disable_sched(struct gk20a *g, struct tsg_gk20a *tsg);
+int nvgpu_tsg_set_interleave(struct tsg_gk20a *tsg, u32 level);
 int gk20a_tsg_set_priority(struct gk20a *g, struct tsg_gk20a *tsg,
 				u32 priority);
 int gk20a_tsg_alloc_sm_error_states_mem(struct gk20a *g,
diff --git a/drivers/gpu/nvgpu/os/linux/ioctl_tsg.c b/drivers/gpu/nvgpu/os/linux/ioctl_tsg.c
index 0812a612f..327de6bb9 100644
--- a/drivers/gpu/nvgpu/os/linux/ioctl_tsg.c
+++ b/drivers/gpu/nvgpu/os/linux/ioctl_tsg.c
@@ -518,7 +518,7 @@ static int gk20a_tsg_ioctl_set_runlist_interleave(struct gk20a *g,
 	}
 
 	level = nvgpu_get_common_runlist_level(level);
-	err = gk20a_tsg_set_runlist_interleave(tsg, level);
+	err = nvgpu_tsg_set_interleave(tsg, level);
 
 	gk20a_idle(g);
 done:
diff --git a/drivers/gpu/nvgpu/os/linux/sched.c b/drivers/gpu/nvgpu/os/linux/sched.c
index 4e5d376e9..41e132154 100644
--- a/drivers/gpu/nvgpu/os/linux/sched.c
+++ b/drivers/gpu/nvgpu/os/linux/sched.c
@@ -276,7 +276,7 @@ static int gk20a_sched_dev_ioctl_tsg_set_runlist_interleave(
 	if (err)
 		goto done;
 
-	err = gk20a_tsg_set_runlist_interleave(tsg, arg->runlist_interleave);
+	err = nvgpu_tsg_set_interleave(tsg, arg->runlist_interleave);
 
 	gk20a_idle(g);