add-uint-support

What problem does it solve?

This Skill automates the complex and error-prone process of adding unsigned integer (uint) type support to PyTorch operators, saving developers time and ensuring consistent implementation across kernels.

Core Features & Use Cases

• Automated Type Expansion: Automatically updates AT_DISPATCH macros to include uint16, uint32, and uint64 types.
• Consistent Implementation: Ensures that PyTorch operators correctly handle unsigned integer data, preventing type-related bugs.
• Decision Tree Guidance: Provides a clear decision tree to determine the best method for adding uint support (explicitly or by upgrading type groups).
• Use Case: You are porting a numerical algorithm to PyTorch that heavily relies on unsigned 32-bit integers. Instead of manually modifying every AT_DISPATCH macro and risking errors, you use this Skill to automatically update the necessary operator definitions, ensuring your algorithm runs correctly with uint types.

Quick Start

Add unsigned integer support to the PyTorch operator defined in my_cuda_kernel.cpp.