How to use cascade streams West -> East

[A-Ayerh]

Hi, I am learning how to use MLIR-AIE for my capstone and I was interested in how we can cascade results from the AIE tiles from West -> East.

Some context:

• I am using the Ryzen AI Phoenix architecture (4 column, 6 rows, AIE-ML)
• I want to implement this cascade dataflow using IRON JIT.
• I am familiar with the Matrix Multiplication cascade example and have a working example of cascading downwards using IRON JIT.
• My end goal currently is to implement a Systolic Array pattern which cascades results both North -> South and West -> East.

I have gotten a look at the cascade_flow operation, but when I try adding it to my python file it doesn't seem to appear in the MLIR...

I'm stuck here.

Questions:

• Is this possible to run in IRON JIT currently?
• Is there a working example to show this cascading pattern outside of the MLIR examples?
• Is this Systolic Array design even possible, in the sense of performing cascade in both directions on the same cycle.

Than you for your time and looking forward to any feedback on how I may solve this.