mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
2c5bf12584
This reverts commit d23ecbfc9ac157560611b242f015743f189dbf48. Reverted https://github.com/pytorch/pytorch/pull/73434 on behalf of https://github.com/albanD
550 lines
33 KiB
C++
550 lines
33 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 <caffe2/serialize/versions.h>
|
|
#include <torch/csrc/jit/mobile/upgrader_mobile.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})
|
|
})},
|
|
{std::string("aten::gelu"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 9, "gelu_0_9", 5})
|
|
})},
|
|
{std::string("aten::gelu.out"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 9, "gelu_out_0_9", 6})
|
|
})},
|
|
{std::string("aten::linspace"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 7, "linspace_0_7", 7})
|
|
})},
|
|
{std::string("aten::linspace.out"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 7, "linspace_out_0_7", 8})
|
|
})},
|
|
{std::string("aten::logspace"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 8, "logspace_0_8", 9})
|
|
})},
|
|
{std::string("aten::logspace.out"),
|
|
std::vector<Upgrader>({
|
|
Upgrader({0, 8, "logspace_out_0_8", 10})
|
|
})},
|
|
});
|
|
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
|
|
}),
|
|
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 jit
|
|
} // namespace torch
|