tarry.singh/Radix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Tarry Singh 2023-02-17 10:47:15 +00:00
parent 639f5cd994
commit 8fa3878fc3
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pjrt/pjrt.zig
View File

@ -668,6 +668,9 @@ pub const Buffer = opaque {
const ret = api.call(.PJRT_Buffer_Dimensions, .{ const ret = api.call(.PJRT_Buffer_Dimensions, .{
.buffer = self.inner(), .buffer = self.inner(),
}) catch unreachable; }) catch unreachable;
if (ret.num_dims == 0) {
return &.{};
}
return ret.dims[0..ret.num_dims]; return ret.dims[0..ret.num_dims];
} }