From 73cb134cca408f28edf8913362c02e4d8e6b1bd7 Mon Sep 17 00:00:00 2001 From: Tim Shen Date: Thu, 3 Dec 2020 16:11:22 -0800 Subject: [PATCH] [MLIR] Add XLA HLO -> LMHLO conversion for all elementwise ops. PiperOrigin-RevId: 345557248 --- include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td | 2 ++ 1 file changed, 2 insertions(+) diff --git a/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td b/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td index 47b100a..3f302f1 100644 --- a/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td +++ b/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td @@ -85,6 +85,8 @@ def LHLO_AbsOp: LHLO_UnaryElementwiseOp<"abs">, BASE_HLO_AbsOp; def LHLO_BitcastConvertOp: LHLO_UnaryElementwiseOp<"bitcast_convert", LHLO_Buffer, [SameOperandsShape]>, BASE_HLO_BitcastConvertOp; +def LHLO_CbrtOp: LHLO_UnaryElementwiseOp<"cbrt", LHLO_FpBuffer>, BASE_HLO_CbrtOp; + def LHLO_CeilOp: LHLO_UnaryElementwiseOp<"ceil", LHLO_FpBuffer>, BASE_HLO_CeilOp; def LHLO_ClzOp: LHLO_UnaryElementwiseOp<"count_leading_zeros", LHLO_IntBuffer>, BASE_HLO_ClzOp;