### Motivation and Expected Behavior The [attention example](https://github.com/cornell-zhang/allo/blob/main/examples/aie/attention.py) currently only supports XDNA1. We should add support for XDNA2. ### Related resources - `exp` and `softmax` kernel: [bf16_exp.cc](https://github.com/Xilinx/mlir-aie/blob/main/aie_kernels/aie2p/bf16_exp.cc), [softmax.cc](https://github.com/Xilinx/mlir-aie/blob/main/aie_kernels/aie2p/softmax.cc) ### TODOs - [x] add `exp` and `softmax` microkernels to our kernel library - [ ] resolve matmul kernel naming conflict
Motivation and Expected Behavior
The attention example currently only supports XDNA1. We should add support for XDNA2.
Related resources
expandsoftmaxkernel: bf16_exp.cc, softmax.ccTODOs
expandsoftmaxmicrokernels to our kernel library