ARM: dts: rockchip: Add pin names for rk3288-veyron-minnie

[linux.git] / drivers / ata / libata-transport.c

diff --git a/drivers/ata/libata-transport.c b/drivers/ata/libata-transport.c
index 19e6e539a061b93a6ac81975a69b3cdba7b1d699..a0b0b4d986f20fe813d565113e23097dee761102 100644 (file)
--- a/drivers/ata/libata-transport.c
+++ b/drivers/ata/libata-transport.c
@@ -224,6 +224,8 @@ static DECLARE_TRANSPORT_CLASS(ata_port_class,
 
 static void ata_tport_release(struct device *dev)
 {
+       struct ata_port *ap = tdev_to_port(dev);
+       ata_host_put(ap->host);
 }
 
 /**
@@ -284,6 +286,7 @@ int ata_tport_add(struct device *parent,
        dev->type = &ata_port_type;
 
        dev->parent = parent;
+       ata_host_get(ap->host);
        dev->release = ata_tport_release;
        dev_set_name(dev, "ata%d", ap->print_id);
        transport_setup_device(dev);
@@ -314,6 +317,7 @@ int ata_tport_add(struct device *parent,
  tport_err:
        transport_destroy_device(dev);
        put_device(dev);
+       ata_host_put(ap->host);
        return error;
 }