Skip to content

Commit 5cdaaa5

Browse files
ArvindSridharnarendasan
authored andcommitted
Minor changes
Signed-off-by: Arvind Sridhar <[email protected]>
1 parent 318520d commit 5cdaaa5

File tree

3 files changed

+1
-3
lines changed

3 files changed

+1
-3
lines changed

core/compiler.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ void AddEngineToGraph(
119119
}
120120

121121
bool CheckMethodOperatorSupport(const torch::jit::script::Module& mod, std::string method_name) {
122-
// Go through Lowering to simplify graph and extract weight parameters
122+
// Go through Lowering to simplify graph
123123
auto graph_and_parameters = lowering::Lower(mod, method_name, lowering::LowerInfo());
124124

125125
auto g = graph_and_parameters.first;

core/lowering/passes/module_fallback.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@ void NotateModuleForFallback(const torch::jit::Module& mod, std::string mod_name
4141
auto user = u.user;
4242
auto delim_start_n = g->create(torch::jit::prim::Enter, 0);
4343
delim_start_n->s_(c10::Symbol::attr("compilation_edge"), "start");
44-
auto num_end_outs = user->outputs().size();
4544
auto delim_end_n = g->create(torch::jit::prim::Exit, 0);
4645
delim_end_n->s_(c10::Symbol::attr("compilation_edge"), "end");
4746
delim_start_n->insertBefore(user);

core/lowering/register_trt_placeholder_ops.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
#include "torch/csrc/jit/runtime/custom_operator.h"
2-
#include "torch/library.h"
32

43
namespace torch {
54
namespace jit {

0 commit comments

Comments
 (0)