new_fl |= S_NOATIME;
        if (flags & EXT4_DIRSYNC_FL)
                new_fl |= S_DIRSYNC;
-       if (test_opt(inode->i_sb, DAX) && S_ISREG(inode->i_mode) &&
-           !ext4_should_journal_data(inode) && !ext4_has_inline_data(inode) &&
-           !(flags & EXT4_ENCRYPT_FL))
+       if (ext4_should_use_dax(inode))
                new_fl |= S_DAX;
 +      if (flags & EXT4_ENCRYPT_FL)
 +              new_fl |= S_ENCRYPTED;
        inode_set_flags(inode, new_fl,
 -                      S_SYNC|S_APPEND|S_IMMUTABLE|S_NOATIME|S_DIRSYNC|S_DAX);
 +                      S_SYNC|S_APPEND|S_IMMUTABLE|S_NOATIME|S_DIRSYNC|S_DAX|
 +                      S_ENCRYPTED);
  }
  
  static blkcnt_t ext4_inode_blocks(struct ext4_inode *raw_inode,