zml/platform: increase maximum device limit to support up to 32 devices per platform.
This commit is contained in:
parent
af0630616c
commit
c081cb9ad6
@ -28,7 +28,7 @@ pub const Platform = struct {
|
|||||||
pjrt_client: *pjrt.Client,
|
pjrt_client: *pjrt.Client,
|
||||||
compilation_options: CompilationOptions = .{},
|
compilation_options: CompilationOptions = .{},
|
||||||
|
|
||||||
pub const MAX_NUM_DEVICES: u8 = 8;
|
pub const MAX_NUM_DEVICES: u8 = 32;
|
||||||
|
|
||||||
pub fn init(target: Target, api: *const pjrt.Api) !Platform {
|
pub fn init(target: Target, api: *const pjrt.Api) !Platform {
|
||||||
const pjrt_client = try pjrt.Client.init(api, &.{});
|
const pjrt_client = try pjrt.Client.init(api, &.{});
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user