mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
e9aa6898ab
Test Plan: revert-hammer
Differential Revision:
D23802296 (d2b045030e)
Original commit changeset: 20c9798a414e
fbshipit-source-id: a28d56039ca404fe94ed7572f1febd1673e3e788
51 lines
1.5 KiB
C++
51 lines
1.5 KiB
C++
#include <torch/csrc/jit/frontend/ir_emitter.h>
|
|
#include <torch/csrc/jit/ir/ir.h>
|
|
#include <torch/csrc/jit/ir/irparser.h>
|
|
#include <torch/csrc/jit/testing/file_check.h>
|
|
#include "test/cpp/jit/test_base.h"
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
|
|
void testCleanUpPasses() {
|
|
// Tests stability of clean up passes when dealing with constant pooling
|
|
// and constant propagation.
|
|
{
|
|
auto graph = std::make_shared<Graph>();
|
|
parseIR(
|
|
R"IR(
|
|
graph(%cond.1 : Tensor,
|
|
%suffix.1 : str):
|
|
%3 : bool = aten::Bool(%cond.1) # o.py:6:7
|
|
%25 : str = prim::If(%3) # o.py:6:4
|
|
block0():
|
|
%a.1 : str = prim::Constant[value="same string"]()
|
|
%b.1 : str = prim::Constant[value=" with a twist"]()
|
|
%7 : str = aten::add(%a.1, %b.1)
|
|
%11 : str = aten::add(%7, %suffix.1) # o.py:10:15
|
|
-> (%11)
|
|
block1():
|
|
%c.1 : str = prim::Constant[value="same string"]()
|
|
%d.1 : str = prim::Constant[value=" with a twist"]()
|
|
%12 : str = aten::add(%c.1, %d.1)
|
|
-> (%12)
|
|
return (%25)
|
|
)IR",
|
|
&*graph);
|
|
runCleanupPasses(graph);
|
|
testing::FileCheck()
|
|
.check_count(
|
|
"prim::Constant[value=\"same string with a twist\"]",
|
|
1,
|
|
/*exactly=*/true)
|
|
->run(*graph);
|
|
|
|
auto graph_after_pass_once = graph->toString();
|
|
runCleanupPasses(graph);
|
|
auto graph_after_pass_twice = graph->toString();
|
|
ASSERT_EQ(graph_after_pass_once, graph_after_pass_twice);
|
|
}
|
|
}
|
|
} // namespace jit
|
|
} // namespace torch
|