mirror of https://github.com/google/gemma.cpp.git
44 lines
895 B
Python
44 lines
895 B
Python
# [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:shared",
|
|
],
|
|
)
|
|
|
|
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",
|
|
],
|
|
)
|