1 files changed, 14 insertions, 0 deletions

diff --git a/drivers/ide/ide-gd.c b/drivers/ide/ide-gd.c
index 4b6b71e2cdf5..214119026b3f 100644
--- a/drivers/ide/ide-gd.c
+++ b/drivers/ide/ide-gd.c
@@ -287,6 +287,19 @@ static int ide_gd_media_changed(struct gendisk *disk)
         return ret;
 }
 
+static unsigned long long ide_gd_set_capacity(struct gendisk *disk,
+                                              unsigned long long capacity)
+{
+        struct ide_disk_obj *idkp = ide_drv_g(disk, ide_disk_obj);
+        ide_drive_t *drive = idkp->drive;
+        const struct ide_disk_ops *disk_ops = drive->disk_ops;
+
+        if (disk_ops->set_capacity)
+                return disk_ops->set_capacity(drive, capacity);
+
+        return drive->capacity64;
+}
+
 static int ide_gd_revalidate_disk(struct gendisk *disk)
 {
         struct ide_disk_obj *idkp = ide_drv_g(disk, ide_disk_obj);
@@ -315,6 +328,7 @@ static struct block_device_operations ide_gd_ops = {
         .locked_ioctl           = ide_gd_ioctl,
         .getgeo                 = ide_gd_getgeo,
         .media_changed          = ide_gd_media_changed,
+        .set_capacity           = ide_gd_set_capacity,
         .revalidate_disk        = ide_gd_revalidate_disk
 };