List of TODOs for the cutlass backend

mirage-project / mirage

A multi-level tensor algebra superoptimizer

Apache License 2.0

262 stars 16 forks source link

Open jiazhihao opened 1 month ago

jiazhihao commented 1 month ago

Support more general shapes by InputLoader (and OutputSaver) other than dimension sizes of 32 and 64.
Overlapping Input loading and TB Matmul by implementing a customized Matmul (so we can remove Input loading operators)
Support customized Epilogue by Matmul