build.zig : install config header (#2216)

This commit is contained in:
Ali Chraghi 2023-07-14 11:50:58 -07:00 committed by GitHub
parent c48c525f87
commit de8342423d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,9 +1,19 @@
const std = @import("std"); const std = @import("std");
const commit_hash = @embedFile(".git/refs/heads/master");
// Zig Version: 0.11.0-dev.3379+629f0d23b // Zig Version: 0.11.0-dev.3986+e05c242cd
pub fn build(b: *std.build.Builder) void { pub fn build(b: *std.build.Builder) void {
const target = b.standardTargetOptions(.{}); const target = b.standardTargetOptions(.{});
const optimize = b.standardOptimizeOption(.{}); const optimize = b.standardOptimizeOption(.{});
const config_header = b.addConfigHeader(
.{ .style = .blank, .include_path = "build-info.h" },
.{
.BUILD_NUMBER = 0,
.BUILD_COMMIT = commit_hash[0 .. commit_hash.len - 1], // omit newline
},
);
const lib = b.addStaticLibrary(.{ const lib = b.addStaticLibrary(.{
.name = "llama", .name = "llama",
.target = target, .target = target,
@ -13,24 +23,21 @@ pub fn build(b: *std.build.Builder) void {
lib.linkLibCpp(); lib.linkLibCpp();
lib.addIncludePath("."); lib.addIncludePath(".");
lib.addIncludePath("./examples"); lib.addIncludePath("./examples");
lib.addCSourceFiles(&.{ lib.addConfigHeader(config_header);
"ggml.c", lib.addCSourceFiles(&.{"ggml.c"}, &.{"-std=c11"});
}, &.{"-std=c11"}); lib.addCSourceFiles(&.{"llama.cpp"}, &.{"-std=c++11"});
lib.addCSourceFiles(&.{
"llama.cpp",
}, &.{"-std=c++11"});
b.installArtifact(lib); b.installArtifact(lib);
const examples = .{ const examples = .{
"main", "main",
"baby-llama", "baby-llama",
"embedding", "embedding",
// "metal", "metal",
"perplexity", "perplexity",
"quantize", "quantize",
"quantize-stats", "quantize-stats",
"save-load-state", "save-load-state",
// "server", "server",
"simple", "simple",
"train-text-from-scratch", "train-text-from-scratch",
}; };
@ -43,16 +50,19 @@ pub fn build(b: *std.build.Builder) void {
}); });
exe.addIncludePath("."); exe.addIncludePath(".");
exe.addIncludePath("./examples"); exe.addIncludePath("./examples");
exe.addConfigHeader(config_header);
exe.addCSourceFiles(&.{ exe.addCSourceFiles(&.{
std.fmt.comptimePrint("examples/{s}/{s}.cpp", .{example_name, example_name}), std.fmt.comptimePrint("examples/{s}/{s}.cpp", .{ example_name, example_name }),
"examples/common.cpp", "examples/common.cpp",
}, &.{"-std=c++11"}); }, &.{"-std=c++11"});
exe.linkLibrary(lib); exe.linkLibrary(lib);
b.installArtifact(exe); b.installArtifact(exe);
const run_cmd = b.addRunArtifact(exe); const run_cmd = b.addRunArtifact(exe);
run_cmd.step.dependOn(b.getInstallStep()); run_cmd.step.dependOn(b.getInstallStep());
if (b.args) |args| run_cmd.addArgs(args); if (b.args) |args| run_cmd.addArgs(args);
const run_step = b.step("run_" ++ example_name, "Run the app");
const run_step = b.step("run-" ++ example_name, "Run the app");
run_step.dependOn(&run_cmd.step); run_step.dependOn(&run_cmd.step);
} }
} }