mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
c5df294940
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71578 Use more robust way of extracting upgrader min and max versions Test Plan: omgitsgreen Reviewed By: cccclai Differential Revision: D33690113 fbshipit-source-id: 79a964acb26d7ca1354e104710a285b8da3f46d1 (cherry picked from commit 9e316ee5c12e7bce9b17edebec2eeb38ecabd336)
302 lines
18 KiB
C++
302 lines
18 KiB
C++
/**
|
|
* @generated
|
|
* This is an auto-generated file. Please do not modify it by hand.
|
|
* To re-generate, please run:
|
|
* cd ~/pytorch && python torch/csrc/jit/mobile/upgrader_mobile.cpp
|
|
*/
|
|
|
|
#include <torch/csrc/jit/mobile/upgrader_mobile.h>
|
|
|
|
#include <ATen/core/ivalue.h>
|
|
#include <caffe2/serialize/versions.h>
|
|
#include <torch/csrc/jit/mobile/type_parser.h>
|
|
|
|
namespace c10 {
|
|
TypePtr parseType(const std::string& pythonStr);
|
|
} // namespace c10
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
|
|
// clang-format off
|
|
|
|
// From operator_versions_map
|
|
|
|
const std::unordered_map<std::string, std::vector<Upgrader>>
|
|
getOperatorVersionMapForMobile() {
|
|
static std::unordered_map<std::string, std::vector<Upgrader>>
|
|
operatorVersionMapForMobile({
|
|
{std::string("aten::div.Scalar"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 3, "div_Scalar_0_3", 0})
|
|
})},
|
|
{std::string("aten::div.Tensor"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 3, "div_Tensor_0_3", 1})
|
|
})},
|
|
{std::string("aten::div.out"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 3, "div_out_0_3", 4})
|
|
})},
|
|
{std::string("aten::div_.Scalar"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 3, "div__Scalar_0_3", 2})
|
|
})},
|
|
{std::string("aten::div_.Tensor"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 3, "div__Tensor_0_3", 3})
|
|
})},
|
|
});
|
|
return operatorVersionMapForMobile;
|
|
}
|
|
|
|
const std::vector<ByteCodeFunctionWithOperator>& getUpgraderBytecodeList() {
|
|
auto generate_upgrader_bytecode_list = []() {
|
|
std::vector<ByteCodeFunctionWithOperator> upgrader_function_list({
|
|
ByteCodeFunctionWithOperator({
|
|
mobile::Function::registerFunc(
|
|
"div_Scalar_0_3",
|
|
std::vector<Instruction>({
|
|
Instruction{OpCode::STOREN, 1, 2},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::OP, 0, 0},
|
|
Instruction{OpCode::JF, 3, 0},
|
|
Instruction{OpCode::LOADC, 1, 0},
|
|
Instruction{OpCode::JMP, 3, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::ISINSTANCE, 0, 1},
|
|
Instruction{OpCode::STORE, 3, 0},
|
|
Instruction{OpCode::MOVE, 3, 0},
|
|
Instruction{OpCode::JF, 5, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::OP, 1, 0},
|
|
Instruction{OpCode::JMP, 6, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::OP, 2, 0},
|
|
Instruction{OpCode::LOADC, 0, 0},
|
|
Instruction{OpCode::OP, 3, 0},
|
|
Instruction{OpCode::STORE, 4, 0},
|
|
Instruction{OpCode::DROPR, 2, 0},
|
|
Instruction{OpCode::DROPR, 1, 0},
|
|
Instruction{OpCode::MOVE, 4, 0},
|
|
Instruction{OpCode::RET, 0, 0},
|
|
}), // instructions list,
|
|
std::vector<c10::IValue>({
|
|
c10::IValue("trunc"),
|
|
c10::IValue(true),
|
|
}), // constants list,
|
|
std::vector<c10::TypePtr>({
|
|
c10::parseType("float"),
|
|
}), // types list,
|
|
4
|
|
),
|
|
std::vector<OperatorString>({
|
|
OperatorString({"aten::is_floating_point", "", 1}),
|
|
OperatorString({"aten::div", "Scalar", 2}),
|
|
OperatorString({"prim::unchecked_cast", "", 1}),
|
|
OperatorString({"aten::div", "Scalar_mode", 3}),
|
|
}), // operators list
|
|
}),
|
|
ByteCodeFunctionWithOperator({
|
|
mobile::Function::registerFunc(
|
|
"div_Tensor_0_3",
|
|
std::vector<Instruction>({
|
|
Instruction{OpCode::STOREN, 1, 2},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::OP, 0, 0},
|
|
Instruction{OpCode::JF, 3, 0},
|
|
Instruction{OpCode::LOADC, 1, 0},
|
|
Instruction{OpCode::JMP, 3, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::OP, 0, 0},
|
|
Instruction{OpCode::STORE, 3, 0},
|
|
Instruction{OpCode::MOVE, 3, 0},
|
|
Instruction{OpCode::JF, 5, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::OP, 1, 0},
|
|
Instruction{OpCode::JMP, 5, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::LOADC, 0, 0},
|
|
Instruction{OpCode::OP, 2, 0},
|
|
Instruction{OpCode::STORE, 4, 0},
|
|
Instruction{OpCode::DROPR, 2, 0},
|
|
Instruction{OpCode::DROPR, 1, 0},
|
|
Instruction{OpCode::MOVE, 4, 0},
|
|
Instruction{OpCode::RET, 0, 0},
|
|
}), // instructions list,
|
|
std::vector<c10::IValue>({
|
|
c10::IValue("trunc"),
|
|
c10::IValue(true),
|
|
}), // constants list,
|
|
std::vector<c10::TypePtr>(), // types list,
|
|
4
|
|
),
|
|
std::vector<OperatorString>({
|
|
OperatorString({"aten::is_floating_point", "", 1}),
|
|
OperatorString({"aten::div", "Tensor", 2}),
|
|
OperatorString({"aten::div", "Tensor_mode", 3}),
|
|
}), // operators list
|
|
}),
|
|
ByteCodeFunctionWithOperator({
|
|
mobile::Function::registerFunc(
|
|
"div__Scalar_0_3",
|
|
std::vector<Instruction>({
|
|
Instruction{OpCode::STOREN, 1, 2},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::OP, 0, 0},
|
|
Instruction{OpCode::JF, 3, 0},
|
|
Instruction{OpCode::LOADC, 1, 0},
|
|
Instruction{OpCode::JMP, 3, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::ISINSTANCE, 0, 1},
|
|
Instruction{OpCode::STORE, 3, 0},
|
|
Instruction{OpCode::MOVE, 3, 0},
|
|
Instruction{OpCode::JF, 5, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::OP, 1, 0},
|
|
Instruction{OpCode::JMP, 6, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::OP, 2, 0},
|
|
Instruction{OpCode::LOADC, 0, 0},
|
|
Instruction{OpCode::OP, 3, 0},
|
|
Instruction{OpCode::STORE, 4, 0},
|
|
Instruction{OpCode::DROPR, 2, 0},
|
|
Instruction{OpCode::DROPR, 1, 0},
|
|
Instruction{OpCode::MOVE, 4, 0},
|
|
Instruction{OpCode::RET, 0, 0},
|
|
}), // instructions list,
|
|
std::vector<c10::IValue>({
|
|
c10::IValue("trunc"),
|
|
c10::IValue(true),
|
|
}), // constants list,
|
|
std::vector<c10::TypePtr>({
|
|
c10::parseType("float"),
|
|
}), // types list,
|
|
4
|
|
),
|
|
std::vector<OperatorString>({
|
|
OperatorString({"aten::is_floating_point", "", 1}),
|
|
OperatorString({"aten::div_", "Scalar", 2}),
|
|
OperatorString({"prim::unchecked_cast", "", 1}),
|
|
OperatorString({"aten::div_", "Scalar_mode", 3}),
|
|
}), // operators list
|
|
}),
|
|
ByteCodeFunctionWithOperator({
|
|
mobile::Function::registerFunc(
|
|
"div__Tensor_0_3",
|
|
std::vector<Instruction>({
|
|
Instruction{OpCode::STOREN, 1, 2},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::OP, 0, 0},
|
|
Instruction{OpCode::JF, 3, 0},
|
|
Instruction{OpCode::LOADC, 1, 0},
|
|
Instruction{OpCode::JMP, 3, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::OP, 0, 0},
|
|
Instruction{OpCode::STORE, 3, 0},
|
|
Instruction{OpCode::MOVE, 3, 0},
|
|
Instruction{OpCode::JF, 5, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::OP, 1, 0},
|
|
Instruction{OpCode::JMP, 5, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::LOADC, 0, 0},
|
|
Instruction{OpCode::OP, 2, 0},
|
|
Instruction{OpCode::STORE, 4, 0},
|
|
Instruction{OpCode::DROPR, 2, 0},
|
|
Instruction{OpCode::DROPR, 1, 0},
|
|
Instruction{OpCode::MOVE, 4, 0},
|
|
Instruction{OpCode::RET, 0, 0},
|
|
}), // instructions list,
|
|
std::vector<c10::IValue>({
|
|
c10::IValue("trunc"),
|
|
c10::IValue(true),
|
|
}), // constants list,
|
|
std::vector<c10::TypePtr>(), // types list,
|
|
4
|
|
),
|
|
std::vector<OperatorString>({
|
|
OperatorString({"aten::is_floating_point", "", 1}),
|
|
OperatorString({"aten::div_", "Tensor", 2}),
|
|
OperatorString({"aten::div_", "Tensor_mode", 3}),
|
|
}), // operators list
|
|
}),
|
|
ByteCodeFunctionWithOperator({
|
|
mobile::Function::registerFunc(
|
|
"div_out_0_3",
|
|
std::vector<Instruction>({
|
|
Instruction{OpCode::STOREN, 1, 3},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::OP, 0, 0},
|
|
Instruction{OpCode::JF, 3, 0},
|
|
Instruction{OpCode::LOADC, 1, 0},
|
|
Instruction{OpCode::JMP, 3, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::OP, 0, 0},
|
|
Instruction{OpCode::JF, 3, 0},
|
|
Instruction{OpCode::LOADC, 1, 0},
|
|
Instruction{OpCode::JMP, 3, 0},
|
|
Instruction{OpCode::LOAD, 3, 0},
|
|
Instruction{OpCode::OP, 0, 0},
|
|
Instruction{OpCode::STORE, 4, 0},
|
|
Instruction{OpCode::MOVE, 4, 0},
|
|
Instruction{OpCode::JF, 6, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::LOAD, 3, 0},
|
|
Instruction{OpCode::OP, 1, 0},
|
|
Instruction{OpCode::JMP, 6, 0},
|
|
Instruction{OpCode::LOAD, 1, 0},
|
|
Instruction{OpCode::LOAD, 2, 0},
|
|
Instruction{OpCode::LOADC, 0, 0},
|
|
Instruction{OpCode::LOAD, 3, 0},
|
|
Instruction{OpCode::OP, 2, 0},
|
|
Instruction{OpCode::STORE, 5, 0},
|
|
Instruction{OpCode::DROPR, 3, 0},
|
|
Instruction{OpCode::DROPR, 2, 0},
|
|
Instruction{OpCode::DROPR, 1, 0},
|
|
Instruction{OpCode::MOVE, 5, 0},
|
|
Instruction{OpCode::RET, 0, 0},
|
|
}), // instructions list,
|
|
std::vector<c10::IValue>({
|
|
c10::IValue("trunc"),
|
|
c10::IValue(true),
|
|
}), // constants list,
|
|
std::vector<c10::TypePtr>(), // types list,
|
|
5
|
|
),
|
|
std::vector<OperatorString>({
|
|
OperatorString({"aten::is_floating_point", "", 1}),
|
|
OperatorString({"aten::div", "out", 3}),
|
|
OperatorString({"aten::div", "out_mode", 4}),
|
|
}), // operators list
|
|
}),
|
|
});
|
|
for (const auto& upgrader_function : upgrader_function_list) {
|
|
for (const auto& op : upgrader_function.operators) {
|
|
upgrader_function.function.append_operator(
|
|
op.name,
|
|
op.overload_name,
|
|
op.num_specified_args,
|
|
caffe2::serialize::kMaxSupportedFileFormatVersion);
|
|
}
|
|
}
|
|
return upgrader_function_list;
|
|
};
|
|
static std::vector<ByteCodeFunctionWithOperator> upgraderBytecodeList =
|
|
generate_upgrader_bytecode_list();
|
|
return upgraderBytecodeList;
|
|
}
|
|
|
|
// clang-format on
|
|
|
|
} // namespace jit
|
|
} // namespace torch
|