From 9885d859b68b06a31d03fb6983a35f59aab61844 Mon Sep 17 00:00:00 2001 From: Ilia Sergachev Date: Wed, 15 May 2024 22:04:01 +0000 Subject: [PATCH] [GPU] Fix OSS compilation of a previously disabled test. --- xla/service/gpu/ir_emitter_triton_test.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/xla/service/gpu/ir_emitter_triton_test.cc b/xla/service/gpu/ir_emitter_triton_test.cc index a9a0ed4ba69e9d..94cad60fec9aea 100644 --- a/xla/service/gpu/ir_emitter_triton_test.cc +++ b/xla/service/gpu/ir_emitter_triton_test.cc @@ -31,7 +31,7 @@ limitations under the License. #include "absl/types/span.h" #include "llvm/IR/LLVMContext.h" #include "llvm/Support/raw_ostream.h" -#include "mlir/IR/MLIRContext.h" // from @llvm-project +#include "mlir/IR/MLIRContext.h" // from @llvm-project #include "mlir/Pass/PassManager.h" // from @llvm-project #include "xla/autotuning.pb.h" #include "xla/error_spec.h" @@ -5235,7 +5235,7 @@ ENTRY e { })"; TritonGemmConfig config(32, 16, 128, 1, 1, 4); - ASSERT_OK( + TF_ASSERT_OK( CreateTritonIrAndFileCheck(hlo_text, config, EmitMatMul, "triton_dot", R"( CHECK: tt.dot CHECK-NOT: inputPrecision = tf32