diff --git a/arch/arm64/boot/dts/rockchip/rk3399-android.dtsi b/arch/arm64/boot/dts/rockchip/rk3399-android.dtsi
index a00953700c8a..720a902c1006 100644
--- a/arch/arm64/boot/dts/rockchip/rk3399-android.dtsi
+++ b/arch/arm64/boot/dts/rockchip/rk3399-android.dtsi
@@ -218,6 +218,8 @@
 		power-domains = <&power RK3399_PD_VCODEC>;
 		name = "vpu_service";
 		dev_mode = <0>;
+		/* 0 means ion, 1 means drm */
+		allocator = <0>;
 	};
 
 	vpu_mmu: vpu_mmu {
@@ -242,6 +244,8 @@
 		power-domains = <&power RK3399_PD_VDU>;
 		dev_mode = <2>;
 		name = "rkvdec";
+		/* 0 means ion, 1 means drm */
+		allocator = <0>;
 	};
 
 	vdec_mmu: vdec_mmu {