dax: Use dax_write_cache* helpers

Use dax_write_cache() and dax_write_cache_enabled() instead of open coding
the bit operations.

Signed-off-by: Ross Zwisler <ross.zwisler@linux.intel.com>
Signed-off-by: Dan Williams <dan.j.williams@intel.com>

1 files changed, 3 insertions, 6 deletions

diff --git a/drivers/dax/super.c b/drivers/dax/super.c
index 2b2332b605e4..c2c46f96b18c 100644
--- a/drivers/dax/super.c
+++ b/drivers/dax/super.c
@@ -182,8 +182,7 @@ static ssize_t write_cache_show(struct device *dev,
         if (!dax_dev)
                 return -ENXIO;
 
-        rc = sprintf(buf, "%d\n", !!test_bit(DAXDEV_WRITE_CACHE,
-                                &dax_dev->flags));
+        rc = sprintf(buf, "%d\n", !!dax_write_cache_enabled(dax_dev));
         put_dax(dax_dev);
         return rc;
 }
@@ -201,10 +200,8 @@ static ssize_t write_cache_store(struct device *dev,
 
         if (rc)
                 len = rc;
-        else if (write_cache)
-                set_bit(DAXDEV_WRITE_CACHE, &dax_dev->flags);
         else
-                clear_bit(DAXDEV_WRITE_CACHE, &dax_dev->flags);
+                dax_write_cache(dax_dev, write_cache);
 
         put_dax(dax_dev);
         return len;
@@ -286,7 +283,7 @@ EXPORT_SYMBOL_GPL(dax_copy_from_iter);
 void arch_wb_cache_pmem(void *addr, size_t size);
 void dax_flush(struct dax_device *dax_dev, void *addr, size_t size)
 {
-        if (unlikely(!test_bit(DAXDEV_WRITE_CACHE, &dax_dev->flags)))
+        if (unlikely(!dax_write_cache_enabled(dax_dev)))
                 return;
 
         arch_wb_cache_pmem(addr, size);