AT_DISPATCHからAT_DISPATCH_V2へのコンバーター

このスキルは、PyTorchのレガシーAT_DISPATCHマクロを新しいAT_DISPATCH_V2形式に変換するのに役立ちます。形式は aten/src/ATen/Dispatch_v2.h で定義されています。

このスキルを使う場面

このスキルは以下の場合に使用します：

• AT_DISPATCH_*マクロをAT_DISPATCH_V2に変換する
• ATenカーネルを新しいディスパッチAPIに移植する
• aten/src/ATen/native/ のファイルでディスパッチマクロを使用している場合
• ユーザーが「AT_DISPATCH」「dispatch v2」「Dispatch_v2.h」またはマクロ変換について言及する場合

クイックリファレンス

旧形式：

AT_DISPATCH_ALL_TYPES_AND3(kBFloat16, kHalf, kBool, dtype, "kernel_name", [&]() {
  // ラムダの本体
});

新形式：

AT_DISPATCH_V2(dtype, "kernel_name", AT_WRAP([&]() {
  // ラムダの本体
}), AT_EXPAND

...