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

Update examples to use the corrected logFn API.

Browse Source
This commit is contained in:
Foke Singh 2024-03-18 13:11:14 +00:00
parent 754656f2f0
commit 602757e7a9
3 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/benchmark/main.zig
View File

@ -6,6 +6,7 @@ const flags = @import("tigerbeetle/flags");
// set log level to debug to print the generated IR // set log level to debug to print the generated IR
pub const std_options = .{ pub const std_options = .{
.log_level = .warn, .log_level = .warn,
.logFn = asynk.logFn(std.log.defaultLog),
}; };
pub fn benchmark(a: zml.Tensor, b: zml.Tensor) zml.Tensor { pub fn benchmark(a: zml.Tensor, b: zml.Tensor) zml.Tensor {

1
examples/llama/main.zig
View File

@ -19,6 +19,7 @@ const log = std.log.scoped(.llama);
pub const std_options = .{ pub const std_options = .{
.log_level = .info, .log_level = .info,
.logFn = asynk.logFn(std.log.defaultLog),
}; };
pub fn tokenizePromptLlama3(allocator: std.mem.Allocator, tokenizer: zml.tokenizer.Tokenizer, config: LlamaLM.Config, prompt: []const u8) ![]u32 { pub fn tokenizePromptLlama3(allocator: std.mem.Allocator, tokenizer: zml.tokenizer.Tokenizer, config: LlamaLM.Config, prompt: []const u8) ![]u32 {

7
examples/mnist/mnist.zig
View File

@ -1,14 +1,13 @@
const asynk = @import("async");
const std = @import("std"); const std = @import("std");
const zml = @import("zml");
const show_mlir = true; const asynk = @import("async");
const zml = @import("zml");
const log = std.log.scoped(.mnist); const log = std.log.scoped(.mnist);
pub const std_options: std.Options = .{ pub const std_options: std.Options = .{
.logFn = asynk.logFn,
.log_level = .info, .log_level = .info,
.logFn = asynk.logFn(std.log.defaultLog),
}; };
/// Model definition /// Model definition