New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add new matmul test with M and N tiled as `scf.for` loop nest instead of `air.launch` #827

Draft

erwei-xilinx wants to merge 1 commit into Xilinx:main from erwei-xilinx:new_matmul_tiling_strategy#

Collaborator

erwei-xilinx commented Dec 24, 2024

Pseudo code for the GEMM tiling strategy:

air.launch in (1, 1):
  for n in range(N):
    for m in range(M):
      zero_fill_acc
      for k in range(K):
        air.herd %h_m, %h_n in (4, 4):
          mmul
      memcpy_acc

Depends on #825 and #826.


          Add new matmul test with M and N tiled as scf.for loops instead of ai…

…r.launch

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet