-
Notifications
You must be signed in to change notification settings - Fork 644
Expand file tree
/
Copy pathBUILD.bazel
More file actions
43 lines (39 loc) · 894 Bytes
/
Copy pathBUILD.bazel
File metadata and controls
43 lines (39 loc) · 894 Bytes
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
# [internal2] load py_binary
# [internal] load py_binary
load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
package(
default_applicable_licenses = [
"//:license", # Placeholder comment, do not modify
],
default_visibility = ["//visibility:public"],
)
pybind_extension(
name = "configs",
srcs = ["configs.cc"],
deps = [
"//:configs",
"//:tensor_info",
"//compression:types",
],
)
pybind_extension(
name = "gemma",
srcs = ["gemma_py.cc"],
deps = [
"//:benchmark_helper",
"//:gemma_args",
"//:gemma_lib",
"//:threading_context",
"@highway//:hwy",
],
)
py_binary(
name = "run_example",
srcs = ["run_example.py"],
deps = [
":gemma",
"@python_deps//absl_py",
# placeholder for absl/flags
"@compression_deps//numpy",
],
)