gemma.cpp/paligemma/BUILD.bazel

50 lines
985 B
Python

package(
default_applicable_licenses = [
"//:license", # Placeholder comment, do not modify
],
default_visibility = [
"//:__subpackages__",
],
)
cc_library(
name = "image",
srcs = ["image.cc"],
hdrs = ["image.h"],
deps = [
"//compression:io",
"@highway//:hwy",
"@highway//:profiler",
],
)
cc_test(
name = "image_test",
srcs = ["image_test.cc"],
data = ["testdata/image.ppm"],
deps = [
":image",
"@googletest//:gtest_main", # buildcleaner: keep
],
)
cc_test(
name = "paligemma_test",
srcs = ["paligemma_test.cc"],
# Requires model files
tags = [
"local",
"manual",
"no_tap",
],
deps = [
"@googletest//:gtest_main", # buildcleaner: keep
"//:benchmark_helper",
"//:common",
"//:gemma_lib",
"//compression:sfp",
"@highway//:hwy",
"@highway//:hwy_test_util",
],
)