Fix build issues when tests are enabled

This commit is contained in:
RangerUFO 2024-08-10 19:55:10 +08:00 committed by Jan Wassenberg
parent 730b6bfc94
commit ea72575e56
3 changed files with 4 additions and 4 deletions

View File

@ -58,10 +58,10 @@ cc_test(
)
cc_test(
name = "matvec_test",
name = "gemma_matvec_test",
size = "small",
timeout = "long",
srcs = ["ops/matvec_test.cc"],
srcs = ["ops/gemma_matvec_test.cc"],
local_defines = ["HWY_IS_TEST"],
# for test_suite.
tags = ["hwy_ops_test"],

View File

@ -154,7 +154,7 @@ set(GEMMA_TEST_FILES
backprop/optimize_test.cc
ops/ops_test.cc
ops/matmul_test.cc
ops/matvec_test.cc
ops/gemma_matvec_test.cc
evals/gemma_test.cc
)

View File

@ -33,7 +33,7 @@
// clang-format off
#undef HWY_TARGET_INCLUDE
#define HWY_TARGET_INCLUDE "ops/matvec_test.cc" // NOLINT
#define HWY_TARGET_INCLUDE "ops/gemma_matvec_test.cc" // NOLINT
// clang-format on
#include "hwy/foreach_target.h" // IWYU pragma: keep
#include "hwy/highway.h"