return valid;
 }
 
-struct ffa_device *ffa_device_register(const uuid_t *uuid, int vm_id)
+struct ffa_device *ffa_device_register(const uuid_t *uuid, int vm_id,
+                                      const struct ffa_dev_ops *ops)
 {
        int ret;
        struct device *dev;
        dev_set_name(&ffa_dev->dev, "arm-ffa-%04x", vm_id);
 
        ffa_dev->vm_id = vm_id;
+       ffa_dev->ops = ops;
        uuid_copy(&ffa_dev->uuid, uuid);
 
        ret = device_register(&ffa_dev->dev);
 
                 * as part of the discovery API, we need to pass the
                 * discovered UUID here instead.
                 */
-               ffa_dev = ffa_device_register(&uuid_null, tpbuf->id);
+               ffa_dev = ffa_device_register(&uuid_null, tpbuf->id, &ffa_ops);
                if (!ffa_dev) {
                        pr_err("%s: failed to register partition ID 0x%x\n",
                               __func__, tpbuf->id);
 
        bool mode_32bit;
        uuid_t uuid;
        struct device dev;
+       const struct ffa_dev_ops *ops;
 };
 
 #define to_ffa_dev(d) container_of(d, struct ffa_device, dev)
 }
 
 #if IS_REACHABLE(CONFIG_ARM_FFA_TRANSPORT)
-struct ffa_device *ffa_device_register(const uuid_t *uuid, int vm_id);
+struct ffa_device *ffa_device_register(const uuid_t *uuid, int vm_id,
+                                      const struct ffa_dev_ops *ops);
 void ffa_device_unregister(struct ffa_device *ffa_dev);
 int ffa_driver_register(struct ffa_driver *driver, struct module *owner,
                        const char *mod_name);
 
 #else
 static inline
-struct ffa_device *ffa_device_register(const uuid_t *uuid, int vm_id)
+struct ffa_device *ffa_device_register(const uuid_t *uuid, int vm_id,
+                                      const struct ffa_dev_ops *ops)
 {
        return NULL;
 }