3 files changed, 5 insertions, 2 deletions

diff --git a/drivers/ide/dtc2278.c b/drivers/ide/dtc2278.c
index 689b2e493413..c6b138122981 100644
--- a/drivers/ide/dtc2278.c
+++ b/drivers/ide/dtc2278.c
@@ -100,7 +100,8 @@ static const struct ide_port_info dtc2278_port_info __initdata = {
                                   IDE_HFLAG_IO_32BIT |
                                   /* disallow ->io_32bit changes */
                                   IDE_HFLAG_NO_IO_32BIT |
-                                  IDE_HFLAG_NO_DMA,
+                                  IDE_HFLAG_NO_DMA |
+                                  IDE_HFLAG_DTC2278,
         .pio_mask               = ATA_PIO4,
 };
 
diff --git a/drivers/ide/ide-probe.c b/drivers/ide/ide-probe.c
index 75b79cc96339..62270f474681 100644
--- a/drivers/ide/ide-probe.c
+++ b/drivers/ide/ide-probe.c
@@ -1061,7 +1061,7 @@ static void ide_init_port(ide_hwif_t *hwif, unsigned int port,
                 hwif->tp_ops = d->tp_ops;
 
         /* ->set_pio_mode for DTC2278 is currently limited to port 0 */
-        if (hwif->chipset != ide_dtc2278 || hwif->channel == 0)
+        if ((hwif->host_flags & IDE_HFLAG_DTC2278) == 0 || hwif->channel == 0)
                 hwif->port_ops = d->port_ops;
 
         hwif->swdma_mask = d->swdma_mask;
diff --git a/include/linux/ide.h b/include/linux/ide.h
index 901d323c7bbe..732a05f3de08 100644
--- a/include/linux/ide.h
+++ b/include/linux/ide.h
@@ -1325,6 +1325,8 @@ enum {
         IDE_HFLAG_ERROR_STOPS_FIFO      = (1 << 19),
         /* serialize ports */
         IDE_HFLAG_SERIALIZE             = (1 << 20),
+        /* host is DTC2278 */
+        IDE_HFLAG_DTC2278               = (1 << 21),
         /* host is TRM290 */
         IDE_HFLAG_TRM290                = (1 << 23),
         /* use 32-bit I/O ops */