• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Description:
2#    Tests for LLVM-based CPU backend for XLA.
3
4load("//tensorflow:tensorflow.bzl", "filegroup")
5load("//tensorflow:tensorflow.bzl", "tf_cc_test")
6
7package(
8    default_visibility = [":friends"],
9    licenses = ["notice"],
10)
11
12package_group(
13    name = "friends",
14    includes = [
15        "//tensorflow/compiler/xla:friends",
16    ],
17)
18
19# Filegroup used to collect source files for dependency checking.
20filegroup(
21    name = "c_srcs",
22    data = glob([
23        "**/*.cc",
24        "**/*.h",
25    ]),
26)
27
28cc_library(
29    name = "cpu_codegen_test",
30    testonly = True,
31    hdrs = ["cpu_codegen_test.h"],
32    deps = [
33        "//tensorflow/compiler/xla/service:cpu_plugin",
34        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
35        "//tensorflow/core:test_main",
36    ],
37)
38
39tf_cc_test(
40    name = "cpu_dyn_shape_test",
41    srcs = ["cpu_dyn_shape_test.cc"],
42    deps = [
43        "//tensorflow/compiler/xla/service:hlo",
44        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
45        "//tensorflow/compiler/xla/service/cpu:test_header_helper",
46        "//tensorflow/compiler/xla/service/cpu/tests:cpu_codegen_test",
47        "//tensorflow/core:lib",
48        "//tensorflow/core:test",
49        "//tensorflow/core:test_main",
50    ],
51)
52
53tf_cc_test(
54    name = "cpu_fusion_test",
55    srcs = ["cpu_fusion_test.cc"],
56    deps = [
57        "//tensorflow/compiler/xla:literal",
58        "//tensorflow/compiler/xla:shape_util",
59        "//tensorflow/compiler/xla:util",
60        "//tensorflow/compiler/xla:xla_data_proto_cc",
61        "//tensorflow/compiler/xla/service:cpu_plugin",
62        "//tensorflow/compiler/xla/service:hlo",
63        "//tensorflow/compiler/xla/service/cpu:cpu_instruction_fusion",
64        "//tensorflow/compiler/xla/tests:hlo_test_base",
65        "//tensorflow/compiler/xla/tests:literal_test_util",
66        "//tensorflow/core:test",
67        "//tensorflow/core:test_main",
68        "@com_google_absl//absl/memory",
69    ],
70)
71
72tf_cc_test(
73    name = "cpu_bytesizeof_test",
74    srcs = ["cpu_bytesizeof_test.cc"],
75    deps = [
76        "//tensorflow/compiler/xla:shape_util",
77        "//tensorflow/compiler/xla/service/llvm_ir:llvm_util",
78        "//tensorflow/core:test",
79        "//tensorflow/core:test_main",
80    ],
81)
82
83tf_cc_test(
84    name = "cpu_external_constants_test",
85    srcs = ["cpu_external_constants_test.cc"],
86    deps = [
87        "//tensorflow/compiler/xla:array2d",
88        "//tensorflow/compiler/xla:shape_util",
89        "//tensorflow/compiler/xla/service:hlo",
90        "//tensorflow/compiler/xla/service/cpu/tests:cpu_codegen_test",
91        "//tensorflow/compiler/xla/tests:filecheck",
92        "//tensorflow/core:test",
93    ],
94)
95
96tf_cc_test(
97    name = "cpu_noalias_test",
98    srcs = ["cpu_noalias_test.cc"],
99    deps = [
100        "//tensorflow/compiler/xla:literal",
101        "//tensorflow/compiler/xla:shape_util",
102        "//tensorflow/compiler/xla:util",
103        "//tensorflow/compiler/xla:xla_data_proto_cc",
104        "//tensorflow/compiler/xla/service:buffer_assignment",
105        "//tensorflow/compiler/xla/service:hlo",
106        "//tensorflow/compiler/xla/service/cpu/tests:cpu_codegen_test",
107        "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis",
108        "//tensorflow/compiler/xla/service/llvm_ir:llvm_util",
109        "//tensorflow/compiler/xla/tests:filecheck",
110        "//tensorflow/core:test",
111        "//tensorflow/core:test_main",
112        "@com_google_absl//absl/memory",
113        "@llvm-project//llvm:Core",
114    ],
115)
116
117tf_cc_test(
118    name = "cpu_intrinsic_test",
119    srcs = ["cpu_intrinsic_test.cc"],
120    deps = [
121        ":cpu_codegen_test",
122        "//tensorflow/compiler/xla/service:hlo",
123        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
124        "//tensorflow/core:lib",
125        "//tensorflow/core:test",
126        "//tensorflow/core:test_main",
127        "@com_google_absl//absl/strings",
128        "@llvm-project//llvm:ARMCodeGen",  # fixdeps: keep
129        "@llvm-project//llvm:Target",
130        "@llvm-project//llvm:X86CodeGen",  # fixdeps: keep
131    ],
132)
133
134tf_cc_test(
135    name = "cpu_eigen_dot_operation_test",
136    srcs = ["cpu_eigen_dot_operation_test.cc"],
137    deps = [
138        "//tensorflow/compiler/xla/service:hlo",
139        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
140        "//tensorflow/compiler/xla/service/cpu:test_header_helper",
141        "//tensorflow/compiler/xla/service/cpu/tests:cpu_codegen_test",
142        "//tensorflow/compiler/xla/tests:test_utils",
143        "//tensorflow/core:lib",
144        "//tensorflow/core:test",
145        "//tensorflow/core:test_main",
146        "@com_google_absl//absl/strings",
147    ],
148)
149
150tf_cc_test(
151    name = "cpu_profiling_test",
152    srcs = ["cpu_profiling_test.cc"],
153    deps = [
154        ":cpu_codegen_test",
155        "//tensorflow/compiler/xla/service:hlo",
156        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
157        "//tensorflow/core:lib",
158        "//tensorflow/core:test",
159        "//tensorflow/core:test_main",
160        "@com_google_absl//absl/strings",
161        "@llvm-project//llvm:ARMCodeGen",  # fixdeps: keep
162        "@llvm-project//llvm:Target",
163        "@llvm-project//llvm:X86CodeGen",  # fixdeps: keep
164    ],
165)
166
167tf_cc_test(
168    name = "tree_reduction_rewriter_test",
169    srcs = ["tree_reduction_rewriter_test.cc"],
170    deps = [
171        ":cpu_codegen_test",
172        "//tensorflow/compiler/xla:statusor",
173        "//tensorflow/compiler/xla:xla_proto_cc",
174        "//tensorflow/compiler/xla/service:hlo",
175        "//tensorflow/compiler/xla/service:hlo_module_config",
176        "//tensorflow/compiler/xla/service:hlo_parser",
177        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
178        "//tensorflow/compiler/xla/tests:codegen_test_base",
179        "//tensorflow/compiler/xla/tests:filecheck",
180        "//tensorflow/compiler/xla/tests:hlo_test_base",
181        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
182        "//tensorflow/compiler/xla/tests:test_utils",
183        "//tensorflow/core:lib",
184        "//tensorflow/core:test",
185        "//tensorflow/core:test_main",
186        "//tensorflow/stream_executor/lib",
187        "@com_google_absl//absl/strings",
188    ],
189)
190
191tf_cc_test(
192    name = "cpu_infeed_test",
193    srcs = ["cpu_infeed_test.cc"],
194    deps = [
195        "//tensorflow/compiler/xla:literal",
196        "//tensorflow/compiler/xla:shape_util",
197        "//tensorflow/compiler/xla:statusor",
198        "//tensorflow/compiler/xla:test_helpers",
199        "//tensorflow/compiler/xla:xla_data_proto_cc",
200        "//tensorflow/compiler/xla/client:global_data",
201        "//tensorflow/compiler/xla/client:local_client",
202        "//tensorflow/compiler/xla/client:xla_builder",
203        "//tensorflow/compiler/xla/client:xla_computation",
204        "//tensorflow/compiler/xla/client/lib:arithmetic",
205        "//tensorflow/compiler/xla/service:cpu_plugin",
206        "//tensorflow/compiler/xla/tests:client_library_test_base",
207        "//tensorflow/compiler/xla/tests:literal_test_util",
208        "//tensorflow/core:lib",
209        "//tensorflow/core:test",
210        "//tensorflow/core:test_main",
211    ],
212)
213
214tf_cc_test(
215    name = "cpu_literal_caching_test",
216    srcs = ["cpu_literal_caching_test.cc"],
217    deps = [
218        "//tensorflow/compiler/xla/service:hlo",
219        "//tensorflow/compiler/xla/service:hlo_parser",
220        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
221        "//tensorflow/compiler/xla/service/cpu:test_header_helper",
222        "//tensorflow/compiler/xla/service/cpu/tests:cpu_codegen_test",
223        "//tensorflow/core:lib",
224        "//tensorflow/core:test",
225        "//tensorflow/core:test_main",
226    ],
227)
228
229tf_cc_test(
230    name = "cpu_outfeed_test",
231    srcs = ["cpu_outfeed_test.cc"],
232    deps = [
233        "//tensorflow/compiler/xla/service:hlo",
234        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
235        "//tensorflow/compiler/xla/service/cpu:test_header_helper",
236        "//tensorflow/compiler/xla/service/cpu/tests:cpu_codegen_test",
237        "//tensorflow/core:lib",
238        "//tensorflow/core:test",
239        "//tensorflow/core:test_main",
240    ],
241)
242
243tf_cc_test(
244    name = "cpu_key_value_sort_test",
245    srcs = ["cpu_key_value_sort_test.cc"],
246    deps = [
247        "//tensorflow/compiler/xla/service:hlo",
248        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
249        "//tensorflow/compiler/xla/service/cpu:test_header_helper",
250        "//tensorflow/compiler/xla/service/cpu/tests:cpu_codegen_test",
251        "//tensorflow/core:lib",
252        "//tensorflow/core:test",
253        "//tensorflow/core:test_main",
254    ],
255)
256
257tf_cc_test(
258    name = "cpu_spmd_compile_test",
259    srcs = ["cpu_spmd_compile_test.cc"],
260    deps = [
261        ":cpu_codegen_test",
262        "//tensorflow/compiler/xla/service:hlo_module_config",
263        "//tensorflow/compiler/xla/service:hlo_parser",
264        "//tensorflow/compiler/xla/service:hlo_query",
265        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
266        "//tensorflow/compiler/xla/service/cpu:test_header_helper",
267        "//tensorflow/compiler/xla/tests:hlo_test_base",
268        "//tensorflow/core:test",
269        "//tensorflow/core:test_main",
270    ],
271)
272
273tf_cc_test(
274    name = "cpu_topk_test",
275    srcs = ["cpu_topk_test.cc"],
276    deps = [
277        ":cpu_codegen_test",
278        "//tensorflow/compiler/xla/client:xla_builder",
279        "//tensorflow/compiler/xla/client/lib:sorting",
280        "//tensorflow/compiler/xla/service:hlo",
281        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
282        "//tensorflow/compiler/xla/service/cpu:test_header_helper",
283        "//tensorflow/core:lib",
284        "//tensorflow/core:test",
285        "//tensorflow/core:test_main",
286    ],
287)
288
289tf_cc_test(
290    name = "cpu_vectorization_test",
291    srcs = ["cpu_vectorization_test.cc"],
292    deps = [
293        ":cpu_codegen_test",
294        "//tensorflow/compiler/xla/service:hlo",
295        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
296        "//tensorflow/core:lib",
297        "//tensorflow/core:test",
298        "//tensorflow/core:test_main",
299        "@com_google_absl//absl/strings",
300        "@llvm-project//llvm:ARMCodeGen",  # fixdeps: keep
301        "@llvm-project//llvm:Target",
302        "@llvm-project//llvm:X86CodeGen",  # fixdeps: keep
303    ],
304)
305
306tf_cc_test(
307    name = "cpu_while_test",
308    srcs = ["cpu_while_test.cc"],
309    deps = [
310        ":cpu_codegen_test",
311        "//tensorflow/compiler/xla/service:hlo",
312        "//tensorflow/compiler/xla/service/cpu:cpu_compiler",
313        "//tensorflow/core:lib",
314        "//tensorflow/core:test",
315        "//tensorflow/core:test_main",
316        "@com_google_absl//absl/strings",
317        "@llvm-project//llvm:ARMCodeGen",  # fixdeps: keep
318        "@llvm-project//llvm:Target",
319        "@llvm-project//llvm:X86CodeGen",  # fixdeps: keep
320    ],
321)
322