Handle reshapes while fusing inputs in MLIR

ROCm / AMDMIGraphX

AMD's graph optimization engine.

https://rocm.docs.amd.com/projects/AMDMIGraphX/en/latest/

MIT License

167 stars 76 forks source link

Handle reshapes while fusing inputs in MLIR #3211

Open umangyadav opened 1 week ago

umangyadav commented 1 week ago

https://github.com/ROCm/AMDMIGraphX/pull/3010/files#r1648254890

With #3010 MIGraphX can fuse pointwise inputs for the Dot/conv instruction for MLIR. It is not handling Reshapes that happen on pointwise instruction inputs. We can add those.

umangyadav commented 1 week ago

In general it should handle following patterns: reshapes (optional) --> pointwise --> reshapes (optional) --> gemm_based_ins --> reshapes(optional) --> pointwise