[conv] Expore different filter layouts for convolution

nod-ai / SHARK-ModelDev

Unified compiler/runtime for interfacing with PyTorch Dynamo.

Apache License 2.0

95 stars 48 forks source link

Closed antiagainst closed 4 months ago

antiagainst commented 6 months ago

Explore different convolution filter layouts (NHWC, HCHW, etc.) to see if they improve conv performance.

antiagainst commented 5 months ago

Update 5/22: added new 1D nhw op and vectorization. seems slow. next to verify numerics and profile.

antiagainst commented 4 months ago

Closing as proven to be not very helpful.