Skip to content

Commit 7130431

Browse files
Sam-HornbySam-Hornby
committed
Adapt tests for new print tensor behaviour
Summary: Tests that rely on print tensor output need to change output format Reviewers: #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, alfiee, jackh Reviewed By: #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, alfiee Maniphest Tasks: T18640, T67122 Differential Revision: https://phabricator.sourcevertex.net/D73855
1 parent 369aaca commit 7130431

File tree

3 files changed

+309
-317
lines changed

3 files changed

+309
-317
lines changed

‎tensorflow/compiler/plugin/poplar/tests/pipeline_grouped_visitor_test.cc‎

Lines changed: 88 additions & 88 deletions
Original file line numberDiff line numberDiff line change
@@ -196,14 +196,14 @@ ENTRY main {
196196
engine.run(0);
197197
device.detach();
198198

199-
const std::string expected = R"(/print-tensor: {306,6012}
200-
/print-tensor: {306,6012}
201-
/print-tensor: {306,6012}
202-
/print-tensor: {306,6012}
203-
/print-tensor: {306,6012}
204-
/print-tensor: {306,6012}
205-
/print-tensor: {306,6012}
206-
/print-tensor: {306,6012}
199+
const std::string expected = R"(/print-tensor: [ 306.0000 6012.0000]
200+
/print-tensor: [ 306.0000 6012.0000]
201+
/print-tensor: [ 306.0000 6012.0000]
202+
/print-tensor: [ 306.0000 6012.0000]
203+
/print-tensor: [ 306.0000 6012.0000]
204+
/print-tensor: [ 306.0000 6012.0000]
205+
/print-tensor: [ 306.0000 6012.0000]
206+
/print-tensor: [ 306.0000 6012.0000]
207207
)";
208208

209209
ASSERT_EQ(expected, ss.str());
@@ -760,78 +760,78 @@ ENTRY main {
760760
engine.run(0);
761761
device.detach();
762762

763-
const std::string expected = R"(/print-tensor: 0
764-
/print-tensor.1: 1
765-
/print-tensor: 0
766-
/print-tensor.2: 2
767-
/print-tensor.1: 1
768-
/print-tensor: 0
769-
/print-tensor.3: 3
770-
/print-tensor.2: 2
771-
/print-tensor.1: 1
772-
/print-tensor: 0
773-
/print-tensor.4: 4
774-
/print-tensor.3: 3
775-
/print-tensor.2: 2
776-
/print-tensor.1: 1
777-
/print-tensor: 0
778-
/print-tensor.5: 5
779-
/print-tensor.4: 4
780-
/print-tensor.3: 3
781-
/print-tensor.2: 2
782-
/print-tensor.1: 1
783-
/print-tensor: 0
784-
/print-tensor.5: 5
785-
/print-tensor.4: 4
786-
/print-tensor.3: 3
787-
/print-tensor.2: 2
788-
/print-tensor.1: 1
789-
/print-tensor: 0
790-
/print-tensor.5: 5
791-
/print-tensor.4: 4
792-
/print-tensor.3: 3
793-
/print-tensor.2: 2
794-
/print-tensor.1: 1
795-
/print-tensor: 0
796-
/print-tensor.5: 5
797-
/print-tensor.4: 4
798-
/print-tensor.3: 3
799-
/print-tensor.2: 2
800-
/print-tensor.1: 1
801-
/print-tensor: 0
802-
/print-tensor.5: 5
803-
/print-tensor.4: 4
804-
/print-tensor.3: 3
805-
/print-tensor.2: 2
806-
/print-tensor.1: 1
807-
/print-tensor: 0
808-
/print-tensor.5: 5
809-
/print-tensor.4: 4
810-
/print-tensor.3: 3
811-
/print-tensor.2: 2
812-
/print-tensor.1: 1
813-
/print-tensor: 0
814-
/print-tensor.5: 5
815-
/print-tensor.4: 4
816-
/print-tensor.3: 3
817-
/print-tensor.2: 2
818-
/print-tensor.1: 1
819-
/print-tensor: 0
820-
/print-tensor.5: 5
821-
/print-tensor.4: 4
822-
/print-tensor.3: 3
823-
/print-tensor.2: 2
824-
/print-tensor.1: 1
825-
/print-tensor.5: 5
826-
/print-tensor.4: 4
827-
/print-tensor.3: 3
828-
/print-tensor.2: 2
829-
/print-tensor.5: 5
830-
/print-tensor.4: 4
831-
/print-tensor.3: 3
832-
/print-tensor.5: 5
833-
/print-tensor.4: 4
834-
/print-tensor.5: 5
763+
const std::string expected = R"(/print-tensor: 0.0000000
764+
/print-tensor.1: 1.0000000
765+
/print-tensor: 0.0000000
766+
/print-tensor.2: 2.0000000
767+
/print-tensor.1: 1.0000000
768+
/print-tensor: 0.0000000
769+
/print-tensor.3: 3.0000000
770+
/print-tensor.2: 2.0000000
771+
/print-tensor.1: 1.0000000
772+
/print-tensor: 0.0000000
773+
/print-tensor.4: 4.0000000
774+
/print-tensor.3: 3.0000000
775+
/print-tensor.2: 2.0000000
776+
/print-tensor.1: 1.0000000
777+
/print-tensor: 0.0000000
778+
/print-tensor.5: 5.0000000
779+
/print-tensor.4: 4.0000000
780+
/print-tensor.3: 3.0000000
781+
/print-tensor.2: 2.0000000
782+
/print-tensor.1: 1.0000000
783+
/print-tensor: 0.0000000
784+
/print-tensor.5: 5.0000000
785+
/print-tensor.4: 4.0000000
786+
/print-tensor.3: 3.0000000
787+
/print-tensor.2: 2.0000000
788+
/print-tensor.1: 1.0000000
789+
/print-tensor: 0.0000000
790+
/print-tensor.5: 5.0000000
791+
/print-tensor.4: 4.0000000
792+
/print-tensor.3: 3.0000000
793+
/print-tensor.2: 2.0000000
794+
/print-tensor.1: 1.0000000
795+
/print-tensor: 0.0000000
796+
/print-tensor.5: 5.0000000
797+
/print-tensor.4: 4.0000000
798+
/print-tensor.3: 3.0000000
799+
/print-tensor.2: 2.0000000
800+
/print-tensor.1: 1.0000000
801+
/print-tensor: 0.0000000
802+
/print-tensor.5: 5.0000000
803+
/print-tensor.4: 4.0000000
804+
/print-tensor.3: 3.0000000
805+
/print-tensor.2: 2.0000000
806+
/print-tensor.1: 1.0000000
807+
/print-tensor: 0.0000000
808+
/print-tensor.5: 5.0000000
809+
/print-tensor.4: 4.0000000
810+
/print-tensor.3: 3.0000000
811+
/print-tensor.2: 2.0000000
812+
/print-tensor.1: 1.0000000
813+
/print-tensor: 0.0000000
814+
/print-tensor.5: 5.0000000
815+
/print-tensor.4: 4.0000000
816+
/print-tensor.3: 3.0000000
817+
/print-tensor.2: 2.0000000
818+
/print-tensor.1: 1.0000000
819+
/print-tensor: 0.0000000
820+
/print-tensor.5: 5.0000000
821+
/print-tensor.4: 4.0000000
822+
/print-tensor.3: 3.0000000
823+
/print-tensor.2: 2.0000000
824+
/print-tensor.1: 1.0000000
825+
/print-tensor.5: 5.0000000
826+
/print-tensor.4: 4.0000000
827+
/print-tensor.3: 3.0000000
828+
/print-tensor.2: 2.0000000
829+
/print-tensor.5: 5.0000000
830+
/print-tensor.4: 4.0000000
831+
/print-tensor.3: 3.0000000
832+
/print-tensor.5: 5.0000000
833+
/print-tensor.4: 4.0000000
834+
/print-tensor.5: 5.0000000
835835
)";
836836

837837
ASSERT_EQ(expected, ss.str());
@@ -971,14 +971,14 @@ ENTRY main {
971971
engine.run(0);
972972
device.detach();
973973

974-
const std::string expected = R"(/print-tensor: {206,212}
975-
/print-tensor: {206,212}
976-
/print-tensor: {206,212}
977-
/print-tensor: {206,212}
978-
/print-tensor: {206,212}
979-
/print-tensor: {206,212}
980-
/print-tensor: {206,212}
981-
/print-tensor: {206,212}
974+
const std::string expected = R"(/print-tensor: [206.00000 212.00000]
975+
/print-tensor: [206.00000 212.00000]
976+
/print-tensor: [206.00000 212.00000]
977+
/print-tensor: [206.00000 212.00000]
978+
/print-tensor: [206.00000 212.00000]
979+
/print-tensor: [206.00000 212.00000]
980+
/print-tensor: [206.00000 212.00000]
981+
/print-tensor: [206.00000 212.00000]
982982
)";
983983

984984
// Check the output of the stage has aliases.

‎tensorflow/compiler/plugin/poplar/tests/pipeline_sequential_visitor_test.cc‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -144,12 +144,12 @@ ENTRY e {
144144
engine.run(0);
145145
device.detach();
146146

147-
const std::string expected = R"(/print-tensor: 3
148-
/print-tensor: 3
149-
/print-tensor: 3
150-
/print-tensor: 3
151-
/print-tensor: 3
152-
/print-tensor: 3
147+
const std::string expected = R"(/print-tensor: 3.0000000
148+
/print-tensor: 3.0000000
149+
/print-tensor: 3.0000000
150+
/print-tensor: 3.0000000
151+
/print-tensor: 3.0000000
152+
/print-tensor: 3.0000000
153153
)";
154154

155155
ASSERT_EQ(expected, ss.str());

0 commit comments

Comments
 (0)