zig-torch/build.zig at main · kitajusSus/zig-torch · GitHub

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
62
63
64
65
66
67
68
69
const std = @import("std");

pub fn build(b: *std.Build) void {
    const target = b.standardTargetOptions(.{});
    const optimize = b.standardOptimizeOption(.{});

    const lib = b.addLibrary(.{
        .name = "zigtorch",
        .linkage = .dynamic,
        .version = .{ .major = 0, .minor = 1, .patch = 1 },
        .root_module = b.createModule(.{
            .root_source_file = b.path("src/c_api/exports.zig"),
            .target = target,
            .optimize = optimize,
        }),
    });

    lib.root_module.addImport("mm", b.createModule(.{
        .root_source_file = b.path("src/ops/mm.zig"),
    }));
    lib.root_module.addImport("add", b.createModule(.{
        .root_source_file = b.path("src/ops/add.zig"),
    }));

    b.installArtifact(lib);

    const run_tests = b.addTest(.{
        .root_module = b.createModule(.{
            .root_source_file = b.path("src/ops/mm.zig"),
            .target = target,
            .optimize = optimize,
        }),
    });
    const test_step = b.step("test", "Run Zig tests");
    const run_main_tests = b.addRunArtifact(run_tests);
    test_step.dependOn(&run_main_tests.step);

    const core_tests = b.addTest(.{
        .root_module = b.createModule(.{
            .root_source_file = b.path("core/mm_test.zig"),
            .target = target,
            .optimize = optimize,
        }),
    });
    core_tests.root_module.addImport("mm", b.createModule(.{
        .root_source_file = b.path("src/ops/mm.zig"),
    }));

    const core_test_step = b.step("test-core", "Run core tests");
    const run_core_tests = b.addRunArtifact(core_tests);
    core_test_step.dependOn(&run_core_tests.step);
    const bench_exe = b.addExecutable(.{
        .name = "bench",
        .root_module = b.createModule(.{
            .root_source_file = b.path("benchmarks/mm_bench.zig"),
            .target = target,
            .optimize = optimize,
        }),
    });
    bench_exe.root_module.addImport("mm", b.createModule(.{
        .root_source_file = b.path("src/ops/mm.zig"),
    }));

    b.installArtifact(bench_exe);

    const bench_step = b.step("bench", "Run benchmark");
    const run_bench = b.addRunArtifact(bench_exe);
    bench_step.dependOn(&run_bench.step);
}