Skip to content
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.

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

feat: allow print_latex(TiledMMA) to colorize sliced thread and add print_latex(ThrMMA) #1656

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

cloudhan
Copy link

Allow colorize only one thread of print_latex, to make mma pattern obvious and reduce eye strain.

For example,

#include "cute/tensor.hpp"
using namespace cute;
int main() {
  auto tiled_mma = make_tiled_mma(SM80_16x8x8_F32F16F16F32_TN{});
  auto thr_mma = tiled_mma.get_thread_slice(11);
  // print_latex(tiled_mma, 11);
  print_latex(thr_mma);  // the same as above
  return 0;
}

produces

latex

Copy link

This PR has been labeled inactive-30d due to no recent activity in the past 30 days. Please close this PR if it is no longer required. Otherwise, please respond with a comment indicating any updates. This PR will be labeled inactive-90d if there is no activity in the next 60 days.

Copy link

This PR has been labeled inactive-90d due to no recent activity in the past 90 days. Please close this PR if it is no longer required. Otherwise, please respond with a comment indicating any updates.

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

Successfully merging this pull request may close these issues.

1 participant