mirror of https://github.com/google/gemma.cpp.git
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
# [internal] load strict.bzl
|
|
load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
|
|
|
|
package(
|
|
default_applicable_licenses = [
|
|
"//:license", # Placeholder comment, do not modify
|
|
],
|
|
default_visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "compression_clif_aux",
|
|
srcs = ["compression_clif_aux.cc"],
|
|
hdrs = ["compression_clif_aux.h"],
|
|
visibility = ["//visibility:private"],
|
|
deps = [
|
|
"@abseil-cpp//absl/types:span",
|
|
"//:common",
|
|
"//:tokenizer",
|
|
"//compression:compress",
|
|
"//compression:io",
|
|
"@highway//:hwy",
|
|
"@highway//:thread_pool",
|
|
],
|
|
)
|
|
|
|
pybind_extension(
|
|
name = "compression",
|
|
srcs = ["compression_extension.cc"],
|
|
deps = [
|
|
":compression_clif_aux",
|
|
"@abseil-cpp//absl/types:span",
|
|
"//:common",
|
|
"//compression:sfp",
|
|
],
|
|
)
|
|
|
|
# py_strict
|
|
py_test(
|
|
name = "compression_test",
|
|
srcs = ["compression_test.py"],
|
|
deps = [
|
|
":compression",
|
|
"@com_google_absl_py//absl/testing:absltest",
|
|
"//python:configs",
|
|
"@compression_deps//numpy",
|
|
],
|
|
)
|