[XLA:GPU] Move Triton scaled dot device tests to dedicated test file.

Extracts scaled dot tests from fusion_emitter_device_test into scaled_dot_device_test to isolate scaled dot test cases and reduce test compilation overhead.

PiperOrigin-RevId: 987286750
This commit is contained in:
Ilya Tikhonovskiy
2026-09-25 04:37:37 -07:00
committed by TensorFlower Gardener
parent 9bb279da4c
commit 65c21117a1
3 changed files with 1479 additions and 1247 deletions
@@ -226,6 +226,59 @@ xla_test(
],
)
xla_test(
name = "scaled_dot_device_test",
srcs = ["scaled_dot_device_test.cc"],
backends = [
"a100",
"h100",
"b200",
"amdgpu_any",
],
shard_count = 20,
tags = ["no_mac"],
deps = [
"//xla:comparison_util",
"//xla:error_spec",
"//xla:literal",
"//xla:literal_util",
"//xla:types",
"//xla:xla_data_proto_cc",
"//xla:xla_proto_cc",
"//xla/backends/gpu/codegen/triton:test_utils",
"//xla/backends/gpu/codegen/triton:xtile_compiler",
"//xla/backends/gpu/codegen/triton:xtile_test_base",
"//xla/backends/gpu/tests:gpu_pjrt_codegen_test",
"//xla/backends/gpu/transforms:composite_rewriter",
"//xla/codegen/xtile:block_level_parameters",
"//xla/hlo/ir:hlo",
"//xla/hlo/parser:hlo_parser",
"//xla/hlo/testlib:filecheck",
"//xla/hlo/testlib:verified_hlo_module",
"//xla/service/gpu:backend_configs_cc",
"//xla/service/gpu:gpu_compiler",
"//xla/service/gpu:gpu_device_info_for_tests",
"//xla/stream_executor:device_description",
"//xla/stream_executor/cuda:cuda_compute_capability",
"//xla/tests:hlo_interpreter_reference_mixin",
"//xla/tests:test_utils",
"//xla/tests:xla_internal_test_main", # fixdeps: keep
"//xla/tsl/platform:statusor",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/random:distributions",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:status_macros",
"@com_google_absl//absl/status:status_matchers",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
],
)
xla_test(
name = "fusion_emitter_large_test",
srcs = ["fusion_emitter_large_test.cc"],
File diff suppressed because it is too large Load Diff
File diff suppressed because it is too large Load Diff