diff --git a/arch/arm64/boot/dts/rockchip/rk3308b-amp.dtsi b/arch/arm64/boot/dts/rockchip/rk3308b-amp.dtsi
index 5a4548d8149d..171424d55d1c 100644
--- a/arch/arm64/boot/dts/rockchip/rk3308b-amp.dtsi
+++ b/arch/arm64/boot/dts/rockchip/rk3308b-amp.dtsi
@@ -5,6 +5,8 @@
 
 #include <dt-bindings/soc/rockchip-amp.h>
 
+#define CPU_GET_AFFINITY(cpu, cluster) ((cpu) << 0 | ((cluster) << 8))
+
 / {
 	rockchip_amp: rockchip-amp {
 		compatible = "rockchip,amp";