diff --git a/arch/arm64/boot/dts/rockchip/rk3566-eink.dtsi b/arch/arm64/boot/dts/rockchip/rk3566-eink.dtsi
index cf4f9f89418d..913c9d170a8e 100644
--- a/arch/arm64/boot/dts/rockchip/rk3566-eink.dtsi
+++ b/arch/arm64/boot/dts/rockchip/rk3566-eink.dtsi
@@ -24,6 +24,8 @@
 		eink_tcon = <&eink>;
 		memory-region = <&display_reserved>;
 		waveform-region = <&waveform_reserved>;
+		nvmem-cells = <&cpu_code>;
+		nvmem-cell-names = "cpu-code";
 		status = "okay";
 	};
 };
diff --git a/arch/arm64/boot/dts/rockchip/rk3576-eink.dtsi b/arch/arm64/boot/dts/rockchip/rk3576-eink.dtsi
index e9959357b042..c615ded870f8 100644
--- a/arch/arm64/boot/dts/rockchip/rk3576-eink.dtsi
+++ b/arch/arm64/boot/dts/rockchip/rk3576-eink.dtsi
@@ -29,6 +29,8 @@
 		ebc_tcon = <&ebc>;
 		memory-region = <&display_reserved>;
 		waveform-region = <&waveform_reserved>;
+		nvmem-cells = <&cpu_code>;
+		nvmem-cell-names = "cpu-code";
 		status = "okay";
 	};
 };