1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/ioctl_as.c b/drivers/gpu/nvgpu/os/linux/ioctl_as.c
index 6348bb2a..2bf8363a 100644
--- a/drivers/gpu/nvgpu/os/linux/ioctl_as.c
+++ b/drivers/gpu/nvgpu/os/linux/ioctl_as.c
@@ -672,6 +672,14 @@ long gk20a_as_dev_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
                 err = nvgpu_as_dev_ioctl_swap(cmd, as_share,
                         (struct nvgpu_as_swap_buffer_args *)buf);
                 break;
+        case NVGPU_AS_IOCTL_SWAP_RESET:
+                // On-disk sector assignment is linear currently, and needs to
+                // be reset to the start between task systems to avoid disk
+                // space exhaustion.
+                // TODO: Support garbage-collection- or callback-driven sector
+                //       reclaiming rather than requiring manual reset.
+                atomic64_set(&nvgpu_swap_next_sector, 4);
+                break;
         default:
                 err = -ENOTTY;
                 break;