#define cls_to_ae_dev(dev) container_of(dev, struct hnae_ae_dev, cls_dev)
 
-static struct class *hnae_class;
+static const struct class hnae_class = {
+       .name = "hnae",
+};
 
 static void
 hnae_list_add(spinlock_t *lock, struct list_head *node, struct list_head *head)
 
        WARN_ON(!fwnode);
 
-       dev = class_find_device(hnae_class, NULL, fwnode, __ae_match);
+       dev = class_find_device(&hnae_class, NULL, fwnode, __ae_match);
 
        return dev ? cls_to_ae_dev(dev) : NULL;
 }
        hdev->owner = owner;
        hdev->id = (int)atomic_inc_return(&id);
        hdev->cls_dev.parent = hdev->dev;
-       hdev->cls_dev.class = hnae_class;
+       hdev->cls_dev.class = &hnae_class;
        hdev->cls_dev.release = hnae_release;
        (void)dev_set_name(&hdev->cls_dev, "hnae%d", hdev->id);
        ret = device_register(&hdev->cls_dev);
 
 static int __init hnae_init(void)
 {
-       hnae_class = class_create("hnae");
-       return PTR_ERR_OR_ZERO(hnae_class);
+       return class_register(&hnae_class);
 }
 
 static void __exit hnae_exit(void)
 {
-       class_destroy(hnae_class);
+       class_unregister(&hnae_class);
 }
 
 subsys_initcall(hnae_init);