1 files changed, 6 insertions, 1 deletions

diff --git a/include/uapi/linux/nvgpu.h b/include/uapi/linux/nvgpu.h
index b8ea59a1..08f8ecda 100644
--- a/include/uapi/linux/nvgpu.h
+++ b/include/uapi/linux/nvgpu.h
@@ -2214,9 +2214,14 @@ struct nvgpu_as_swap_buffer_args {
         _IOW(NVGPU_AS_IOCTL_MAGIC,  17, struct nvgpu_as_swap_buffer_args)
 #define NVGPU_AS_IOCTL_READ_SWAP_BUFFER_ASYNC_FINISH    \
         _IOW(NVGPU_AS_IOCTL_MAGIC,  18, struct nvgpu_as_swap_buffer_args)
+// Reset sector assignment to 0. Results in loss of all swapped data.
+// TODO: Don't require this - support dynamic task systems.
+#define NVGPU_AS_IOCTL_SWAP_RESET \
+        _IO(NVGPU_AS_IOCTL_MAGIC,  19)
+
 
 #define NVGPU_AS_IOCTL_LAST            \
-        _IOC_NR(NVGPU_AS_IOCTL_READ_SWAP_BUFFER_ASYNC_FINISH)
+        _IOC_NR(NVGPU_AS_IOCTL_SWAP_RESET)
 #define NVGPU_AS_IOCTL_MAX_ARG_SIZE     \
         sizeof(struct nvgpu_as_map_buffer_ex_args)