2025-06-20 13:23:06 +00:00
|
|
|
load("@bazel_skylib//rules:build_test.bzl", "build_test")
|
2025-07-04 15:10:46 +00:00
|
|
|
load("@rules_zig//zig:defs.bzl", "zig_binary")
|
2023-01-03 10:21:07 +00:00
|
|
|
|
2025-07-04 15:10:46 +00:00
|
|
|
zig_binary(
|
2023-01-03 10:21:07 +00:00
|
|
|
name = "mnist",
|
|
|
|
|
args = [
|
2025-07-23 12:53:46 +00:00
|
|
|
"$(location @mnist//:mnist.safetensors)",
|
2025-06-20 13:23:06 +00:00
|
|
|
"$(location @mnist//:t10k-images.idx3-ubyte)",
|
2023-01-03 10:21:07 +00:00
|
|
|
],
|
|
|
|
|
data = [
|
2025-07-23 12:53:46 +00:00
|
|
|
"@mnist//:mnist.safetensors",
|
2025-06-20 13:23:06 +00:00
|
|
|
"@mnist//:t10k-images.idx3-ubyte",
|
2023-01-03 10:21:07 +00:00
|
|
|
],
|
|
|
|
|
main = "mnist.zig",
|
|
|
|
|
deps = [
|
2025-06-20 13:23:06 +00:00
|
|
|
"//async",
|
|
|
|
|
"//zml",
|
2023-01-03 10:21:07 +00:00
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
|
2025-06-20 13:23:06 +00:00
|
|
|
build_test(
|
|
|
|
|
name = "test",
|
|
|
|
|
targets = [":mnist"],
|
2023-01-03 10:21:07 +00:00
|
|
|
)
|