#include <linux/kernel.h>
 #include <linux/init.h>
 #include <linux/module.h>
+#include <linux/of_device.h>
 #include <linux/slab.h>
 #include <linux/delay.h>
 #include <linux/mutex.h>
 };
 MODULE_DEVICE_TABLE(i2c, at24_ids);
 
+static const struct of_device_id at24_of_match[] = {
+       {
+               .compatible = "atmel,24c00",
+               .data = (void *)AT24_DEVICE_MAGIC(128 / 8, AT24_FLAG_TAKE8ADDR)
+       },
+       {
+               .compatible = "atmel,24c01",
+               .data = (void *)AT24_DEVICE_MAGIC(1024 / 8, 0)
+       },
+       {
+               .compatible = "atmel,24c02",
+               .data = (void *)AT24_DEVICE_MAGIC(2048 / 8, 0)
+       },
+       {
+               .compatible = "atmel,spd",
+               .data = (void *)AT24_DEVICE_MAGIC(2048 / 8,
+                               AT24_FLAG_READONLY | AT24_FLAG_IRUGO)
+       },
+       {
+               .compatible = "atmel,24c04",
+               .data = (void *)AT24_DEVICE_MAGIC(4096 / 8, 0)
+       },
+       {
+               .compatible = "atmel,24c08",
+               .data = (void *)AT24_DEVICE_MAGIC(8192 / 8, 0)
+       },
+       {
+               .compatible = "atmel,24c16",
+               .data = (void *)AT24_DEVICE_MAGIC(16384 / 8, 0)
+       },
+       {
+               .compatible = "atmel,24c32",
+               .data = (void *)AT24_DEVICE_MAGIC(32768 / 8, AT24_FLAG_ADDR16)
+       },
+       {
+               .compatible = "atmel,24c64",
+               .data = (void *)AT24_DEVICE_MAGIC(65536 / 8, AT24_FLAG_ADDR16)
+       },
+       {
+               .compatible = "atmel,24c128",
+               .data = (void *)AT24_DEVICE_MAGIC(131072 / 8, AT24_FLAG_ADDR16)
+       },
+       {
+               .compatible = "atmel,24c256",
+               .data = (void *)AT24_DEVICE_MAGIC(262144 / 8, AT24_FLAG_ADDR16)
+       },
+       {
+               .compatible = "atmel,24c512",
+               .data = (void *)AT24_DEVICE_MAGIC(524288 / 8, AT24_FLAG_ADDR16)
+       },
+       {
+               .compatible = "atmel,24c1024",
+               .data = (void *)AT24_DEVICE_MAGIC(1048576 / 8, AT24_FLAG_ADDR16)
+       },
+       { },
+};
+MODULE_DEVICE_TABLE(of, at24_of_match);
+
 static const struct acpi_device_id at24_acpi_ids[] = {
        { "INT3499", AT24_DEVICE_MAGIC(8192 / 8, 0) },
        { }
        if (client->dev.platform_data) {
                chip = *(struct at24_platform_data *)client->dev.platform_data;
        } else {
-               if (id) {
+               /*
+                * The I2C core allows OF nodes compatibles to match against the
+                * I2C device ID table as a fallback, so check not only if an OF
+                * node is present but also if it matches an OF device ID entry.
+                */
+               if (client->dev.of_node &&
+                   of_match_device(at24_of_match, &client->dev)) {
+                       magic = (kernel_ulong_t)
+                               of_device_get_match_data(&client->dev);
+               } else if (id) {
                        magic = id->driver_data;
                } else {
                        const struct acpi_device_id *aid;
 static struct i2c_driver at24_driver = {
        .driver = {
                .name = "at24",
+               .of_match_table = at24_of_match,
                .acpi_match_table = ACPI_PTR(at24_acpi_ids),
        },
        .probe = at24_probe,