return IRQ_HANDLED;
 }
 
-static int tc_probe(struct i2c_client *client, const struct i2c_device_id *id)
+static int tc_probe_edp_bridge_endpoint(struct tc_data *tc)
 {
-       struct device *dev = &client->dev;
+       struct device *dev = tc->dev;
        struct drm_panel *panel;
-       struct tc_data *tc;
        int ret;
 
-       tc = devm_kzalloc(dev, sizeof(*tc), GFP_KERNEL);
-       if (!tc)
-               return -ENOMEM;
-
-       tc->dev = dev;
-
        /* port@2 is the output port */
        ret = drm_of_find_panel_or_bridge(dev->of_node, 2, 0, &panel, NULL);
        if (ret && ret != -ENODEV)
                tc->bridge.type = DRM_MODE_CONNECTOR_DisplayPort;
        }
 
+       return ret;
+}
+
+static int tc_probe(struct i2c_client *client, const struct i2c_device_id *id)
+{
+       struct device *dev = &client->dev;
+       struct tc_data *tc;
+       int ret;
+
+       tc = devm_kzalloc(dev, sizeof(*tc), GFP_KERNEL);
+       if (!tc)
+               return -ENOMEM;
+
+       tc->dev = dev;
+
+       ret = tc_probe_edp_bridge_endpoint(tc);
+       if (ret)
+               return ret;
+
        /* Shut down GPIO is optional */
        tc->sd_gpio = devm_gpiod_get_optional(dev, "shutdown", GPIOD_OUT_HIGH);
        if (IS_ERR(tc->sd_gpio))