unsigned int            read_dirty_data:1;
        unsigned int            cache_missed:1;
 
+       struct hd_struct        *part;
        unsigned long           start_time;
 
        struct btree_op         op;
 {
        if (s->orig_bio) {
                /* Count on bcache device */
-               disk_end_io_acct(s->d->disk, bio_op(s->orig_bio), s->start_time);
+               part_end_io_acct(s->part, s->orig_bio, s->start_time);
 
                trace_bcache_request_end(s->d, s->orig_bio);
                s->orig_bio->bi_status = s->iop.status;
        s->write                = op_is_write(bio_op(bio));
        s->read_dirty_data      = 0;
        /* Count on the bcache device */
-       s->start_time           = disk_start_io_acct(d->disk, bio_sectors(bio), bio_op(bio));
+       s->start_time           = part_start_io_acct(d->disk, &s->part, bio);
        s->iop.c                = d->c;
        s->iop.bio              = NULL;
        s->iop.inode            = d->id;
        unsigned long           start_time;
        bio_end_io_t            *bi_end_io;
        void                    *bi_private;
+       struct hd_struct        *part;
 };
 
 static void detached_dev_end_io(struct bio *bio)
        bio->bi_private = ddip->bi_private;
 
        /* Count on the bcache device */
-       disk_end_io_acct(ddip->d->disk, bio_op(bio), ddip->start_time);
+       part_end_io_acct(ddip->part, bio, ddip->start_time);
 
        if (bio->bi_status) {
                struct cached_dev *dc = container_of(ddip->d,
        ddip = kzalloc(sizeof(struct detached_dev_io_private), GFP_NOIO);
        ddip->d = d;
        /* Count on the bcache device */
-       ddip->start_time = disk_start_io_acct(d->disk, bio_sectors(bio), bio_op(bio));
+       ddip->start_time = part_start_io_acct(d->disk, &ddip->part, bio);
        ddip->bi_end_io = bio->bi_end_io;
        ddip->bi_private = bio->bi_private;
        bio->bi_end_io = detached_dev_end_io;