Register index_put and support new cases

Reviewed By: kunmingho Differential Revision: D48245280
pytorch · Nov 9, 2023 · 415c021 · 415c021
1 parent 0f1ab96
commit 415c021
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/ClassGen/InstrGen.cpp b/tools/ClassGen/InstrGen.cpp
@@ -1182,6 +1182,7 @@ int main(int argc, char **argv) {
       .addOperand("Values", OperandKind::In)
       .addMember(MemberType::Boolean, "Accumulate")
       .addMember(MemberType::Int64, "NumIndices")
+      .addMember(MemberType::Int64, "StartDim")
       .inplaceOperand({"Dest", "Input"})
       .autoVerify(VerifyKind::NoVerify);
   //===--------------------------------------------------------------------===//