File tree Expand file tree Collapse file tree 6 files changed +6
-6
lines changed Expand file tree Collapse file tree 6 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -48,7 +48,7 @@ void LowerGraph(std::shared_ptr<torch::jit::Graph>& g) {
48
48
// passes::UnpackBatchNorm(g);
49
49
passes::UnpackLogSoftmax (g);
50
50
passes::RemoveNOPs (g);
51
- passes::ElementWisePass (g);
51
+ passes::AliasOperators (g);
52
52
torch::jit::EliminateDeadCode (g);
53
53
LOG_GRAPH (*g);
54
54
}
Original file line number Diff line number Diff line change @@ -25,7 +25,7 @@ cc_library(
25
25
"unpack_addmm.cpp" ,
26
26
"unpack_batch_norm.cpp" ,
27
27
"unpack_log_softmax.cpp" ,
28
- "element_wise_pass .cpp"
28
+ "op_aliasing .cpp"
29
29
],
30
30
deps = [
31
31
"//core/util:prelude" ,
Original file line number Diff line number Diff line change @@ -7,7 +7,7 @@ namespace core {
7
7
namespace lowering {
8
8
namespace passes {
9
9
10
- void ElementWisePass (std::shared_ptr<torch::jit::Graph>& graph) {
10
+ void AliasOperators (std::shared_ptr<torch::jit::Graph>& graph) {
11
11
std::string true_divide_pattern = R"IR(
12
12
graph(%s, %o):
13
13
%1 : Tensor = aten::true_divide(%s, %o)
Original file line number Diff line number Diff line change @@ -19,7 +19,7 @@ void RemoveNOPs(std::shared_ptr<torch::jit::Graph> graph);
19
19
void UnpackAddMM (std::shared_ptr<torch::jit::Graph>& graph);
20
20
void UnpackBatchNorm (std::shared_ptr<torch::jit::Graph>& graph);
21
21
void UnpackLogSoftmax (std::shared_ptr<torch::jit::Graph>& graph);
22
- void ElementWisePass (std::shared_ptr<torch::jit::Graph>& graph);
22
+ void AliasOperators (std::shared_ptr<torch::jit::Graph>& graph);
23
23
24
24
} // namespace passes
25
25
} // namespace lowering
Original file line number Diff line number Diff line change @@ -20,7 +20,7 @@ lowering_test(
20
20
)
21
21
22
22
lowering_test (
23
- name = "test_element_wise_pass " ,
23
+ name = "test_operator_aliasing_pass " ,
24
24
)
25
25
26
26
test_suite (
@@ -29,6 +29,6 @@ test_suite(
29
29
":test_remove_contiguous_pass" ,
30
30
":test_remove_to" ,
31
31
":test_remove_detach_pass" ,
32
- ":test_element_wise_pass "
32
+ ":test_operator_aliasing_pass "
33
33
]
34
34
)
File renamed without changes.
You can’t perform that action at this time.
0 commit comments