Skip to content

Commit d3de227

Browse files
authored
Fix FusionMaxRootDomainInfoSpanningTreePrintTwice_CUDA (csarofeen#1781)
1 parent ecc7a87 commit d3de227

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

torch/csrc/jit/codegen/cuda/test/test_gpu.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -23866,13 +23866,13 @@ TEST_F(NVFuserTest, FusionMaxRootDomainInfoSpanningTreePrintTwice_CUDA) {
2386623866
std::stringstream ss;
2386723867
virtual void propagateTvPasC(TensorView* from, TensorView* to) override {
2386823868
ss << "propagateTvPasC" << std::endl;
23869-
ss << "from: " << from << std::endl;
23870-
ss << "to: " << to << std::endl;
23869+
ss << "from: " << from->name() << std::endl;
23870+
ss << "to: " << to->name() << std::endl;
2387123871
}
2387223872
virtual void propagateTvCasP(TensorView* from, TensorView* to) override {
2387323873
ss << "propagateTvCasP" << std::endl;
23874-
ss << "from: " << from << std::endl;
23875-
ss << "to: " << to << std::endl;
23874+
ss << "from: " << from->name() << std::endl;
23875+
ss << "to: " << to->name() << std::endl;
2387623876
}
2387723877
} printer1, printer2;
2387823878
printer1.ss << std::endl;
@@ -23884,11 +23884,11 @@ TEST_F(NVFuserTest, FusionMaxRootDomainInfoSpanningTreePrintTwice_CUDA) {
2388423884

2388523885
auto expect = R"ESCAPE(
2388623886
propagateTvPasC
23887-
from: T1_l[ rS8{( ceilDiv(i1, 10) )}, rS9{10}, iS4{i2}, iS5{i3} ]
23888-
to: T0_g[ iS0{i1}, iS1{i2}, iS2{i3} ]
23887+
from: 1
23888+
to: 0
2388923889
propagateTvCasP
23890-
from: T1_l[ rS8{( ceilDiv(i1, 10) )}, rS9{10}, iS4{i2}, iS5{i3} ]
23891-
to: T2_g[ iS6{i2}, iS7{i3} ]
23890+
from: 1
23891+
to: 2
2389223892
)ESCAPE";
2389323893
TORCH_CHECK(printer1.ss.str() == expect);
2389423894
TORCH_CHECK(printer2.ss.str() == expect);

0 commit comments

Comments
 (0)