frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert D18350353: dump operator names of a module and its sub-modules.

Browse Source 
Test Plan: revert-hammer

Differential Revision:
D18350353

Original commit changeset: 2026c8ab7650

fbshipit-source-id: 401f34cb276c3ea34a5439de4c3415969a04ab2a
This commit is contained in:
Pieter Noordhuis
2019-11-07 05:26:59 -08:00
committed by Facebook Github Bot
parent 58005382c8
commit 78a34d3205
2 changed files with 0 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
binaries/CMakeLists.txt
View File

@ -97,5 +97,3 @@ endif()
# ---[ tutorials
caffe2_binary_target("tutorial_blob.cc")
caffe2_binary_target("dump_operator_names.cc")

82
binaries/dump_operator_names.cc
View File

@ -1,82 +0,0 @@
/**
* Copyright (c) 2016-present, Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <torch/csrc/jit/script/module.h>
#include <torch/csrc/jit/mobile/module.h>
#include <torch/csrc/jit/import.h>
#include <torch/csrc/jit/instruction.h>
#include <c10/util/Flags.h>
#include <fstream>
namespace torch {
namespace jit {
void dump_opnames(const script::Module& m, std::unordered_set<std::string>& opnames) {
auto methods = m.get_methods();
for (const auto& method : methods) {
const auto& func = method.function();
std::cout << "function name: " << func.name() << std::endl;
torch::jit::Code code(func.graph());
for (size_t i = 0; i < code.instructions().size(); ++i) {
auto ins = code.instructions()[i];
auto node = code.instructions_source()[i];
if (ins.op == OpCode::OP) {
auto opname = node->schema().operator_name();
std::string namestr = opname.name;
if (!opname.overload_name.empty())
namestr += "." + opname.overload_name;
std::cout << " " << namestr << std::endl;
opnames.emplace(namestr);
}
}
}
auto modules = m.get_modules();
for (const auto& sub_m : modules) {
std::cout << "sub module name: " << sub_m.name << std::endl;
dump_opnames(sub_m.module, opnames);
}
}
}
}
C10_DEFINE_string(model, "", "The given torch script model.");
C10_DEFINE_string(output, "", "The output yaml file of operator list.");
int main(int argc, char** argv) {
c10::SetUsageMessage(
"Dump operators in a script module and its sub modules.\n"
"Example usage:\n"
"./dump_operator_names"
" --model=<model_file>"
" --output=<output.yaml>");
if (!c10::ParseCommandLineFlags(&argc, &argv)) {
std::cerr << "Failed to parse command line flags!" << std::endl;
return 1;
}
CAFFE_ENFORCE_GE(FLAGS_model.size(), 0, "Model file must be specified.");
CAFFE_ENFORCE_GE(FLAGS_output.size(), 0, "Output yaml file must be specified.");
auto m = torch::jit::load(FLAGS_model);
std::unordered_set<std::string> opnames;
torch::jit::dump_opnames(m, opnames);
std::ofstream ofile(FLAGS_output);
for (const auto& name : opnames) {
ofile << "- " << name << std::endl;
}
ofile.close();
}