mirror of
https://github.com/tensorflow/tensorflow.git
synced 2026-09-29 16:58:55 +08:00
[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:
committed by
TensorFlower Gardener
parent
9bb279da4c
commit
65c21117a1
@@ -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"],
|
||||
|
||||
-1247
File diff suppressed because it is too large
Load Diff
+1426
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user