From c081cb9ad648a616e6418dfad7ada48b96d858b1 Mon Sep 17 00:00:00 2001 From: Tarry Singh Date: Thu, 24 Aug 2023 12:23:07 +0000 Subject: [PATCH] zml/platform: increase maximum device limit to support up to 32 devices per platform. --- zml/platform.zig | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/zml/platform.zig b/zml/platform.zig index 85b253c..c594ba6 100644 --- a/zml/platform.zig +++ b/zml/platform.zig @@ -28,7 +28,7 @@ pub const Platform = struct { pjrt_client: *pjrt.Client, 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 { const pjrt_client = try pjrt.Client.init(api, &.{});