diff --git a/drivers/platform/tegra/rtcpu/ivc-bus.c b/drivers/platform/tegra/rtcpu/ivc-bus.c
index dae868b7..8d97516a 100644
--- a/drivers/platform/tegra/rtcpu/ivc-bus.c
+++ b/drivers/platform/tegra/rtcpu/ivc-bus.c
@@ -9,7 +9,11 @@
 #include <linux/of_device.h>
 #include <linux/pm_runtime.h>
 #include <linux/slab.h>
+#if defined(NV_TEGRA_IVC_USE_IVC_EXT_DRIVER)
 #include <soc/tegra/ivc_ext.h>
+#else
+#include <soc/tegra/ivc-priv.h>
+#endif
 #include <linux/tegra-ivc-bus.h>
 #include <linux/bitops.h>
 #include <linux/version.h>
@@ -201,7 +205,11 @@ static struct tegra_ivc_channel *tegra_ivc_channel_create(
 
 	chan->group = channel_group;
 
+#if defined(NV_TEGRA_IVC_USE_IVC_EXT_DRIVER)
 	tegra_ivc_channel_reset(&chan->ivc);
+#else
+	tegra_ivc_reset(&chan->ivc);
+#endif
 
 	/* Fill channel descriptor */
 	tlv = (struct camrtc_tlv_ivc_setup *)
@@ -245,7 +253,11 @@ static void tegra_ivc_channel_notify(struct tegra_ivc_channel *chan)
 {
 	const struct tegra_ivc_channel_ops *ops;
 
+#if defined(NV_TEGRA_IVC_USE_IVC_EXT_DRIVER)
 	if (tegra_ivc_channel_notified(&chan->ivc) != 0)
+#else
+	if (tegra_ivc_notified(&chan->ivc) != 0)
+#endif
 		return;
 
 	if (!chan->is_ready)