OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
master
pytorch/torch/csrc/jit/mobile/upgrader_mobile.cpp

687 lines
41 KiB
C++
Raw Permalink Blame History

/**
* @generated
* This is an auto-generated file. Please do not modify it by hand.
* To re-generate, please run:
* cd ~/pytorch && python torchgen/operator_versions/gen_mobile_upgraders.py
*/
#include <caffe2/serialize/versions.h>
#include <torch/csrc/jit/mobile/upgrader_mobile.h>
namespace c10 {
TypePtr parseType(const std::string& pythonStr);
} // namespace c10
namespace torch::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.Scalar_mode"),
std::vector<Upgrader>({
Upgrader({0, 3, "div_Scalar_mode_0_3", 1})
})},
{std::string("aten::div.Tensor"),
std::vector<Upgrader>({
Upgrader({0, 3, "div_Tensor_0_3", 2})
})},
{std::string("aten::div.Tensor_mode"),
std::vector<Upgrader>({
Upgrader({0, 3, "div_Tensor_mode_0_3", 3})
})},
{std::string("aten::div.out"),
std::vector<Upgrader>({
Upgrader({0, 3, "div_out_0_3", 8})
})},
{std::string("aten::div.out_mode"),
std::vector<Upgrader>({
Upgrader({0, 3, "div_out_mode_0_3", 9})
})},
{std::string("aten::div_.Scalar"),
std::vector<Upgrader>({
Upgrader({0, 3, "div__Scalar_0_3", 4})
})},
{std::string("aten::div_.Scalar_mode"),
std::vector<Upgrader>({
Upgrader({0, 3, "div__Scalar_mode_0_3", 5})
})},
{std::string("aten::div_.Tensor"),
std::vector<Upgrader>({
Upgrader({0, 3, "div__Tensor_0_3", 6})
})},
{std::string("aten::div_.Tensor_mode"),
std::vector<Upgrader>({
Upgrader({0, 3, "div__Tensor_mode_0_3", 7})
})},
{std::string("aten::gelu"),
std::vector<Upgrader>({
Upgrader({0, 9, "gelu_0_9", 11})
})},
{std::string("aten::gelu.out"),
std::vector<Upgrader>({
Upgrader({0, 9, "gelu_out_0_9", 12})
})},
{std::string("aten::linspace"),
std::vector<Upgrader>({
Upgrader({0, 7, "linspace_0_7", 13})
})},
{std::string("aten::linspace.out"),
std::vector<Upgrader>({
Upgrader({0, 7, "linspace_out_0_7", 14})
})},
{std::string("aten::logspace"),
std::vector<Upgrader>({
Upgrader({0, 8, "logspace_0_8", 15})
})},
{std::string("aten::logspace.out"),
std::vector<Upgrader>({
Upgrader({0, 8, "logspace_out_0_8", 16})
})},
});
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_Scalar_mode_0_3",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 3},
Instruction{OpCode::MOVE, 1, 0},
Instruction{OpCode::MOVE, 2, 0},
Instruction{OpCode::MOVE, 3, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>(), // constants list,
std::vector<c10::TypePtr>(), // types list,
3
),
std::vector<OperatorString>({
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_Tensor_mode_0_3",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 3},
Instruction{OpCode::MOVE, 1, 0},
Instruction{OpCode::MOVE, 2, 0},
Instruction{OpCode::MOVE, 3, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>(), // constants list,
std::vector<c10::TypePtr>(), // types list,
3
),
std::vector<OperatorString>({
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__Scalar_mode_0_3",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 3},
Instruction{OpCode::MOVE, 1, 0},
Instruction{OpCode::MOVE, 2, 0},
Instruction{OpCode::MOVE, 3, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>(), // constants list,
std::vector<c10::TypePtr>(), // types list,
3
),
std::vector<OperatorString>({
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__Tensor_mode_0_3",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 3},
Instruction{OpCode::MOVE, 1, 0},
Instruction{OpCode::MOVE, 2, 0},
Instruction{OpCode::MOVE, 3, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>(), // constants list,
std::vector<c10::TypePtr>(), // types list,
3
),
std::vector<OperatorString>({
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
}),
ByteCodeFunctionWithOperator({
mobile::Function::registerFunc(
"div_out_mode_0_3",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 4},
Instruction{OpCode::MOVE, 1, 0},
Instruction{OpCode::MOVE, 2, 0},
Instruction{OpCode::MOVE, 3, 0},
Instruction{OpCode::MOVE, 4, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>(), // constants list,
std::vector<c10::TypePtr>(), // types list,
4
),
std::vector<OperatorString>({
OperatorString({"aten::div", "out_mode", 4}),
}), // operators list
}),
ByteCodeFunctionWithOperator({
mobile::Function::registerFunc(
"full_names_0_4",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 7},
Instruction{OpCode::MOVE, 1, 0},
Instruction{OpCode::MOVE, 2, 0},
Instruction{OpCode::MOVE, 3, 0},
Instruction{OpCode::MOVE, 4, 0},
Instruction{OpCode::MOVE, 5, 0},
Instruction{OpCode::MOVE, 6, 0},
Instruction{OpCode::MOVE, 7, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>(), // constants list,
std::vector<c10::TypePtr>(), // types list,
7
),
std::vector<OperatorString>({
OperatorString({"aten::full", "names", 7}),
}), // operators list
}),
ByteCodeFunctionWithOperator({
mobile::Function::registerFunc(
"gelu_0_9",
std::vector<Instruction>({
Instruction{OpCode::STORE, 1, 0},
Instruction{OpCode::MOVE, 1, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>({
c10::IValue("none"),
}), // constants list,
std::vector<c10::TypePtr>(), // types list,
1
),
std::vector<OperatorString>({
OperatorString({"aten::gelu", "", 1}),
}), // operators list
}),
ByteCodeFunctionWithOperator({
mobile::Function::registerFunc(
"gelu_out_0_9",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 2},
Instruction{OpCode::MOVE, 1, 0},
Instruction{OpCode::MOVE, 2, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>({
c10::IValue("none"),
}), // constants list,
std::vector<c10::TypePtr>(), // types list,
2
),
std::vector<OperatorString>({
OperatorString({"aten::gelu", "out", 2}),
}), // operators list
}),
ByteCodeFunctionWithOperator({
mobile::Function::registerFunc(
"linspace_0_7",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 7},
Instruction{OpCode::LOAD, 3, 0},
Instruction{OpCode::LOADC, 0, 0},
Instruction{OpCode::__IS__, 0, 0},
Instruction{OpCode::JF, 10, 0},
Instruction{OpCode::LOAD, 1, 0},
Instruction{OpCode::LOAD, 2, 0},
Instruction{OpCode::LOADC, 1, 0},
Instruction{OpCode::LOAD, 4, 0},
Instruction{OpCode::LOAD, 5, 0},
Instruction{OpCode::LOAD, 6, 0},
Instruction{OpCode::LOAD, 7, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::JMP, 10, 0},
Instruction{OpCode::LOAD, 1, 0},
Instruction{OpCode::LOAD, 2, 0},
Instruction{OpCode::LOAD, 3, 0},
Instruction{OpCode::OP, 1, 0},
Instruction{OpCode::LOAD, 4, 0},
Instruction{OpCode::LOAD, 5, 0},
Instruction{OpCode::LOAD, 6, 0},
Instruction{OpCode::LOAD, 7, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::STORE, 8, 0},
Instruction{OpCode::DROPR, 7, 0},
Instruction{OpCode::DROPR, 6, 0},
Instruction{OpCode::DROPR, 5, 0},
Instruction{OpCode::DROPR, 4, 0},
Instruction{OpCode::DROPR, 2, 0},
Instruction{OpCode::DROPR, 1, 0},
Instruction{OpCode::DROPR, 3, 0},
Instruction{OpCode::MOVE, 8, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>({
c10::IValue(),
c10::IValue(100),
}), // constants list,
std::vector<c10::TypePtr>(), // types list,
8
),
std::vector<OperatorString>({
OperatorString({"aten::linspace", "", 7}),
OperatorString({"prim::unchecked_cast", "", 1}),
}), // operators list
}),
ByteCodeFunctionWithOperator({
mobile::Function::registerFunc(
"linspace_out_0_7",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 4},
Instruction{OpCode::LOAD, 3, 0},
Instruction{OpCode::LOADC, 0, 0},
Instruction{OpCode::__IS__, 0, 0},
Instruction{OpCode::JF, 7, 0},
Instruction{OpCode::LOAD, 1, 0},
Instruction{OpCode::LOAD, 2, 0},
Instruction{OpCode::LOADC, 1, 0},
Instruction{OpCode::LOAD, 4, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::JMP, 7, 0},
Instruction{OpCode::LOAD, 1, 0},
Instruction{OpCode::LOAD, 2, 0},
Instruction{OpCode::LOAD, 3, 0},
Instruction{OpCode::OP, 1, 0},
Instruction{OpCode::LOAD, 4, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::STORE, 5, 0},
Instruction{OpCode::DROPR, 4, 0},
Instruction{OpCode::DROPR, 2, 0},
Instruction{OpCode::DROPR, 1, 0},
Instruction{OpCode::DROPR, 3, 0},
Instruction{OpCode::MOVE, 5, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>({
c10::IValue(),
c10::IValue(100),
}), // constants list,
std::vector<c10::TypePtr>(), // types list,
5
),
std::vector<OperatorString>({
OperatorString({"aten::linspace", "out", 4}),
OperatorString({"prim::unchecked_cast", "", 1}),
}), // operators list
}),
ByteCodeFunctionWithOperator({
mobile::Function::registerFunc(
"logspace_0_8",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 8},
Instruction{OpCode::LOAD, 3, 0},
Instruction{OpCode::LOADC, 0, 0},
Instruction{OpCode::__IS__, 0, 0},
Instruction{OpCode::JF, 11, 0},
Instruction{OpCode::LOAD, 1, 0},
Instruction{OpCode::LOAD, 2, 0},
Instruction{OpCode::LOADC, 1, 0},
Instruction{OpCode::LOAD, 4, 0},
Instruction{OpCode::LOAD, 5, 0},
Instruction{OpCode::LOAD, 6, 0},
Instruction{OpCode::LOAD, 7, 0},
Instruction{OpCode::LOAD, 8, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::JMP, 11, 0},
Instruction{OpCode::LOAD, 1, 0},
Instruction{OpCode::LOAD, 2, 0},
Instruction{OpCode::LOAD, 3, 0},
Instruction{OpCode::OP, 1, 0},
Instruction{OpCode::LOAD, 4, 0},
Instruction{OpCode::LOAD, 5, 0},
Instruction{OpCode::LOAD, 6, 0},
Instruction{OpCode::LOAD, 7, 0},
Instruction{OpCode::LOAD, 8, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::STORE, 9, 0},
Instruction{OpCode::DROPR, 8, 0},
Instruction{OpCode::DROPR, 7, 0},
Instruction{OpCode::DROPR, 6, 0},
Instruction{OpCode::DROPR, 5, 0},
Instruction{OpCode::DROPR, 4, 0},
Instruction{OpCode::DROPR, 2, 0},
Instruction{OpCode::DROPR, 1, 0},
Instruction{OpCode::DROPR, 3, 0},
Instruction{OpCode::MOVE, 9, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>({
c10::IValue(),
c10::IValue(100),
}), // constants list,
std::vector<c10::TypePtr>(), // types list,
9
),
std::vector<OperatorString>({
OperatorString({"aten::logspace", "", 8}),
OperatorString({"prim::unchecked_cast", "", 1}),
}), // operators list
}),
ByteCodeFunctionWithOperator({
mobile::Function::registerFunc(
"logspace_out_0_8",
std::vector<Instruction>({
Instruction{OpCode::STOREN, 1, 5},
Instruction{OpCode::LOAD, 3, 0},
Instruction{OpCode::LOADC, 0, 0},
Instruction{OpCode::__IS__, 0, 0},
Instruction{OpCode::JF, 8, 0},
Instruction{OpCode::LOAD, 1, 0},
Instruction{OpCode::LOAD, 2, 0},
Instruction{OpCode::LOADC, 1, 0},
Instruction{OpCode::LOAD, 4, 0},
Instruction{OpCode::LOAD, 5, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::JMP, 8, 0},
Instruction{OpCode::LOAD, 1, 0},
Instruction{OpCode::LOAD, 2, 0},
Instruction{OpCode::LOAD, 3, 0},
Instruction{OpCode::OP, 1, 0},
Instruction{OpCode::LOAD, 4, 0},
Instruction{OpCode::LOAD, 5, 0},
Instruction{OpCode::OP, 0, 0},
Instruction{OpCode::STORE, 6, 0},
Instruction{OpCode::DROPR, 5, 0},
Instruction{OpCode::DROPR, 4, 0},
Instruction{OpCode::DROPR, 2, 0},
Instruction{OpCode::DROPR, 1, 0},
Instruction{OpCode::DROPR, 3, 0},
Instruction{OpCode::MOVE, 6, 0},
Instruction{OpCode::RET, 0, 0},
}), // instructions list,
std::vector<c10::IValue>({
c10::IValue(),
c10::IValue(100),
}), // constants list,
std::vector<c10::TypePtr>(), // types list,
6
),
std::vector<OperatorString>({
OperatorString({"aten::logspace", "out", 5}),
OperatorString({"prim::unchecked_cast", "", 1}),
}), // 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);
}
}
return upgrader_function_list;
};
static std::vector<ByteCodeFunctionWithOperator> upgraderBytecodeList =
generate_upgrader_bytecode_list();
return upgraderBytecodeList;
}
// clang-format on
} // namespace torch::jit
Reference in New Issue View Git Blame Copy Permalink