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

module-cache: raise max_pjrt_executable_size limit to 400 MB to accommodate large PJRT executables.

Browse Source
This commit is contained in:
Tarry Singh 2023-07-14 17:58:22 +00:00
parent 88c7a74ccf
commit 0f9a92f27d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
zml/module.zig
View File

@ -1089,7 +1089,7 @@ fn computeModuleHash(platform: Platform, module: mlir.Module) u64 {
return hasher.final(); return hasher.final();
} }
const max_pjrt_executable_size = 20 * 1024 * 1024; const max_pjrt_executable_size = 400 * 1024 * 1024;
fn loadPjrtExecutable(arena: std.mem.Allocator, platform: Platform, module_hash: u64, compilation_cache_location: []const u8) !*pjrt.LoadedExecutable { fn loadPjrtExecutable(arena: std.mem.Allocator, platform: Platform, module_hash: u64, compilation_cache_location: []const u8) !*pjrt.LoadedExecutable {
const resolved_path = try std.fs.cwd().realpathAlloc(arena, compilation_cache_location); const resolved_path = try std.fs.cwd().realpathAlloc(arena, compilation_cache_location);