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

Implement register stealing #3584

Open

zasdfgbnm opened this issue Dec 12, 2024 · 0 comments

Labels

Matmuls

Collaborator

zasdfgbnm commented Dec 12, 2024 •

edited

Loading

Change our warp specialized kernel to something like:

if load-warp:
  asm volatile("{setmaxnreg.dec.sync.aligned.u32 56; \n\t}");
  do the work
  return; # Super important!
else:
  asm volatile("{setmaxnreg.inc.sync.aligned.u32 224; \n\t}");
  do the work

This seems to improve performance, see experiment at #3566

zasdfgbnm added the Matmuls label

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