PJRT: Add handling for rank‑0 case in getDimensions to avoid null pointer usage when num_dims is zero.

This commit is contained in:
Tarry Singh 2023-02-17 10:47:15 +00:00
parent 639f5cd994
commit 8fa3878fc3

View File

@ -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];
}