struct platform_device *platform_dev)
 {
        struct device *dev = csi_dev->dev;
-       unsigned long clock_mod_rate;
+       const struct sun6i_csi_variant *variant;
        void __iomem *io_base;
        int ret;
        int irq;
 
+       variant = of_device_get_match_data(dev);
+       if (!variant)
+               return -EINVAL;
+
        /* Registers */
 
        io_base = devm_platform_ioremap_resource(platform_dev, 0);
                return PTR_ERR(csi_dev->clock_ram);
        }
 
-       if (of_device_is_compatible(dev->of_node, "allwinner,sun50i-a64-csi"))
-               clock_mod_rate = 300000000;
-       else
-               clock_mod_rate = 297000000;
-
-       ret = clk_set_rate_exclusive(csi_dev->clock_mod, clock_mod_rate);
+       ret = clk_set_rate_exclusive(csi_dev->clock_mod,
+                                    variant->clock_mod_rate);
        if (ret) {
                dev_err(dev, "failed to set mod clock rate\n");
                return ret;
        return 0;
 }
 
+static const struct sun6i_csi_variant sun6i_a31_csi_variant = {
+       .clock_mod_rate = 297000000,
+};
+
+static const struct sun6i_csi_variant sun50i_a64_csi_variant = {
+       .clock_mod_rate = 300000000,
+};
+
 static const struct of_device_id sun6i_csi_of_match[] = {
-       { .compatible = "allwinner,sun6i-a31-csi", },
-       { .compatible = "allwinner,sun8i-a83t-csi", },
-       { .compatible = "allwinner,sun8i-h3-csi", },
-       { .compatible = "allwinner,sun8i-v3s-csi", },
-       { .compatible = "allwinner,sun50i-a64-csi", },
+       {
+               .compatible     = "allwinner,sun6i-a31-csi",
+               .data           = &sun6i_a31_csi_variant,
+       },
+       {
+               .compatible     = "allwinner,sun8i-a83t-csi",
+               .data           = &sun6i_a31_csi_variant,
+       },
+       {
+               .compatible     = "allwinner,sun8i-h3-csi",
+               .data           = &sun6i_a31_csi_variant,
+       },
+       {
+               .compatible     = "allwinner,sun8i-v3s-csi",
+               .data           = &sun6i_a31_csi_variant,
+       },
+       {
+               .compatible     = "allwinner,sun50i-a64-csi",
+               .data           = &sun50i_a64_csi_variant,
+       },
        {},
 };