From 8fa3878fc391c0c695f287f4e3f875c79dee9ef7 Mon Sep 17 00:00:00 2001
From: Tarry Singh <tarry.singh@deepkapha.com>
Date: Fri, 17 Feb 2023 10:47:15 +0000
Subject: [PATCH] =?UTF-8?q?PJRT:=20Add=20handling=20for=20rank=E2=80=910?=
 =?UTF-8?q?=20case=20in=20`getDimensions`=20to=20avoid=20null=20pointer=20?=
 =?UTF-8?q?usage=20when=20`num=5Fdims`=20is=20zero.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pjrt/pjrt.zig | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pjrt/pjrt.zig b/pjrt/pjrt.zig
index 642d97a..e80fdf8 100644
--- a/pjrt/pjrt.zig
+++ b/pjrt/pjrt.zig
@@ -668,6 +668,9 @@ pub const Buffer = opaque {
         const ret = api.call(.PJRT_Buffer_Dimensions, .{
             .buffer = self.inner(),
         }) catch unreachable;
+        if (ret.num_dims == 0) {
+            return &.{};
+        }
         return ret.dims[0..ret.num_dims];
     }