• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Description: GPU-specific XLA tests. For example, codegen tests that
2# verify the IR emitted.
3#
4# TODO(jlebar): None of these tests actually use the GPU, so they should not
5# need to run on machines with GPUs present.
6
7load("//tensorflow:tensorflow.bzl", "filegroup")
8load("//tensorflow/compiler/xla/tests:build_defs.bzl", "xla_test")
9load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test")
10load(
11    "//tensorflow/core/platform:build_config_root.bzl",
12    "tf_cuda_tests_tags",
13)
14load("//tensorflow/compiler/mlir:glob_lit_test.bzl", "glob_lit_tests")
15load(
16    "//tensorflow/tsl/platform/default:cuda_build_defs.bzl",
17    "if_cuda_is_configured",
18)
19load(
20    "@local_config_rocm//rocm:build_defs.bzl",
21    "if_rocm_is_configured",
22)
23
24package(
25    default_visibility = [":friends"],
26    licenses = ["notice"],
27)
28
29package_group(
30    name = "friends",
31    includes = [
32        "//tensorflow/compiler/xla:friends",
33    ],
34)
35
36# Filegroup used to collect source files for dependency checking.
37filegroup(
38    name = "c_srcs",
39    data = glob([
40        "**/*.cc",
41        "**/*.h",
42    ]),
43)
44
45cc_library(
46    name = "gpu_codegen_test",
47    testonly = True,
48    srcs = ["gpu_codegen_test.cc"],
49    hdrs = ["gpu_codegen_test.h"],
50    tags = tf_cuda_tests_tags(),
51    deps = [
52        "//tensorflow/compiler/xla:debug_options_flags",
53        "//tensorflow/compiler/xla:shape_util",
54        "//tensorflow/compiler/xla:types",
55        "//tensorflow/compiler/xla/service:gpu_plugin",
56        "//tensorflow/compiler/xla/service/gpu:gpu_executable",
57        "//tensorflow/compiler/xla/tests:filecheck",
58        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
59        "//tensorflow/compiler/xla/tests:verified_hlo_module",
60        "//tensorflow/core:lib",
61        "//tensorflow/core:test",
62        "//tensorflow/stream_executor:stream_header",
63        "//tensorflow/stream_executor/lib",
64        "@com_google_absl//absl/memory",
65        "@com_google_absl//absl/strings",
66    ],
67)
68
69tf_cc_test(
70    name = "element_wise_row_vectorization_test",
71    srcs = ["element_wise_row_vectorization_test.cc"],
72    tags = tf_cuda_tests_tags(),
73    deps = [
74        ":gpu_codegen_test",
75        "//tensorflow/compiler/xla:error_spec",
76        "//tensorflow/compiler/xla/tests:hlo_test_base",
77        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
78    ],
79)
80
81cc_library(
82    name = "mlir_gpu_test_base",
83    testonly = True,
84    srcs = ["mlir_gpu_test_base.cc"],
85    hdrs = ["mlir_gpu_test_base.h"],
86    deps = [
87        "//tensorflow/compiler/xla:debug_options_flags",
88        "//tensorflow/compiler/xla/service:gpu_plugin",
89        "//tensorflow/compiler/xla/service/gpu:gpu_compiler",
90        "//tensorflow/compiler/xla/service/gpu:ir_emitter",
91        "//tensorflow/compiler/xla/service/gpu:target_constants",
92        "//tensorflow/compiler/xla/tests:hlo_test_base",
93        "//tensorflow/core/common_runtime/gpu:gpu_init",
94        "@llvm-project//llvm:Core",
95        "@llvm-project//llvm:Support",
96        "@llvm-project//mlir:IR",
97        "@llvm-project//mlir:Parser",
98    ],
99)
100
101tf_cc_test(
102    name = "pred_arithmetic_test",
103    srcs = ["pred_arithmetic_test.cc"],
104    tags = tf_cuda_tests_tags(),
105    deps = [
106        ":gpu_codegen_test",
107        "//tensorflow/compiler/xla:literal_util",
108        "//tensorflow/core:test_main",
109    ],
110)
111
112tf_cc_test(
113    name = "fusion_logical_index_test",
114    srcs = ["fusion_logical_index_test.cc"],
115    tags = tf_cuda_tests_tags(),
116    deps = [
117        ":gpu_codegen_test",
118        "//tensorflow/compiler/xla/tests:hlo_test_base",
119        "//tensorflow/core:test_main",
120    ],
121)
122
123tf_cc_test(
124    name = "mlir_fft_test",
125    srcs = ["mlir_fft_test.cc"],
126    tags = tf_cuda_tests_tags(),
127    deps = [
128        ":mlir_gpu_test_base",
129        "//tensorflow/compiler/jit:xla_gpu_jit",
130        "//tensorflow/core:test",
131        "//tensorflow/core:test_main",
132    ],
133)
134
135tf_cc_test(
136    name = "mlir_gemm_test",
137    srcs = ["mlir_gemm_test.cc"],
138    tags = tf_cuda_tests_tags(),
139    deps = [
140        ":mlir_gpu_test_base",
141        "//tensorflow/compiler/jit:xla_gpu_jit",
142        "//tensorflow/core:test",
143        "//tensorflow/core:test_main",
144    ],
145)
146
147tf_cc_test(
148    name = "mlir_sorting_test",
149    srcs = ["mlir_sorting_test.cc"],
150    tags = tf_cuda_tests_tags(),
151    deps = [
152        ":mlir_gpu_test_base",
153        "//tensorflow/core:test",
154        "//tensorflow/core:test_main",
155    ],
156)
157
158tf_cc_test(
159    name = "mlir_gpu_compile_test",
160    srcs = ["mlir_gpu_compile_test.cc"],
161    tags = tf_cuda_tests_tags(),
162    deps = [
163        ":mlir_gpu_test_base",
164        "//tensorflow/core:test",
165        "//tensorflow/core:test_main",
166    ],
167)
168
169tf_cc_test(
170    name = "gpu_reduce_scatter_creator_test",
171    srcs = ["gpu_reduce_scatter_creator_test.cc"],
172    deps = [
173        "//tensorflow/compiler/xla:util",
174        "//tensorflow/compiler/xla:xla_data_proto_cc",
175        "//tensorflow/compiler/xla/service:hlo",
176        "//tensorflow/compiler/xla/service:hlo_matchers",
177        "//tensorflow/compiler/xla/service:hlo_parser",
178        "//tensorflow/compiler/xla/service:hlo_pass_pipeline",
179        "//tensorflow/compiler/xla/service:hlo_verifier",
180        "//tensorflow/compiler/xla/service/gpu:gpu_reduce_scatter_creator",
181        "//tensorflow/compiler/xla/tests:hlo_test_base",
182        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
183        "//tensorflow/core:test",
184    ],
185)
186
187tf_cc_test(
188    name = "gpu_spmd_e2e_compile_test",
189    size = "small",
190    srcs = ["gpu_spmd_e2e_compile_test.cc"],
191    tags = tf_cuda_tests_tags(),
192    deps = [
193        ":gpu_codegen_test",
194        "//tensorflow/compiler/xla/service:hlo_module_config",
195        "//tensorflow/compiler/xla/service:hlo_parser",
196        "//tensorflow/compiler/xla/service:hlo_query",
197        "//tensorflow/compiler/xla/tests:hlo_test_base",
198        "//tensorflow/core:test",
199        "//tensorflow/core:test_main",
200    ],
201)
202
203tf_cc_test(
204    name = "gemm_rewrite_test",
205    srcs = [
206        "gemm_rewrite_test.cc",
207    ],
208    tags = tf_cuda_tests_tags() + [
209        "no_rocm",
210    ],
211    deps = [
212        ":gpu_codegen_test",
213        "//tensorflow/compiler/xla:debug_options_flags",
214        "//tensorflow/compiler/xla:statusor",
215        "//tensorflow/compiler/xla:xla_proto_cc",
216        "//tensorflow/compiler/xla/service:gpu_plugin",
217        "//tensorflow/compiler/xla/service:hlo",
218        "//tensorflow/compiler/xla/service:hlo_module_config",
219        "//tensorflow/compiler/xla/service:hlo_parser",
220        "//tensorflow/compiler/xla/service:pattern_matcher",
221        "//tensorflow/compiler/xla/service:pattern_matcher_gmock",
222        "//tensorflow/compiler/xla/service/gpu:gemm_rewriter",
223        "//tensorflow/compiler/xla/service/gpu:gpu_executable",
224        "//tensorflow/compiler/xla/service/gpu:stream_executor_util",
225        "//tensorflow/compiler/xla/tests:filecheck",
226        "//tensorflow/compiler/xla/tests:hlo_test_base",
227        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
228        "//tensorflow/core:lib",
229        "//tensorflow/core:test",
230        "//tensorflow/core:test_main",
231        "//tensorflow/stream_executor/lib",
232        "@com_google_absl//absl/memory",
233    ],
234)
235
236tf_cc_test(
237    name = "gemm_broadcast_folding_rewrite_test",
238    srcs = [
239        "gemm_broadcast_folding_rewrite_test.cc",
240    ],
241    tags = tf_cuda_tests_tags() + [
242        "no_rocm",
243    ],
244    deps = [
245        ":gpu_codegen_test",
246        "//tensorflow/compiler/xla:error_spec",
247        "//tensorflow/compiler/xla/service:gpu_plugin",
248        "//tensorflow/compiler/xla/service/gpu:gemm_broadcast_folding_rewriter",
249        "//tensorflow/compiler/xla/service/gpu:gemm_rewriter",
250        "//tensorflow/core:test",
251        "//tensorflow/core:test_main",
252    ],
253)
254
255tf_cc_test(
256    name = "gpu_too_many_blocks_test",
257    srcs = [
258        "gpu_too_many_blocks_test.cc",
259    ],
260    tags = tf_cuda_tests_tags(),
261    deps = [
262        ":gpu_codegen_test",
263        "//tensorflow/compiler/xla:statusor",
264        "//tensorflow/compiler/xla/tests:hlo_test_base",
265        "//tensorflow/core:test",
266        "//tensorflow/core:test_main",
267    ],
268)
269
270tf_cc_test(
271    name = "reduction_degenerate_dim_remover_test",
272    srcs = [
273        "reduction_degenerate_dim_remover_test.cc",
274    ],
275    deps = [
276        "//tensorflow/compiler/xla:debug_options_flags",
277        "//tensorflow/compiler/xla:statusor",
278        "//tensorflow/compiler/xla/service:hlo",
279        "//tensorflow/compiler/xla/service:hlo_module_config",
280        "//tensorflow/compiler/xla/service:hlo_parser",
281        "//tensorflow/compiler/xla/service/gpu:reduction_degenerate_dim_remover",
282        "//tensorflow/compiler/xla/tests:filecheck",
283        "//tensorflow/compiler/xla/tests:hlo_test_base",
284        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
285        "//tensorflow/core:lib",
286        "//tensorflow/core:test",
287        "//tensorflow/core:test_main",
288        "//tensorflow/stream_executor/lib",
289        "@com_google_absl//absl/memory",
290    ],
291)
292
293tf_cc_test(
294    name = "reduction_layout_normalizer_test",
295    srcs = [
296        "reduction_layout_normalizer_test.cc",
297    ],
298    deps = [
299        "//tensorflow/compiler/xla:debug_options_flags",
300        "//tensorflow/compiler/xla:statusor",
301        "//tensorflow/compiler/xla/service:hlo",
302        "//tensorflow/compiler/xla/service:hlo_module_config",
303        "//tensorflow/compiler/xla/service:hlo_parser",
304        "//tensorflow/compiler/xla/service/gpu:reduction_layout_normalizer",
305        "//tensorflow/compiler/xla/tests:filecheck",
306        "//tensorflow/compiler/xla/tests:hlo_test_base",
307        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
308        "//tensorflow/core:lib",
309        "//tensorflow/core:test",
310        "//tensorflow/core:test_main",
311        "@com_google_absl//absl/memory",
312    ],
313)
314
315tf_cc_test(
316    name = "tree_reduction_rewriter_test",
317    srcs = [
318        "tree_reduction_rewriter_test.cc",
319    ],
320    deps = [
321        "//tensorflow/compiler/xla:debug_options_flags",
322        "//tensorflow/compiler/xla:statusor",
323        "//tensorflow/compiler/xla/service:hlo",
324        "//tensorflow/compiler/xla/service:hlo_module_config",
325        "//tensorflow/compiler/xla/service:hlo_parser",
326        "//tensorflow/compiler/xla/service/gpu:tree_reduction_rewriter",
327        "//tensorflow/compiler/xla/tests:filecheck",
328        "//tensorflow/compiler/xla/tests:hlo_test_base",
329        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
330        "//tensorflow/core:lib",
331        "//tensorflow/core:test",
332        "//tensorflow/core:test_main",
333        "//tensorflow/stream_executor/lib",
334        "@com_google_absl//absl/memory",
335        "@com_google_absl//absl/strings",
336    ],
337)
338
339tf_cc_test(
340    name = "swap_conv_operands_test",
341    srcs = [
342        "swap_conv_operands_test.cc",
343    ],
344    tags = ["no_rocm"] + tf_cuda_tests_tags(),
345    deps = [
346        ":gpu_codegen_test",
347        "//tensorflow/compiler/xla:debug_options_flags",
348        "//tensorflow/compiler/xla:statusor",
349        "//tensorflow/compiler/xla/service:gpu_plugin",
350        "//tensorflow/compiler/xla/service:hlo",
351        "//tensorflow/compiler/xla/service:hlo_module_config",
352        "//tensorflow/compiler/xla/service:hlo_parser",
353        "//tensorflow/compiler/xla/service/gpu:gemm_rewriter",
354        "//tensorflow/compiler/xla/service/gpu:gpu_executable",
355        "//tensorflow/compiler/xla/tests:filecheck",
356        "//tensorflow/compiler/xla/tests:hlo_test_base",
357        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
358        "//tensorflow/core:lib",
359        "//tensorflow/core:test",
360        "//tensorflow/core:test_main",
361        "//tensorflow/stream_executor/lib",
362        "@com_google_absl//absl/memory",
363    ],
364)
365
366tf_cc_test(
367    name = "reduction_vectorization_test",
368    srcs = [
369        "reduction_vectorization_test.cc",
370    ],
371    tags = tf_cuda_tests_tags(),
372    deps = [
373        ":gpu_codegen_test",
374        "//tensorflow/compiler/xla:debug_options_flags",
375        "//tensorflow/compiler/xla:statusor",
376        "//tensorflow/compiler/xla/service:gpu_plugin",
377        "//tensorflow/compiler/xla/service:hlo",
378        "//tensorflow/compiler/xla/service:hlo_module_config",
379        "//tensorflow/compiler/xla/service:hlo_parser",
380        "//tensorflow/compiler/xla/service/gpu:gemm_rewriter",
381        "//tensorflow/compiler/xla/service/gpu:gpu_executable",
382        "//tensorflow/compiler/xla/tests:filecheck",
383        "//tensorflow/compiler/xla/tests:hlo_test_base",
384        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
385        "//tensorflow/core:lib",
386        "//tensorflow/core:test",
387        "//tensorflow/core:test_main",
388        "//tensorflow/stream_executor/lib",
389        "@com_google_absl//absl/memory",
390        "@com_google_absl//absl/strings",
391    ],
392)
393
394tf_cc_test(
395    name = "reduction_dimension_grouper_test",
396    srcs = [
397        "reduction_dimension_grouper_test.cc",
398    ],
399    deps = [
400        "//tensorflow/compiler/xla:debug_options_flags",
401        "//tensorflow/compiler/xla:statusor",
402        "//tensorflow/compiler/xla/service:hlo",
403        "//tensorflow/compiler/xla/service:hlo_module_config",
404        "//tensorflow/compiler/xla/service:hlo_parser",
405        "//tensorflow/compiler/xla/service/gpu:reduction_dimension_grouper",
406        "//tensorflow/compiler/xla/tests:filecheck",
407        "//tensorflow/compiler/xla/tests:hlo_test_base",
408        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
409        "//tensorflow/core:lib",
410        "//tensorflow/core:test",
411        "//tensorflow/core:test_main",
412        "//tensorflow/stream_executor/lib",
413        "@com_google_absl//absl/memory",
414    ],
415)
416
417tf_cc_test(
418    name = "parallel_reduction_test",
419    srcs = [
420        "parallel_reduction_test.cc",
421    ],
422    tags = tf_cuda_tests_tags(),
423    deps = [
424        ":gpu_codegen_test",
425        "//tensorflow/compiler/xla/service:gpu_plugin",
426        "//tensorflow/compiler/xla/service:hlo",
427        "//tensorflow/compiler/xla/service:hlo_module_config",
428        "//tensorflow/compiler/xla/service:hlo_parser",
429        "//tensorflow/compiler/xla/service/gpu:gpu_executable",
430        "//tensorflow/compiler/xla/tests:filecheck",
431        "//tensorflow/compiler/xla/tests:hlo_test_base",
432        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
433        "//tensorflow/core:lib",
434        "//tensorflow/core:test",
435        "//tensorflow/core:test_main",
436    ],
437)
438
439tf_cc_test(
440    name = "gpu_compilation_parallelism_test",
441    srcs = [
442        "gpu_compilation_parallelism_test.cc",
443    ],
444    tags = tf_cuda_tests_tags(),
445    deps = [
446        ":gpu_codegen_test",
447        "//tensorflow/compiler/xla/service:hlo",
448        "//tensorflow/compiler/xla/service:hlo_module_config",
449        "//tensorflow/core:test_main",
450    ],
451)
452
453tf_cc_test(
454    name = "gpu_copy_test",
455    srcs = ["gpu_copy_test.cc"],
456    tags = tf_cuda_tests_tags(),
457    deps = [
458        ":gpu_codegen_test",
459        "//tensorflow/compiler/xla:literal",
460        "//tensorflow/compiler/xla:literal_util",
461        "//tensorflow/compiler/xla:util",
462        "//tensorflow/compiler/xla:xla_data_proto_cc",
463        "//tensorflow/compiler/xla/service:hlo",
464        "//tensorflow/core:test",
465        "//tensorflow/core:test_main",
466        "@com_google_absl//absl/memory",
467    ],
468)
469
470tf_cc_test(
471    name = "gpu_copy_alone_test",
472    srcs = [
473        "gpu_copy_alone_test.cc",
474    ],
475    tags = tf_cuda_tests_tags(),
476    deps = [
477        ":gpu_codegen_test",
478        "//tensorflow/compiler/xla/service:hlo",
479        "//tensorflow/compiler/xla/service:hlo_module_config",
480        "//tensorflow/core:test_main",
481    ],
482)
483
484tf_cc_test(
485    name = "gpu_dyn_shape_test",
486    srcs = ["gpu_dyn_shape_test.cc"],
487    tags = tf_cuda_tests_tags(),
488    deps = [
489        ":gpu_codegen_test",
490        "//tensorflow/compiler/xla/service:hlo",
491        "//tensorflow/compiler/xla/service:hlo_module_config",
492        "//tensorflow/compiler/xla/service:hlo_parser",
493        "//tensorflow/compiler/xla/tests:hlo_test_base",
494        "//tensorflow/core:test",
495        "//tensorflow/core:test_main",
496    ],
497)
498
499tf_cc_test(
500    name = "gpu_ftz_test",
501    srcs = ["gpu_ftz_test.cc"],
502    tags = tf_cuda_tests_tags(),
503    deps = [
504        ":gpu_codegen_test",
505        "//tensorflow/compiler/xla/tests:verified_hlo_module",
506        "//tensorflow/core:test_main",
507    ],
508)
509
510tf_cc_test(
511    name = "gpu_index_test",
512    srcs = ["gpu_index_test.cc"],
513    tags = tf_cuda_tests_tags(),
514    deps = [
515        ":gpu_codegen_test",
516        "//tensorflow/compiler/xla:literal",
517        "//tensorflow/compiler/xla:shape_util",
518        "//tensorflow/compiler/xla:util",
519        "//tensorflow/compiler/xla:xla_data_proto_cc",
520        "//tensorflow/compiler/xla:xla_proto_cc",
521        "//tensorflow/compiler/xla/service:hlo",
522        "//tensorflow/compiler/xla/service:hlo_module_config",
523        "//tensorflow/compiler/xla/service:hlo_parser",
524        "//tensorflow/compiler/xla/tests:hlo_test_base",
525        "//tensorflow/core:test",
526        "//tensorflow/core:test_main",
527        "@com_google_absl//absl/memory",
528    ],
529)
530
531tf_cc_test(
532    name = "gpu_infeed_test",
533    srcs = ["infeed_test.cc"],
534    tags = tf_cuda_tests_tags(),
535    deps = [
536        ":gpu_codegen_test",
537        "//tensorflow/compiler/xla:literal",
538        "//tensorflow/compiler/xla:shape_util",
539        "//tensorflow/compiler/xla:test_helpers",
540        "//tensorflow/compiler/xla:xla_data_proto_cc",
541        "//tensorflow/compiler/xla/client:global_data",
542        "//tensorflow/compiler/xla/client:local_client",
543        "//tensorflow/compiler/xla/client:xla_builder",
544        "//tensorflow/compiler/xla/client/lib:arithmetic",
545        "//tensorflow/compiler/xla/tests:client_library_test_base",
546        "//tensorflow/compiler/xla/tests:literal_test_util",
547        "//tensorflow/core:lib",
548        "//tensorflow/core:test_main",
549    ],
550)
551
552tf_cc_test(
553    name = "gpu_kernel_tiling_test",
554    srcs = ["gpu_kernel_tiling_test.cc"],
555    tags = tf_cuda_tests_tags(),
556    deps = [
557        ":gpu_codegen_test",
558        "//tensorflow/compiler/xla/service:hlo",
559        "//tensorflow/compiler/xla/service:hlo_module_config",
560        "//tensorflow/compiler/xla/service:hlo_parser",
561        "//tensorflow/compiler/xla/tests:hlo_test_base",
562        "//tensorflow/core:test",
563        "//tensorflow/core:test_main",
564        "@com_google_absl//absl/strings",
565    ],
566)
567
568tf_cc_test(
569    name = "gpu_ldg_test",
570    srcs = ["gpu_ldg_test.cc"],
571    tags = tf_cuda_tests_tags(),
572    deps = [
573        ":gpu_codegen_test",
574        "//tensorflow/compiler/xla:literal",
575        "//tensorflow/compiler/xla:shape_util",
576        "//tensorflow/compiler/xla:util",
577        "//tensorflow/compiler/xla:xla_data_proto_cc",
578        "//tensorflow/compiler/xla/service:hlo",
579        "//tensorflow/core:test",
580        "//tensorflow/core:test_main",
581        "@com_google_absl//absl/memory",
582    ],
583)
584
585tf_cc_test(
586    name = "gpu_noalias_test",
587    srcs = ["gpu_noalias_test.cc"],
588    tags = tf_cuda_tests_tags(),
589    deps = [
590        ":gpu_codegen_test",
591        "//tensorflow/compiler/xla:literal",
592        "//tensorflow/compiler/xla:shape_util",
593        "//tensorflow/compiler/xla:util",
594        "//tensorflow/compiler/xla:xla_data_proto_cc",
595        "//tensorflow/compiler/xla/service:hlo",
596        "//tensorflow/core:test",
597        "//tensorflow/core:test_main",
598        "@com_google_absl//absl/memory",
599    ],
600)
601
602tf_cc_test(
603    name = "gpu_fusion_test",
604    srcs = ["gpu_fusion_test.cc"],
605    tags = tf_cuda_tests_tags(),
606    deps = [
607        ":gpu_codegen_test",
608        "//tensorflow/compiler/xla/service:hlo_module_config",
609        "//tensorflow/compiler/xla/service:hlo_parser",
610        "//tensorflow/compiler/xla/service/gpu:gpu_fusible",
611        "//tensorflow/compiler/xla/service/gpu:instruction_fusion",
612        "//tensorflow/compiler/xla/tests:hlo_test_base",
613        "//tensorflow/core:test",
614        "//tensorflow/core:test_main",
615    ],
616)
617
618tf_cc_test(
619    name = "gpu_unrolling_test",
620    srcs = ["gpu_unrolling_test.cc"],
621    tags = tf_cuda_tests_tags() + ["no_rocm"],
622    deps = [
623        ":gpu_codegen_test",
624        "//tensorflow/compiler/xla/service:hlo_module_config",
625        "//tensorflow/compiler/xla/service:hlo_parser",
626        "//tensorflow/compiler/xla/tests:hlo_test_base",
627        "//tensorflow/core:test",
628        "//tensorflow/core:test_main",
629    ],
630)
631
632tf_cc_test(
633    name = "gpu_alignment_test",
634    testonly = True,
635    srcs = ["gpu_alignment_test.cc"],
636    tags = tf_cuda_tests_tags(),
637    deps = [
638        ":gpu_codegen_test",
639        "//tensorflow/compiler/xla/service:custom_call_target_registry",
640        "//tensorflow/compiler/xla/service:gpu_plugin",
641        "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis",
642        "//tensorflow/compiler/xla/tests:filecheck",
643        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
644        "//tensorflow/core:test",
645        "//tensorflow/core:test_main",
646    ],
647)
648
649tf_cc_test(
650    name = "gpu_atomic_test",
651    srcs = ["gpu_atomic_test.cc"],
652    tags = tf_cuda_tests_tags(),
653    deps = [
654        ":gpu_codegen_test",
655        "//tensorflow/compiler/xla/tests:filecheck",
656        "//tensorflow/core:test",
657        "//tensorflow/core:test_main",
658    ],
659)
660
661tf_cc_test(
662    name = "gpu_input_fusible_slice_test",
663    srcs = ["gpu_input_fusible_slice_test.cc"],
664    tags = tf_cuda_tests_tags(),
665    deps = [
666        ":gpu_codegen_test",
667        "//tensorflow/compiler/xla/service:hlo",
668        "//tensorflow/compiler/xla/service:hlo_module_config",
669        "//tensorflow/compiler/xla/service:hlo_parser",
670        "//tensorflow/compiler/xla/tests:hlo_test_base",
671        "//tensorflow/core:test",
672        "//tensorflow/core:test_main",
673    ],
674)
675
676xla_test(
677    name = "gpu_convolution_regression_test",
678    srcs = ["gpu_convolution_regression_test.cc"],
679    backend_args = {"gpu": [
680        "--xla_enable_hlo_passes_only=layout-assignment,gpu-conv-algorithm-picker",
681        "--xla_gpu_crash_on_verification_failures",
682    ]},
683    backends = ["gpu"],
684    tags = [
685        "manual",
686        "no_oss",
687        "notap",
688    ],
689    deps = [
690        "//tensorflow/compiler/xla/tests:hlo_test_base",
691        "//tensorflow/compiler/xla/tests:xla_internal_test_main",
692    ],
693)
694
695tf_cc_test(
696    name = "select_and_scatter_test",
697    srcs = ["select_and_scatter_test.cc"],
698    tags = tf_cuda_tests_tags(),
699    deps = [
700        ":gpu_codegen_test",
701        "//tensorflow/compiler/xla:error_spec",
702        "//tensorflow/compiler/xla/tests:hlo_test_base",
703        "//tensorflow/core:test_main",
704    ],
705)
706
707tf_cc_test(
708    name = "sorting_test",
709    srcs = ["sorting_test.cc"],
710    tags = tf_cuda_tests_tags(),
711    deps = [
712        ":gpu_codegen_test",
713        "//tensorflow/compiler/xla:debug_options_flags",
714        "//tensorflow/compiler/xla:statusor",
715        "//tensorflow/compiler/xla:xla_proto_cc",
716        "//tensorflow/compiler/xla/service:gpu_plugin",
717        "//tensorflow/compiler/xla/service:hlo",
718        "//tensorflow/compiler/xla/service:hlo_module_config",
719        "//tensorflow/compiler/xla/service:hlo_parser",
720        "//tensorflow/compiler/xla/service/gpu:gpu_executable",
721        "//tensorflow/compiler/xla/tests:filecheck",
722        "//tensorflow/compiler/xla/tests:hlo_test_base",
723        "//tensorflow/compiler/xla/tests:llvm_irgen_test_base",
724        "//tensorflow/core:lib",
725        "//tensorflow/core:test",
726        "//tensorflow/core:test_main",
727        "//tensorflow/stream_executor/lib",
728        "@com_google_absl//absl/memory",
729    ],
730)
731
732tf_cc_binary(
733    name = "hlo_to_llvm_ir",
734    srcs = ["hlo_to_llvm_ir.cc"],
735    copts = if_cuda_is_configured(["-DGOOGLE_CUDA=1"]),
736    deps = [
737        "@llvm-project//llvm:Target",
738        "//tensorflow/compiler/xla:status",
739        "//tensorflow/compiler/xla/service:hlo",
740        "//tensorflow/compiler/xla/service/gpu:gpu_compiler",
741        "//tensorflow/compiler/xla/service/gpu:gpu_device_info",
742        "//tensorflow/compiler/xla/service/gpu:target_constants",
743        "//tensorflow/compiler/xla/service/gpu/llvm_gpu_backend",
744        "//tensorflow/compiler/xla/tools:hlo_module_loader",
745        "//tensorflow/core:framework_internal",
746        "//tensorflow/core:lib",
747        "//tensorflow/stream_executor/cuda:cuda_platform_id",
748    ] + if_cuda_is_configured([
749        "//tensorflow/compiler/xla/service/gpu:nvptx_compiler_impl",
750        "//tensorflow/compiler/xla/service/gpu:nvptx_helper",
751    ]) + if_rocm_is_configured([
752        "//tensorflow/compiler/xla/service/gpu:amdgpu_compiler_impl",
753    ]),
754)
755
756glob_lit_tests(
757    data = [":test_utilities"],
758    default_tags = tf_cuda_tests_tags() + [
759        "no_pip",
760    ],
761    driver = "@llvm-project//mlir:run_lit.sh",
762    exclude = ["execute_memzero_thunk.mlir"],
763    tags_override = {
764        "reduction_vectorization_sm_all.hlo": ["no_rocm"],
765        "element_wise_row_vectorization.hlo": ["no_rocm"],
766    },
767    test_file_exts = ["hlo"],
768)
769
770# Bundle together all of the test utilities that are used by tests.
771filegroup(
772    name = "test_utilities",
773    testonly = True,
774    data = [
775        ":hlo_to_llvm_ir",
776        "@llvm-project//llvm:FileCheck",
777    ],
778)
779
780tf_cc_test(
781    name = "kernel_launch_test",
782    srcs = ["kernel_launch_test.cc"],
783    tags = tf_cuda_tests_tags(),
784    deps = [
785        ":gpu_codegen_test",
786        "//tensorflow/compiler/xla/tests:hlo_test_base",
787        "//tensorflow/core:lib",
788        "//tensorflow/core:test",
789        "//tensorflow/core:test_main",
790    ],
791)
792
793tf_cc_test(
794    name = "in_place_op_test",
795    srcs = ["in_place_op_test.cc"],
796    tags = tf_cuda_tests_tags(),
797    deps = [
798        "//tensorflow/compiler/xla/service:gpu_plugin",
799        "//tensorflow/compiler/xla/tests:hlo_test_base",
800        "//tensorflow/core:test_main",
801    ],
802)
803
804py_test(
805    name = "mnist",
806    srcs = ["mnist.py"],
807    python_version = "PY3",
808    tags = [
809        "manual",
810        "no_oss",
811        "notap",
812        "requires-gpu-nvidia",
813        "requires-net:external",
814    ],
815    deps = [
816        "//tensorflow:tensorflow_py",
817        "@absl_py//absl:app",
818    ],
819)
820