From 78d51e0418710f21132d75242ec9130b3a597388 Mon Sep 17 00:00:00 2001
From: romerojosh <joshr@nvidia.com>
Date: Wed, 6 May 2026 22:59:53 +0000
Subject: [PATCH] Fix zero-displacement shifted rank

Signed-off-by: romerojosh <joshr@nvidia.com>
---
 src/cudecomp.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/cudecomp.cc b/src/cudecomp.cc
index 659737e..f7254a4 100644
--- a/src/cudecomp.cc
+++ b/src/cudecomp.cc
@@ -1354,6 +1354,11 @@ cudecompResult_t cudecompGetShiftedRank(cudecompHandle_t handle, cudecompGridDes
     if (dim < 0 || dim > 2) { THROW_INVALID_USAGE("dim argument out of range"); }
     if (!shifted_rank) { THROW_INVALID_USAGE("shifted_rank argument cannot be null."); }
 
+    if (displacement == 0) {
+      *shifted_rank = handle->rank;
+      return CUDECOMP_RESULT_SUCCESS;
+    }
+
     if (dim == axis) {
       *shifted_rank = periodic ? handle->rank : -1;
       return CUDECOMP_RESULT_SUCCESS;