DEFINE_RANGE_EVENT(iomap_release_folio);
 DEFINE_RANGE_EVENT(iomap_invalidate_folio);
 DEFINE_RANGE_EVENT(iomap_dio_invalidate_fail);
+DEFINE_RANGE_EVENT(iomap_dio_rw_queued);
 
 #define IOMAP_TYPE_STRINGS \
        { IOMAP_HOLE,           "HOLE" }, \
        { IOMAP_F_BUFFER_HEAD,  "BH" }, \
        { IOMAP_F_SIZE_CHANGED, "SIZE_CHANGED" }
 
+#define IOMAP_DIO_STRINGS \
+       {IOMAP_DIO_FORCE_WAIT,  "DIO_FORCE_WAIT" }, \
+       {IOMAP_DIO_OVERWRITE_ONLY, "DIO_OVERWRITE_ONLY" }, \
+       {IOMAP_DIO_PARTIAL,     "DIO_PARTIAL" }
+
 DECLARE_EVENT_CLASS(iomap_class,
        TP_PROTO(struct inode *inode, struct iomap *iomap),
        TP_ARGS(inode, iomap),
                   (void *)__entry->caller)
 );
 
+TRACE_EVENT(iomap_dio_rw_begin,
+       TP_PROTO(struct kiocb *iocb, struct iov_iter *iter,
+                unsigned int dio_flags, size_t done_before),
+       TP_ARGS(iocb, iter, dio_flags, done_before),
+       TP_STRUCT__entry(
+               __field(dev_t,  dev)
+               __field(ino_t,  ino)
+               __field(loff_t, isize)
+               __field(loff_t, pos)
+               __field(size_t, count)
+               __field(size_t, done_before)
+               __field(int,    ki_flags)
+               __field(unsigned int,   dio_flags)
+               __field(bool,   aio)
+       ),
+       TP_fast_assign(
+               __entry->dev = file_inode(iocb->ki_filp)->i_sb->s_dev;
+               __entry->ino = file_inode(iocb->ki_filp)->i_ino;
+               __entry->isize = file_inode(iocb->ki_filp)->i_size;
+               __entry->pos = iocb->ki_pos;
+               __entry->count = iov_iter_count(iter);
+               __entry->done_before = done_before;
+               __entry->ki_flags = iocb->ki_flags;
+               __entry->dio_flags = dio_flags;
+               __entry->aio = !is_sync_kiocb(iocb);
+       ),
+       TP_printk("dev %d:%d ino 0x%lx size 0x%llx offset 0x%llx length 0x%zx done_before 0x%zx flags %s dio_flags %s aio %d",
+                 MAJOR(__entry->dev), MINOR(__entry->dev),
+                 __entry->ino,
+                 __entry->isize,
+                 __entry->pos,
+                 __entry->count,
+                 __entry->done_before,
+                 __print_flags(__entry->ki_flags, "|", TRACE_IOCB_STRINGS),
+                 __print_flags(__entry->dio_flags, "|", IOMAP_DIO_STRINGS),
+                 __entry->aio)
+);
+
+TRACE_EVENT(iomap_dio_complete,
+       TP_PROTO(struct kiocb *iocb, int error, ssize_t ret),
+       TP_ARGS(iocb, error, ret),
+       TP_STRUCT__entry(
+               __field(dev_t,  dev)
+               __field(ino_t,  ino)
+               __field(loff_t, isize)
+               __field(loff_t, pos)
+               __field(int,    ki_flags)
+               __field(bool,   aio)
+               __field(int,    error)
+               __field(ssize_t, ret)
+       ),
+       TP_fast_assign(
+               __entry->dev = file_inode(iocb->ki_filp)->i_sb->s_dev;
+               __entry->ino = file_inode(iocb->ki_filp)->i_ino;
+               __entry->isize = file_inode(iocb->ki_filp)->i_size;
+               __entry->pos = iocb->ki_pos;
+               __entry->ki_flags = iocb->ki_flags;
+               __entry->aio = !is_sync_kiocb(iocb);
+               __entry->error = error;
+               __entry->ret = ret;
+       ),
+       TP_printk("dev %d:%d ino 0x%lx size 0x%llx offset 0x%llx flags %s aio %d error %d ret %zd",
+                 MAJOR(__entry->dev), MINOR(__entry->dev),
+                 __entry->ino,
+                 __entry->isize,
+                 __entry->pos,
+                 __print_flags(__entry->ki_flags, "|", TRACE_IOCB_STRINGS),
+                 __entry->aio,
+                 __entry->error,
+                 __entry->ret)
+);
+
 #endif /* _IOMAP_TRACE_H */
 
 #undef TRACE_INCLUDE_PATH