colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add MLIR generated ExpOp GPU kernel for complex types. 

We lower lmhlo::ExpOp to mlir::complex::ExpOp for complex types.

PiperOrigin-RevId: 379432147
This commit is contained in:
Adrian Kuegel 2021-06-15 00:44:45 -07:00 committed by TensorFlow MLIR Team
parent f9f7a63870
commit 399dae666d
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/mlir-hlo/Dialect/mhlo/transforms/map_lmhlo_to_scalar_op.h
View File

@ -323,7 +323,8 @@ inline Value MapLhloOpToStdScalarOp<lmhlo::ExpOp>(Location loc,
ArrayRef<Type> arg_types,
ArrayRef<Value> args,
OpBuilder* b) {
return MapLhloOpToScalarOpImpl<isFloatType, ::mlir::math::ExpOp>{}(
return MapLhloOpToScalarOpImpl<isFloatType, ::mlir::math::ExpOp,
isComplexType, ::mlir::complex::ExpOp>{}(
loc, result_types, arg_types, args, b);
}