Open jiazhihao opened 4 days ago
Threadblock Matmul operators currently do not perform epilogue: https://github.com/mirage-project/mirage/blob/main/src/transpiler/transpiler_tb.cc#L715-L725
This issue can be reproduced by running the demo/demo_gqa.py example, where the exp operator after matmul is not shown in the transpiled code.
demo/demo_gqa.py
exp
matmul
it should be done at here https://github.com/mirage-project/mirage/blob/main/include/mirage/transpiler/runtime/threadblock/matmul.h#L353 I can add other element unary ops
Threadblock Matmul operators currently do not perform epilogue: https://github.com/mirage-project/mirage/blob/main/src/transpiler/transpiler_tb.cc#L715-L725
This issue can be reproduced by running the
demo/demo_gqa.py
example, where theexp
operator aftermatmul
is not shown in the transpiled code.