From 31da0a28ae8cb7d69f6fb9506e14712edef9df49 Mon Sep 17 00:00:00 2001 From: Adrian Kuegel Date: Thu, 4 Apr 2024 01:33:36 -0700 Subject: [PATCH] Internal CI configuration to run tests on H100 Do some required BUILD file adjustments PiperOrigin-RevId: 621777011 --- xla/service/gpu/BUILD | 2 +- xla/service/gpu/kernels/BUILD | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/xla/service/gpu/BUILD b/xla/service/gpu/BUILD index 98e1e21a06a84..5bc5b739529cc 100644 --- a/xla/service/gpu/BUILD +++ b/xla/service/gpu/BUILD @@ -654,7 +654,7 @@ xla_test( "requires-gpu-sm80", ]}, backends = [ - "gpu", + "gpu_a100", "gpu_h100", ], tags = [ diff --git a/xla/service/gpu/kernels/BUILD b/xla/service/gpu/kernels/BUILD index 7f800cd132b70..d754e975dfb52 100644 --- a/xla/service/gpu/kernels/BUILD +++ b/xla/service/gpu/kernels/BUILD @@ -105,6 +105,8 @@ xla_test( name = "cutlass_gemm_fusion_test", srcs = ["cutlass_gemm_fusion_test.cc"], backends = ["gpu"], + # TODO(b/332820384): Enable when it passes on H100. + disabled_backends = ["gpu_h100"], tags = ["no_rocm"], deps = [ ":custom_kernel_fusion_pattern",