at-dispatch-v2

This skill helps convert PyTorch's legacy AT_DISPATCH macros to the new AT_DISPATCH_V2 format, as defined in aten/src/ATen/Dispatch_v2.h . Use this skill when: Old format: New format: Add the v2 header near the existing #include : Keep the old Dispatch.h include for now (other code may still need it). Common patterns to convert: Identify which type group macro corresponds to the base types: For combined patterns, use multiple AT_EXPAND() entries: From AT_DISPATCH_*_AND2(type1, type2, ...) or AT_DISPATCH_*_AND3(type1, type2, type3, ...) , extract the individual types (type1, type2, etc.). These become the trailing arguments after the type group: Apply the transformation: Pattern: Example transformation: For lambdas with internal commas or complex expressions, AT_WRAP is essential: Check that: Available type group macros (use with AT_EXPAND() ): When asked to convert AT_DISPATCH macros: Do NOT compile or test the code - focus on accurate conversion only. - Converting AT_DISPATCH_* macros to AT_DISPATCH_V2 - Porting ATen kernels to use the new dispatch API - Working with files in aten/src/ATen/native/ that use dispatch macros - User mentions "AT_DISPATCH", "dispatch v2", "Dispatch_v2.h", or macro conversion - Reorder arguments : scalar_type and name come first, then lambda, then types - Wrap the lambda : Use AT_WRAP(lambda) to handle internal commas - Expand type groups : Use...