[Triton-MLIR] Add ex2.approx implementation for ExpOp and fix smem allocation for ReduceOpConversion (#875)

2022-11-15 09:27:32 +08:00
parent c28cfd821b
commit 4c4159c6fa
5 changed files with 58 additions and 0 deletions
--- a/python/tests/test_elementwise.py
+++ b/python/tests/test_elementwise.py
@@ -61,6 +61,7 @@ def get_tensor(shape, data_type, b_positive=False):
                             ('sqrt', 'float64', 'float64'),
                             ('abs', 'float32', 'float32'),
                             ('exp', 'float32', 'float32'),
+                             ('exp', 'float64', 'float64'),
                             ('sigmoid', 'float32', 'float32'),
                          ])
 def test_single_input(expr, output_type, input0_type):