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( cc_test(
name = "matvec_test", name = "gemma_matvec_test",
size = "small", size = "small",
timeout = "long", timeout = "long",
srcs = ["ops/matvec_test.cc"], srcs = ["ops/gemma_matvec_test.cc"],
local_defines = ["HWY_IS_TEST"], local_defines = ["HWY_IS_TEST"],
# for test_suite. # for test_suite.
tags = ["hwy_ops_test"], tags = ["hwy_ops_test"],

View File

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

View File

@ -33,7 +33,7 @@
// clang-format off // clang-format off
#undef HWY_TARGET_INCLUDE #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 // clang-format on
#include "hwy/foreach_target.h" // IWYU pragma: keep #include "hwy/foreach_target.h" // IWYU pragma: keep
#include "hwy/highway.h" #include "hwy/highway.h"