#include <soc/bcm2835/raspberrypi-firmware.h>
 
-#define RPI_FIRMWARE_ARM_CLK_ID                0x00000003
+enum rpi_firmware_clk_id {
+       RPI_FIRMWARE_EMMC_CLK_ID = 1,
+       RPI_FIRMWARE_UART_CLK_ID,
+       RPI_FIRMWARE_ARM_CLK_ID,
+       RPI_FIRMWARE_CORE_CLK_ID,
+       RPI_FIRMWARE_V3D_CLK_ID,
+       RPI_FIRMWARE_H264_CLK_ID,
+       RPI_FIRMWARE_ISP_CLK_ID,
+       RPI_FIRMWARE_SDRAM_CLK_ID,
+       RPI_FIRMWARE_PIXEL_CLK_ID,
+       RPI_FIRMWARE_PWM_CLK_ID,
+       RPI_FIRMWARE_HEVC_CLK_ID,
+       RPI_FIRMWARE_EMMC2_CLK_ID,
+       RPI_FIRMWARE_M2MC_CLK_ID,
+       RPI_FIRMWARE_PIXEL_BVB_CLK_ID,
+       RPI_FIRMWARE_NUM_CLK_ID,
+};
 
 #define RPI_FIRMWARE_STATE_ENABLE_BIT  BIT(0)
 #define RPI_FIRMWARE_STATE_WAIT_BIT    BIT(1)
 
 #define A2W_PLL_FRAC_BITS              20
 
-#define NUM_FW_CLKS                    16
-
 struct raspberrypi_clk {
        struct device *dev;
        struct rpi_firmware *firmware;
        rpi->firmware = firmware;
        platform_set_drvdata(pdev, rpi);
 
-       clk_data = devm_kzalloc(dev, struct_size(clk_data, hws, NUM_FW_CLKS),
+       clk_data = devm_kzalloc(dev, struct_size(clk_data, hws,
+                                                RPI_FIRMWARE_NUM_CLK_ID),
                                GFP_KERNEL);
        if (!clk_data)
                return -ENOMEM;