Skip to content
This repository was archived by the owner on Jan 25, 2023. It is now read-only.

[MLIR] Fusing optimizations #200

Merged
Hardcode84 merged 15 commits into from
Mar 11, 2021
Merged

[MLIR] Fusing optimizations #200

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
bac167e
PostPlierToLinalgPass pass
Feb 26, 2021
6829be2
refactor out populate_common_opts_patterns
Feb 26, 2021
4e6c841
propagate if const values
Feb 26, 2021
bb8007e
simplify expand dims
Mar 10, 2021
4f63b37
plier inliner interface
Mar 10, 2021
0e7603b
simplify if and select
Mar 10, 2021
30af88b
fixes
Mar 10, 2021
aa0b0b6
fixes
Mar 10, 2021
6afed31
EnforceShapeOp
Mar 10, 2021
8a5854d
copy removal pass and parallel loop fusion
Mar 11, 2021
f70efc6
fixes
Mar 11, 2021
1b8e7ad
more SelectOp folding
Mar 11, 2021
0cec129
Add non-recursive CSE
Mar 11, 2021
9a35ad3
more efficient reshape
Mar 11, 2021
21e8265
run few rounds ParallelOp fusing
Mar 11, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
more efficient reshape
  • Loading branch information
Butygin
Butygin committed Mar 11, 2021
commit 9a35ad3deba93a305f0db116945d9725b252bb5a
11 changes: 5 additions & 6 deletions numba/mlir/numpy/funcs.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -211,12 +211,11 @@ def reshape_impl(builder, arg, new_shape):
flat = flatten(builder, arg, src_count)
init = builder.init_tensor(new_shape, arg.dtype)

iterators = ['parallel']
# dims1 = ','.join(['d%s' % i for i in range(count)])
# dims2 = ','.join(['d%s' % i if i == size_index else '0' for i in range(count)])
dims3 = ','.join(['d0' if i == size_index else '0' for i in range(count)])
expr1 = f'(d0) -> (d0)'
expr2 = f'(d0) -> ({dims3})'
iterators = ['parallel' for _ in range(count)]
dims1 = ','.join(['d%s' % i for i in range(count)])
dims3 = ','.join(['d%s' % i if i == size_index else '0' for i in range(count)])
expr1 = f'({dims1}) -> (d{size_index})'
expr2 = f'({dims1}) -> ({dims1})'
maps = [expr1, expr2]

def body(a, b):
Expand Down