From 73d1584c3e29e0084d18cf1f99c344b1ab8c007e Mon Sep 17 00:00:00 2001 From: Zoran Jovanovic Date: Wed, 29 May 2024 17:04:24 +0000 Subject: [PATCH] [ROCm] Fix build break in xla/service/gpu/ir_emitter_triton_rocm.cc --- xla/service/gpu/ir_emitter_triton_rocm.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/xla/service/gpu/ir_emitter_triton_rocm.cc b/xla/service/gpu/ir_emitter_triton_rocm.cc index ead31a2fb6fcb..0401a8c67a534 100644 --- a/xla/service/gpu/ir_emitter_triton_rocm.cc +++ b/xla/service/gpu/ir_emitter_triton_rocm.cc @@ -79,8 +79,8 @@ absl::Status CreateTritonPipeline( // TODO ROCm Check if we want to compare MI100 and greater pm.addPass(mt::gpu::createTritonGPUOptimizeDotOperands({true})); pm.addPass(mlir::createCSEPass()); - pm.addPass(mt::gpu::createTritonGPUPipeline( - {config.num_stages, config.num_warps, config.num_ctas, ccAsInt})); + pm.addPass(mt::gpu::createTritonGPUPipeline({config.num_stages, config.num_warps, + config.num_ctas, ccAsInt})); pm.addPass(mt::gpu::createTritonGPUPrefetch()); // TODO ROCm Check if we want to compare MI100 and greater