Is that possible to do operations other than element-wise operations in epilogue?

[daadaada]

Hi all,

I want to transform the output of matrix multiplication, but the transformation is not about element-wise operation. Actually, I want to do a transformation, which is based on a small region of matrix multiplication result. For example, I want to calculate the average of all 2x2 tiles in matrix multiplication result, and write the average number to the corresponding tile region.

And, if I want to do something like above before matrix multiplication, how can I solve this with CUTLASS (or is that possible/suitable to do this with CUTLASS)?

Thanks!

[daadaada]

Just want to show you what I want to do:

If the matrix multiplication result is:
1 2 3 4
2 3 4 5
2 3 4 5
4 5 6 7

I want to get:
2 2 4 4
2 2 4 4
3.5 3.5 5.5 5.5
3.5 3.5 5.5 5.5