26 lines
532 B
Python
26 lines
532 B
Python
load("@bazel_skylib//rules:build_test.bzl", "build_test")
|
|
load("@rules_zig//zig:defs.bzl", "zig_binary")
|
|
|
|
zig_binary(
|
|
name = "mnist",
|
|
args = [
|
|
"$(location @mnist//:mnist.safetensors)",
|
|
"$(location @mnist//:t10k-images.idx3-ubyte)",
|
|
],
|
|
data = [
|
|
"@mnist//:mnist.safetensors",
|
|
"@mnist//:t10k-images.idx3-ubyte",
|
|
],
|
|
zigopts = ["-fllvm"],
|
|
main = "mnist.zig",
|
|
deps = [
|
|
"//async",
|
|
"//zml",
|
|
],
|
|
)
|
|
|
|
build_test(
|
|
name = "test",
|
|
targets = [":mnist"],
|
|
)
|