diff --git a/tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc b/tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc index 79e56e4a96f..176e7af5d8c 100644 --- a/tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc +++ b/tensorflow/compiler/xla/service/mlir_gpu/kernel_lowering.cc @@ -327,7 +327,7 @@ class LowerToNVVMPass ::mlir::gpu::GPUModuleOp m = getOperation(); ::mlir::OwningRewritePatternList patterns; - ::mlir::LinalgTypeConverter converter(m.getContext()); + ::mlir::LLVMTypeConverter converter(m.getContext()); ::mlir::populateStdToLLVMConversionPatterns(converter, patterns); // TODO(b/145824979) Remove linalg once sliceop is in std. ::mlir::populateLinalgToLLVMConversionPatterns(converter, patterns,