FilesExpand file tree

 master

/

build.zig

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

61 lines (51 loc) · 1.77 KB

 master

/

build.zig

Copy path

Top

File metadata and controls

• Code
• Blame

61 lines (51 loc) · 1.77 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

const std = @import("std");

pub fn build(b: *std.build.Builder) void {

const target = b.standardTargetOptions(.{});

const optimize = b.standardReleaseOptions();

const want_lto = b.option(bool, "lto", "Want -fLTO");

const lib = b.addStaticLibrary("llama", null);

lib.want_lto = want_lto;

lib.setTarget(target);

lib.setBuildMode(optimize);

lib.linkLibCpp();

lib.addIncludePath(".");

lib.addIncludePath("examples");

lib.addCSourceFiles(&.{

"ggml.c",

}, &.{"-std=c11"});

lib.addCSourceFiles(&.{

"llama.cpp",

}, &.{"-std=c++11"});

lib.install();

const build_args = .{ .b = b, .lib = lib, .target = target, .optimize = optimize, .want_lto = want_lto };

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 want_lto = args.want_lto;

const exe = b.addExecutable(name, null);

exe.want_lto = want_lto;

lib.setTarget(args.target);

lib.setBuildMode(args.optimize);

exe.addIncludePath(".");

exe.addIncludePath("examples");

exe.addCSourceFiles(&.{

std.fmt.comptimePrint("examples/{s}/{s}.cpp", .{name, name}),

"examples/common.cpp",

}, &.{"-std=c++11"});

exe.linkLibrary(lib);

exe.install();

return exe;

}