@@ -23866,13 +23866,13 @@ TEST_F(NVFuserTest, FusionMaxRootDomainInfoSpanningTreePrintTwice_CUDA) {
23866
23866
std::stringstream ss;
23867
23867
virtual void propagateTvPasC(TensorView* from, TensorView* to) override {
23868
23868
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;
23871
23871
}
23872
23872
virtual void propagateTvCasP(TensorView* from, TensorView* to) override {
23873
23873
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;
23876
23876
}
23877
23877
} printer1, printer2;
23878
23878
printer1.ss << std::endl;
@@ -23884,11 +23884,11 @@ TEST_F(NVFuserTest, FusionMaxRootDomainInfoSpanningTreePrintTwice_CUDA) {
23884
23884
23885
23885
auto expect = R"ESCAPE(
23886
23886
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
23889
23889
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
23892
23892
)ESCAPE";
23893
23893
TORCH_CHECK(printer1.ss.str() == expect);
23894
23894
TORCH_CHECK(printer2.ss.str() == expect);
0 commit comments