@@ -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-
439374cc_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" ,
0 commit comments