| # Description: GPU-specific XLA tests. For example, codegen tests that |
| # verify the IR emitted. |
| # |
| # TODO(jlebar): None of these tests actually use the GPU, so they should not |
| # need to run on machines with GPUs present. |
| |
| load("//tensorflow/compiler/xla/tests:build_defs.bzl", "xla_test") |
| load("//tensorflow:tensorflow.bzl", "tf_cc_test") |
| load( |
| "//tensorflow/core/platform:build_config_root.bzl", |
| "tf_cuda_tests_tags", |
| ) |
| |
| package( |
| default_visibility = [":friends"], |
| licenses = ["notice"], # Apache 2.0 |
| ) |
| |
| package_group( |
| name = "friends", |
| includes = [ |
| "//tensorflow/compiler/xla:friends", |
| ], |
| ) |
| |
| # Filegroup used to collect source files for dependency checking. |
| filegroup( |
| name = "c_srcs", |
| data = glob([ |
| "**/*.cc", |
| "**/*.h", |
| ]), |
| ) |
| |
| cc_library( |
| name = "gpu_codegen_test", |
| testonly = True, |
| srcs = ["gpu_codegen_test.cc"], |
| hdrs = ["gpu_codegen_test.h"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| "//tensorflow/compiler/xla:debug_options_flags", |
| "//tensorflow/compiler/xla:shape_util", |
| "//tensorflow/compiler/xla/service:gpu_plugin", |
| "//tensorflow/compiler/xla/service/gpu:gpu_executable", |
| "//tensorflow/compiler/xla/tests:filecheck", |
| "//tensorflow/compiler/xla/tests:llvm_irgen_test_base", |
| "//tensorflow/compiler/xla/tests:verified_hlo_module", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:test", |
| "//tensorflow/stream_executor/lib", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gemm_rewrite_test", |
| srcs = [ |
| "gemm_rewrite_test.cc", |
| ], |
| tags = tf_cuda_tests_tags() + [ |
| "no_rocm", |
| ], |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:debug_options_flags", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/compiler/xla:xla_proto_cc", |
| "//tensorflow/compiler/xla/service:gpu_plugin", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/service/gpu:gemm_rewriter", |
| "//tensorflow/compiler/xla/service/gpu:gpu_executable", |
| "//tensorflow/compiler/xla/tests:filecheck", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/compiler/xla/tests:llvm_irgen_test_base", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "//tensorflow/stream_executor/lib", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "reduction_degenerate_dim_remover_test", |
| srcs = [ |
| "reduction_degenerate_dim_remover_test.cc", |
| ], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:debug_options_flags", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/compiler/xla/service:gpu_plugin", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/service/gpu:gemm_rewriter", |
| "//tensorflow/compiler/xla/service/gpu:gpu_executable", |
| "//tensorflow/compiler/xla/tests:filecheck", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/compiler/xla/tests:llvm_irgen_test_base", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "//tensorflow/stream_executor/lib", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "reduction_layout_normalizer_test", |
| srcs = [ |
| "reduction_layout_normalizer_test.cc", |
| ], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:debug_options_flags", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/compiler/xla/service:gpu_plugin", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/service/gpu:gemm_rewriter", |
| "//tensorflow/compiler/xla/service/gpu:gpu_executable", |
| "//tensorflow/compiler/xla/tests:filecheck", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/compiler/xla/tests:llvm_irgen_test_base", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "//tensorflow/stream_executor/lib", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "tree_reduction_rewriter_test", |
| srcs = [ |
| "tree_reduction_rewriter_test.cc", |
| ], |
| tags = tf_cuda_tests_tags() + ["no_rocm"], |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:debug_options_flags", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/compiler/xla/service:gpu_plugin", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/service/gpu:gemm_rewriter", |
| "//tensorflow/compiler/xla/service/gpu:gpu_executable", |
| "//tensorflow/compiler/xla/tests:filecheck", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/compiler/xla/tests:llvm_irgen_test_base", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "//tensorflow/stream_executor/lib", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "reduction_vectorization_test", |
| srcs = [ |
| "reduction_vectorization_test.cc", |
| ], |
| tags = tf_cuda_tests_tags() + ["no_rocm"], |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:debug_options_flags", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/compiler/xla/service:gpu_plugin", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/service/gpu:gemm_rewriter", |
| "//tensorflow/compiler/xla/service/gpu:gpu_executable", |
| "//tensorflow/compiler/xla/tests:filecheck", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/compiler/xla/tests:llvm_irgen_test_base", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "//tensorflow/stream_executor/lib", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "reduction_dimension_grouper_test", |
| srcs = [ |
| "reduction_dimension_grouper_test.cc", |
| ], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:debug_options_flags", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/compiler/xla/service:gpu_plugin", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/service/gpu:gemm_rewriter", |
| "//tensorflow/compiler/xla/service/gpu:gpu_executable", |
| "//tensorflow/compiler/xla/tests:filecheck", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/compiler/xla/tests:llvm_irgen_test_base", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "//tensorflow/stream_executor/lib", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_copy_test", |
| srcs = ["gpu_copy_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:literal", |
| "//tensorflow/compiler/xla:literal_util", |
| "//tensorflow/compiler/xla:util", |
| "//tensorflow/compiler/xla:xla_data_proto_cc", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_ftz_test", |
| srcs = ["gpu_ftz_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla/tests:verified_hlo_module", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_index_test", |
| srcs = ["gpu_index_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:literal", |
| "//tensorflow/compiler/xla:shape_util", |
| "//tensorflow/compiler/xla:util", |
| "//tensorflow/compiler/xla:xla_data_proto_cc", |
| "//tensorflow/compiler/xla:xla_proto_cc", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_infeed_test", |
| srcs = ["infeed_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:literal", |
| "//tensorflow/compiler/xla:shape_util", |
| "//tensorflow/compiler/xla:test_helpers", |
| "//tensorflow/compiler/xla:xla_data_proto_cc", |
| "//tensorflow/compiler/xla/client:global_data", |
| "//tensorflow/compiler/xla/client:local_client", |
| "//tensorflow/compiler/xla/client:xla_builder", |
| "//tensorflow/compiler/xla/client/lib:arithmetic", |
| "//tensorflow/compiler/xla/tests:client_library_test_base", |
| "//tensorflow/compiler/xla/tests:literal_test_util", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_kernel_tiling_test", |
| srcs = ["gpu_kernel_tiling_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_ldg_test", |
| srcs = ["gpu_ldg_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:literal", |
| "//tensorflow/compiler/xla:shape_util", |
| "//tensorflow/compiler/xla:util", |
| "//tensorflow/compiler/xla:xla_data_proto_cc", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_noalias_test", |
| srcs = ["gpu_noalias_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla:literal", |
| "//tensorflow/compiler/xla:shape_util", |
| "//tensorflow/compiler/xla:util", |
| "//tensorflow/compiler/xla:xla_data_proto_cc", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "@com_google_absl//absl/memory", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_fusion_test", |
| srcs = ["gpu_fusion_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_unrolling_test", |
| srcs = ["gpu_unrolling_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_alignment_test", |
| testonly = True, |
| srcs = ["gpu_alignment_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla/service:custom_call_target_registry", |
| "//tensorflow/compiler/xla/service:gpu_plugin", |
| "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis", |
| "//tensorflow/compiler/xla/tests:filecheck", |
| "//tensorflow/compiler/xla/tests:llvm_irgen_test_base", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_atomic_test", |
| srcs = ["gpu_atomic_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla/tests:filecheck", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "gpu_input_fusible_slice_test", |
| srcs = ["gpu_input_fusible_slice_test.cc"], |
| tags = tf_cuda_tests_tags(), |
| deps = [ |
| ":gpu_codegen_test", |
| "//tensorflow/compiler/xla/service:hlo", |
| "//tensorflow/compiler/xla/service:hlo_module_config", |
| "//tensorflow/compiler/xla/service:hlo_parser", |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| xla_test( |
| name = "gpu_convolution_regression_test", |
| srcs = ["gpu_convolution_regression_test.cc"], |
| backend_args = {"gpu": [ |
| "--xla_enable_hlo_passes_only=layout-assignment,gpu-conv-algorithm-picker", |
| "--xla_gpu_crash_on_verification_failures", |
| ]}, |
| backends = ["gpu"], |
| tags = [ |
| "manual", |
| "no_oss", |
| "notap", |
| ], |
| deps = [ |
| "//tensorflow/compiler/xla/tests:hlo_test_base", |
| "//tensorflow/compiler/xla/tests:xla_internal_test_main", |
| ], |
| ) |