[Enhancement] Use cute::elect_one_sync() for slightly better performance

[Rachmanino]

Summary by CodeRabbit

• Refactor
  • Improved internal code path handling for specific GPU thread configurations to enhance execution efficiency.

_{✏️ Tip: You can customize this high-level summary in your review settings.}

[github-actions]

👋 Hi! Thank you for contributing to the TileLang project.

Please remember to run pre-commit run --all-files in the root directory of the project to ensure your changes are properly linted and formatted. This will help ensure your contribution passes the format check.

We appreciate you taking this step! Our team will review your contribution, and we look forward to your awesome work! 🚀

[coderabbitai]

📝 Walkthrough

Walkthrough

A special-case branch was added to the tl_shuffle_elect function in the CUDA intrin header for thread_extent == 32, allowing it to directly return cute::elect_one_sync() instead of evaluating the general case logic path.

Changes

Cohort / File(s)	Summary
CUDA elect optimization src/tl_templates/cuda/intrin.h	Added special-case branch for thread_extent == 32 in tl_shuffle_elect, returning cute::elect_one_sync() directly; precedes general case to skip unnecessary condition evaluation for this specific warp width.

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~10 minutes

Poem

🐰 A warp of thirty-two runs swift and true,
No lengthy conditionals to parse through,
Elect sync returns, direct as morning dew,
Fast paths emerge where special cases brew! ✨

🚥 Pre-merge checks | ✅ 2 | ❌ 1

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Docstring Coverage	⚠️ Warning	Docstring coverage is 0.00% which is insufficient. The required threshold is 80.00%.	Write docstrings for the functions missing them to satisfy the coverage threshold.

✅ Passed checks (2 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly describes the main change: using cute::elect_one_sync() for performance improvement in the shuffle election operation.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

✨ Finishing touches

• 📝 Generate docstrings

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}