#include <linux/module.h>
 #include <linux/property.h>
 #include <linux/slab.h>
+#include <linux/spi/offload/consumer.h>
 #include <linux/spi/spi.h>
 #include <linux/spinlock.h>
 #include <linux/string.h>
 #include <linux/unaligned.h>
 
 #include <linux/iio/adc/ad_sigma_delta.h>
+#include <linux/iio/buffer-dmaengine.h>
 #include <linux/iio/buffer.h>
 #include <linux/iio/iio.h>
 #include <linux/iio/trigger_consumer.h>
        struct ad_sigma_delta *sigma_delta = iio_device_get_drvdata(indio_dev);
        const struct iio_scan_type *scan_type = &indio_dev->channels[0].scan_type;
        struct spi_transfer *xfer = sigma_delta->sample_xfer;
-       unsigned int i, slot, samples_buf_size;
-       unsigned int channel, scan_size;
+       unsigned int i, slot, channel;
        u8 *samples_buf;
        int ret;
 
        sigma_delta->active_slots = slot;
        sigma_delta->current_slot = 0;
 
-       if (sigma_delta->active_slots > 1) {
-               ret = ad_sigma_delta_append_status(sigma_delta, true);
-               if (ret)
-                       return ret;
-       }
+       if (ad_sigma_delta_has_spi_offload(sigma_delta)) {
+               xfer[1].offload_flags = SPI_OFFLOAD_XFER_RX_STREAM;
+               xfer[1].bits_per_word = scan_type->realbits;
+               xfer[1].len = spi_bpw_to_bytes(scan_type->realbits);
+       } else {
+               unsigned int samples_buf_size, scan_size;
 
-       samples_buf_size = ALIGN(slot * BITS_TO_BYTES(scan_type->storagebits),
-                                sizeof(s64));
-       samples_buf_size += sizeof(s64);
-       samples_buf = devm_krealloc(&sigma_delta->spi->dev, sigma_delta->samples_buf,
-                                   samples_buf_size, GFP_KERNEL);
-       if (!samples_buf)
-               return -ENOMEM;
+               if (sigma_delta->active_slots > 1) {
+                       ret = ad_sigma_delta_append_status(sigma_delta, true);
+                       if (ret)
+                               return ret;
+               }
 
-       sigma_delta->samples_buf = samples_buf;
-       scan_size = BITS_TO_BYTES(scan_type->realbits + scan_type->shift);
-       /* For 24-bit data, there is an extra byte of padding. */
-       xfer[1].rx_buf = &sigma_delta->rx_buf[scan_size == 3 ? 1 : 0];
-       xfer[1].len = scan_size + (sigma_delta->status_appended ? 1 : 0);
+               samples_buf_size =
+                       ALIGN(slot * BITS_TO_BYTES(scan_type->storagebits),
+                             sizeof(s64));
+               samples_buf_size += sizeof(s64);
+               samples_buf = devm_krealloc(&sigma_delta->spi->dev,
+                                           sigma_delta->samples_buf,
+                                           samples_buf_size, GFP_KERNEL);
+               if (!samples_buf)
+                       return -ENOMEM;
+
+               sigma_delta->samples_buf = samples_buf;
+               scan_size = BITS_TO_BYTES(scan_type->realbits + scan_type->shift);
+               /* For 24-bit data, there is an extra byte of padding. */
+               xfer[1].rx_buf = &sigma_delta->rx_buf[scan_size == 3 ? 1 : 0];
+               xfer[1].len = scan_size + (sigma_delta->status_appended ? 1 : 0);
+       }
        xfer[1].cs_change = 1;
 
        if (sigma_delta->info->has_registers) {
                                                &xfer[1], 1);
        }
 
+       sigma_delta->sample_msg.offload = sigma_delta->offload;
+
        ret = spi_optimize_message(sigma_delta->spi, &sigma_delta->sample_msg);
        if (ret)
                return ret;
        if (ret)
                goto err_unlock;
 
-       ad_sd_enable_irq(sigma_delta);
+       if (ad_sigma_delta_has_spi_offload(sigma_delta)) {
+               struct spi_offload_trigger_config config = {
+                       .type = SPI_OFFLOAD_TRIGGER_DATA_READY,
+               };
+
+               ret = spi_offload_trigger_enable(sigma_delta->offload,
+                                                sigma_delta->offload_trigger,
+                                                &config);
+               if (ret)
+                       goto err_unlock;
+       } else {
+               ad_sd_enable_irq(sigma_delta);
+       }
 
        return 0;
 
 {
        struct ad_sigma_delta *sigma_delta = iio_device_get_drvdata(indio_dev);
 
-       reinit_completion(&sigma_delta->completion);
-       wait_for_completion_timeout(&sigma_delta->completion, HZ);
+       if (ad_sigma_delta_has_spi_offload(sigma_delta)) {
+               spi_offload_trigger_disable(sigma_delta->offload,
+                                           sigma_delta->offload_trigger);
+       } else {
+               reinit_completion(&sigma_delta->completion);
+               wait_for_completion_timeout(&sigma_delta->completion, HZ);
 
-       ad_sd_disable_irq(sigma_delta);
+               ad_sd_disable_irq(sigma_delta);
+       }
 
        sigma_delta->keep_cs_asserted = false;
        ad_sigma_delta_set_mode(sigma_delta, AD_SD_MODE_IDLE);
        if ((!sigma_delta->rdy_gpiod || gpiod_get_value(sigma_delta->rdy_gpiod)) &&
            ad_sd_disable_irq(sigma_delta)) {
                complete(&sigma_delta->completion);
-               iio_trigger_poll(sigma_delta->trig);
+               if (sigma_delta->trig)
+                       iio_trigger_poll(sigma_delta->trig);
 
                return IRQ_HANDLED;
        }
        unsigned long irq_flags = irq_get_trigger_type(sigma_delta->irq_line);
        int ret;
 
-       if (dev != &sigma_delta->spi->dev) {
-               dev_err(dev, "Trigger parent should be '%s', got '%s'\n",
-                       dev_name(dev), dev_name(&sigma_delta->spi->dev));
-               return -EFAULT;
-       }
-
-       sigma_delta->trig = devm_iio_trigger_alloc(dev, "%s-dev%d", indio_dev->name,
-                                                  iio_device_id(indio_dev));
-       if (sigma_delta->trig == NULL)
-               return -ENOMEM;
-
        init_completion(&sigma_delta->completion);
 
        sigma_delta->irq_dis = true;
        if (ret)
                return ret;
 
-       iio_trigger_set_drvdata(sigma_delta->trig, sigma_delta);
+       if (ad_sigma_delta_has_spi_offload(sigma_delta)) {
+               sigma_delta->offload_trigger =
+                       devm_spi_offload_trigger_get(dev, sigma_delta->offload,
+                                                    SPI_OFFLOAD_TRIGGER_DATA_READY);
+               if (IS_ERR(sigma_delta->offload_trigger))
+                       return dev_err_probe(dev, PTR_ERR(sigma_delta->offload_trigger),
+                                            "Failed to get SPI offload trigger\n");
+       } else {
+               if (dev != &sigma_delta->spi->dev)
+                       return dev_err_probe(dev, -EFAULT,
+                               "Trigger parent should be '%s', got '%s'\n",
+                               dev_name(dev), dev_name(&sigma_delta->spi->dev));
 
-       ret = devm_iio_trigger_register(dev, sigma_delta->trig);
-       if (ret)
-               return ret;
+               sigma_delta->trig = devm_iio_trigger_alloc(dev, "%s-dev%d",
+                       indio_dev->name, iio_device_id(indio_dev));
+               if (!sigma_delta->trig)
+                       return -ENOMEM;
 
-       /* select default trigger */
-       indio_dev->trig = iio_trigger_get(sigma_delta->trig);
+               iio_trigger_set_drvdata(sigma_delta->trig, sigma_delta);
+
+               ret = devm_iio_trigger_register(dev, sigma_delta->trig);
+               if (ret)
+                       return ret;
+
+               /* select default trigger */
+               indio_dev->trig = iio_trigger_get(sigma_delta->trig);
+       }
 
        return 0;
 }
        if (!sigma_delta->slots)
                return -ENOMEM;
 
-       ret = devm_iio_triggered_buffer_setup(dev, indio_dev,
-                                             &iio_pollfunc_store_time,
-                                             &ad_sd_trigger_handler,
-                                             &ad_sd_buffer_setup_ops);
-       if (ret)
-               return ret;
+       if (ad_sigma_delta_has_spi_offload(sigma_delta)) {
+               struct dma_chan *rx_dma;
+
+               rx_dma = devm_spi_offload_rx_stream_request_dma_chan(dev,
+                       sigma_delta->offload);
+               if (IS_ERR(rx_dma))
+                       return dev_err_probe(dev, PTR_ERR(rx_dma),
+                                            "Failed to get RX DMA channel\n");
+
+               ret = devm_iio_dmaengine_buffer_setup_with_handle(dev, indio_dev,
+                       rx_dma, IIO_BUFFER_DIRECTION_IN);
+               if (ret)
+                       return dev_err_probe(dev, ret, "Cannot setup DMA buffer\n");
+
+               indio_dev->setup_ops = &ad_sd_buffer_setup_ops;
+       } else {
+               ret = devm_iio_triggered_buffer_setup(dev, indio_dev,
+                                                     &iio_pollfunc_store_time,
+                                                     &ad_sd_trigger_handler,
+                                                     &ad_sd_buffer_setup_ops);
+               if (ret)
+                       return ret;
+       }
 
        return devm_ad_sd_probe_trigger(dev, indio_dev);
 }
                        return sigma_delta->irq_line;
        }
 
+       if (info->supports_spi_offload) {
+               struct spi_offload_config offload_config = {
+                       .capability_flags = SPI_OFFLOAD_CAP_TRIGGER |
+                                           SPI_OFFLOAD_CAP_RX_STREAM_DMA,
+               };
+               int ret;
+
+               sigma_delta->offload = devm_spi_offload_get(&spi->dev, spi,
+                                                           &offload_config);
+               ret = PTR_ERR_OR_ZERO(sigma_delta->offload);
+               if (ret && ret != -ENODEV)
+                       return dev_err_probe(&spi->dev, ret, "Failed to get SPI offload\n");
+       }
+
        iio_device_set_drvdata(indio_dev, sigma_delta);
 
        return 0;
 MODULE_AUTHOR("Lars-Peter Clausen <lars@metafoo.de>");
 MODULE_DESCRIPTION("Analog Devices Sigma-Delta ADCs");
 MODULE_LICENSE("GPL v2");
+MODULE_IMPORT_NS("IIO_DMAENGINE_BUFFER");