mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-12-25 13:58:46 +01:00
zig : add build.zig (#773)
Co-authored-by: Locria Cyber <74560659+locriacyber@users.noreply.github.com>
This commit is contained in:
parent
0c44427df1
commit
ed1c214e66
3
.gitignore
vendored
3
.gitignore
vendored
@ -33,3 +33,6 @@ compile_commands.json
|
|||||||
.venv
|
.venv
|
||||||
__pycache__
|
__pycache__
|
||||||
.swiftpm
|
.swiftpm
|
||||||
|
|
||||||
|
zig-out/
|
||||||
|
zig-cache/
|
||||||
|
62
build.zig
Normal file
62
build.zig
Normal file
@ -0,0 +1,62 @@
|
|||||||
|
const std = @import("std");
|
||||||
|
|
||||||
|
pub fn build(b: *std.Build) void {
|
||||||
|
const target = b.standardTargetOptions(.{});
|
||||||
|
const optimize = b.standardOptimizeOption(.{});
|
||||||
|
|
||||||
|
const lib = b.addStaticLibrary(.{
|
||||||
|
.name = "llama",
|
||||||
|
.target = target,
|
||||||
|
.optimize = optimize,
|
||||||
|
});
|
||||||
|
lib.linkLibCpp();
|
||||||
|
lib.addIncludePath(".");
|
||||||
|
lib.addIncludePath("examples");
|
||||||
|
lib.addCSourceFiles(&.{
|
||||||
|
"ggml.c",
|
||||||
|
}, &.{"-std=c11"});
|
||||||
|
lib.addCSourceFiles(&.{
|
||||||
|
"llama.cpp",
|
||||||
|
"examples/common.cpp",
|
||||||
|
}, &.{"-std=c++11"});
|
||||||
|
lib.install();
|
||||||
|
|
||||||
|
const build_args = .{ .b = b, .lib = lib, .target = target, .optimize = optimize };
|
||||||
|
const exe = build_example("main", build_args);
|
||||||
|
_ = build_example("quantize", build_args);
|
||||||
|
_ = build_example("perplexity", build_args);
|
||||||
|
_ = build_example("embedding", build_args);
|
||||||
|
|
||||||
|
// create "zig build run" command for ./main
|
||||||
|
|
||||||
|
const run_cmd = exe.run();
|
||||||
|
run_cmd.step.dependOn(b.getInstallStep());
|
||||||
|
if (b.args) |args| {
|
||||||
|
run_cmd.addArgs(args);
|
||||||
|
}
|
||||||
|
|
||||||
|
const run_step = b.step("run", "Run the app");
|
||||||
|
run_step.dependOn(&run_cmd.step);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn build_example(comptime name: []const u8, args: anytype) *std.build.LibExeObjStep {
|
||||||
|
const b = args.b;
|
||||||
|
const lib = args.lib;
|
||||||
|
const target = args.target;
|
||||||
|
const optimize = args.optimize;
|
||||||
|
|
||||||
|
const exe = b.addExecutable(.{
|
||||||
|
.name = name,
|
||||||
|
.target = target,
|
||||||
|
.optimize = optimize,
|
||||||
|
});
|
||||||
|
exe.addIncludePath(".");
|
||||||
|
exe.addIncludePath("examples");
|
||||||
|
exe.addCSourceFiles(&.{
|
||||||
|
std.fmt.comptimePrint("examples/{s}/{s}.cpp", .{name, name}),
|
||||||
|
}, &.{"-std=c++11"});
|
||||||
|
exe.linkLibrary(lib);
|
||||||
|
exe.install();
|
||||||
|
|
||||||
|
return exe;
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user