PJRT: Add handling for rank‑0 case in getDimensions to avoid null pointer usage when num_dims is zero.
This commit is contained in:
parent
639f5cd994
commit
8fa3878fc3
@ -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];
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user