zml/platform: increase maximum device limit to support up to 32 devices per platform.

This commit is contained in:
Tarry Singh 2023-08-24 12:23:07 +00:00
parent af0630616c
commit c081cb9ad6

View File

@ -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, &.{});