Skip to content

Commit 7974b4c

Browse files
pifon2aGoogle-ML-Automation
authored andcommitted
[XLA:GPU] Remove cutlass & custom kernel fusions from XLA:GPU.
PiperOrigin-RevId: 900651828
1 parent 85c65a2 commit 7974b4c

38 files changed

+4
-5550
lines changed

xla/backends/gpu/autotuner/BUILD

Lines changed: 0 additions & 69 deletions
Original file line numberDiff line numberDiff line change
@@ -371,71 +371,6 @@ xla_test(
371371
],
372372
)
373373

374-
cc_library(
375-
name = "custom_kernel",
376-
srcs = ["custom_kernel.cc"],
377-
hdrs = ["custom_kernel.h"],
378-
deps = [
379-
":gpu_codegen_backend",
380-
"//xla:autotune_results_proto_cc",
381-
"//xla:autotuning_proto_cc",
382-
"//xla:xla_proto_cc",
383-
"//xla/backends/autotuner:backends_proto_cc",
384-
"//xla/backends/autotuner:codegen_backend",
385-
"//xla/backends/gpu/codegen/kernels:custom_kernel",
386-
"//xla/backends/gpu/codegen/kernels:custom_kernel_fusion",
387-
"//xla/hlo/ir:hlo",
388-
"//xla/service:compiler",
389-
"//xla/service/gpu:backend_configs_cc",
390-
"//xla/service/gpu:ir_emission_utils",
391-
"//xla/stream_executor:device_description",
392-
"//xla/stream_executor:stream_executor_h",
393-
"//xla/tsl/platform:errors",
394-
"//xla/tsl/platform:statusor",
395-
"@com_google_absl//absl/log",
396-
"@com_google_absl//absl/status",
397-
"@com_google_absl//absl/status:statusor",
398-
"@com_google_absl//absl/strings",
399-
],
400-
)
401-
402-
xla_test(
403-
name = "custom_kernel_test",
404-
srcs = ["custom_kernel_test.cc"],
405-
backends = [
406-
"a100",
407-
"h100",
408-
"b200",
409-
"amdgpu_any",
410-
],
411-
tags = [
412-
"gpu",
413-
"no_mac",
414-
],
415-
use_legacy_runtime = True,
416-
deps = [
417-
":custom_kernel",
418-
"//xla:autotuning_proto_cc",
419-
"//xla:xla_proto_cc",
420-
"//xla/backends/autotuner:codegen_backend",
421-
"//xla/hlo/ir:hlo",
422-
"//xla/hlo/testlib:filecheck",
423-
"//xla/hlo/testlib:hlo_hardware_independent_test_base",
424-
"//xla/service:compiler",
425-
"//xla/service:platform_util",
426-
"//xla/stream_executor:device_description_proto_cc",
427-
"//xla/stream_executor:platform",
428-
"//xla/stream_executor:stream_executor_h",
429-
"//xla/tsl/lib/core:status_test_util",
430-
"//xla/tsl/platform:statusor",
431-
"//xla/tsl/util/proto:proto_matchers",
432-
"@com_google_absl//absl/status",
433-
"@com_google_absl//absl/status:status_matchers",
434-
"@com_google_absl//absl/status:statusor",
435-
"@com_google_googletest//:gtest_main",
436-
],
437-
)
438-
439374
cc_library(
440375
name = "triton",
441376
srcs = ["triton.cc"],
@@ -650,14 +585,12 @@ cc_library(
650585
":cublas",
651586
":cublaslt",
652587
":cudnn",
653-
":custom_kernel",
654588
":factory",
655589
":fission_backend",
656590
":triton",
657591
"//xla:xla_proto_cc",
658592
"//xla/backends/autotuner:backends_proto_cc",
659593
"//xla/backends/autotuner:codegen_backend",
660-
"//xla/backends/gpu/transforms:custom_kernel_fusion_rewriter",
661594
"//xla/backends/gpu/transforms:dot_algorithm_rewriter",
662595
"//xla/backends/gpu/transforms:gemm_rewriter",
663596
"//xla/backends/gpu/transforms:scaled_dot_rewriter",
@@ -1000,15 +933,13 @@ xla_test(
1000933
tags = ["gpu"],
1001934
use_legacy_runtime = True,
1002935
deps = [
1003-
":custom_kernel",
1004936
":fission_backend",
1005937
":gpu_codegen_backend",
1006938
"//xla:autotune_results_proto_cc",
1007939
"//xla:autotuning_proto_cc",
1008940
"//xla:xla_proto_cc",
1009941
"//xla/backends/autotuner:backends_proto_cc",
1010942
"//xla/backends/autotuner:codegen_backend",
1011-
"//xla/backends/gpu/transforms:custom_kernel_fusion_rewriter",
1012943
"//xla/backends/gpu/transforms:dot_algorithm_rewriter",
1013944
"//xla/backends/gpu/transforms:gemm_rewriter",
1014945
"//xla/backends/gpu/transforms:scaled_dot_rewriter",

xla/backends/gpu/autotuner/custom_kernel.cc

Lines changed: 0 additions & 152 deletions
This file was deleted.

xla/backends/gpu/autotuner/custom_kernel.h

Lines changed: 0 additions & 60 deletions
This file was deleted.

0 commit comments

Comments
 (0)