diff --git a/drivers/video/tegra/tsec/tsec_boot.c b/drivers/video/tegra/tsec/tsec_boot.c
index c9f74e9d..6b39945c 100644
--- a/drivers/video/tegra/tsec/tsec_boot.c
+++ b/drivers/video/tegra/tsec/tsec_boot.c
@@ -173,10 +173,7 @@ static int tsec_read_img_and_desc(struct platform_device *dev,
 	for (w = 0; w < fw_image->size/sizeof(u32); w++)
 		rv_data->backdoor_img_va[w] = le32_to_cpu(((__le32 *)fw_image->data)[w]);
 #if (KERNEL_VERSION(5, 14, 0) <= LINUX_VERSION_CODE)
-#ifdef CONFIG_EXPORT_DCACHE_OPS
-	dcache_clean_inval_poc((unsigned long)rv_data->backdoor_img_va,
-		(unsigned long)rv_data->backdoor_img_va + rv_data->backdoor_img_size);
-#endif
+	arch_invalidate_pmem(rv_data->backdoor_img_va, rv_data->backdoor_img_size);
 #else
 	__flush_dcache_area((void *)rv_data->backdoor_img_va, fw_image->size);
 #endif
diff --git a/drivers/video/tegra/tsec/tsec_linux.h b/drivers/video/tegra/tsec/tsec_linux.h
index d56aa9ae..a265d44e 100644
--- a/drivers/video/tegra/tsec/tsec_linux.h
+++ b/drivers/video/tegra/tsec/tsec_linux.h
@@ -35,11 +35,12 @@
 #include <linux/version.h>                 /* for KERNEL_VERSION */
 #include <linux/interrupt.h>               /* for enable_irq */
 #include <linux/firmware.h>                /* for request_firmware */
-#include <asm/cacheflush.h>                /* for __flush_dcache_area */
 #if (KERNEL_VERSION(5, 14, 0) <= LINUX_VERSION_CODE)
 #include <soc/tegra/mc.h>                  /* for tegra_mc_get_carveout_info */
+#include <linux/libnvdimm.h>               /* for arch_invalidate_pmem */
 #else
 #include <linux/platform/tegra/tegra_mc.h> /* for mc_get_carveout_info */
+#include <asm/cacheflush.h>                /* for __flush_dcache_area */
 #endif
 
 #endif /* TSEC_LINUX_H */