BUG: scheduling while atomic in __iomap_dio

BUG: scheduling while atomic: syz-executor.1/8382/0x00000002
Modules linked in:
Preemption disabled at:
[<ffffffff8165ea9a>] is_module_text_address+0x1a/0x140 kernel/module.c:4811
CPU: 1 PID: 8382 Comm: syz-executor.1 Tainted: G        W         5.15.148-syzkaller-00718-g993bed180178 #0
Hardware name: Google Google Compute Engine/Google Compute Engine, BIOS Google 03/27/2024
Call Trace:
 <TASK>
 __dump_stack lib/dump_stack.c:88 [inline]
 dump_stack_lvl+0x151/0x1b7 lib/dump_stack.c:106
 dump_stack+0x15/0x17 lib/dump_stack.c:113
 __schedule_bug+0x195/0x260 kernel/sched/core.c:5707
 schedule_debug kernel/sched/core.c:5734 [inline]
 __schedule+0xd19/0x1590 kernel/sched/core.c:6402
 schedule+0x11f/0x1e0 kernel/sched/core.c:6595
 schedule_timeout+0x18c/0x370 kernel/time/timer.c:1890
 io_schedule_timeout+0x9e/0x130 kernel/sched/core.c:8629
 blk_io_schedule+0x58/0x70 block/blk-core.c:1774
 __iomap_dio_rw+0x1a52/0x1ed0 fs/iomap/direct-io.c:658
 iomap_dio_rw+0x3e/0x90 fs/iomap/direct-io.c:680
 ext4_dio_write_iter fs/ext4/file.c:572 [inline]
 ext4_file_write_iter+0x111c/0x1c80 fs/ext4/file.c:683
 call_write_iter include/linux/fs.h:2202 [inline]
 new_sync_write fs/read_write.c:507 [inline]
 vfs_write+0xd5d/0x1110 fs/read_write.c:594
 ksys_write+0x199/0x2c0 fs/read_write.c:647
 __do_sys_write fs/read_write.c:659 [inline]
 __se_sys_write fs/read_write.c:656 [inline]
 __x64_sys_write+0x7b/0x90 fs/read_write.c:656
 do_syscall_x64 arch/x86/entry/common.c:50 [inline]
 do_syscall_64+0x3d/0xb0 arch/x86/entry/common.c:80
 entry_SYSCALL_64_after_hwframe+0x61/0xcb
RIP: 0033:0x7f7618c25e69
Code: 28 00 00 00 75 05 48 83 c4 28 c3 e8 e1 20 00 00 90 48 89 f8 48 89 f7 48 89 d6 48 89 ca 4d 89 c2 4d 89 c8 4c 8b 4c 24 08 0f 05 <48> 3d 01 f0 ff ff 73 01 c3 48 c7 c1 b0 ff ff ff f7 d8 64 89 01 48
RSP: 002b:00007f76179990c8 EFLAGS: 00000246 ORIG_RAX: 0000000000000001
RAX: ffffffffffffffda RBX: 00007f7618d53f80 RCX: 00007f7618c25e69
RDX: 0000000000ffe000 RSI: 0000000020000000 RDI: 0000000000000005
RBP: 00007f7618c7247a R08: 0000000000000000 R09: 0000000000000000
R10: 0000000000000000 R11: 0000000000000246 R12: 0000000000000000
R13: 000000000000000b R14: 00007f7618d53f80 R15: 00007ffc7c8c4d68
 </TASK>