This PR adds support for CPU weight prepacking. To be specific, this PR adds a new pass AttachAttrLayoutFreeBuffers to attach layout free buffers to the weight parameters, so that we can leverage MetaSchedule to optimize the prepacking process.
After the pass and tuning, we introduce a new pass SplitLayoutRewritePreproc to split the layout rewrite pass into multiple functions, so that we can lift the parameters transform pass function with existing pass.
This PR adds support for CPU weight prepacking. To be specific, this PR adds a new pass
AttachAttrLayoutFreeBuffers
to attach layout free buffers to the weight parameters, so that we can leverage MetaSchedule to optimize the prepacking process.After the pass and tuning, we introduce a new pass
SplitLayoutRewritePreproc
to split the layout rewrite pass into multiple functions, so that we can lift the parameters transform pass function with existing pass.