Skip to content

Commit 850fcfa

Browse files
committed
Comment test auxiliary code to be used later
1 parent e2fe179 commit 850fcfa

File tree

1 file changed

+25
-16
lines changed

1 file changed

+25
-16
lines changed

source/source_base/module_container/ATen/kernels/test/lapack_test.cpp

Lines changed: 25 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -133,22 +133,31 @@ TYPED_TEST(LapackTest, GeqrfInPlace) {
133133
&beta,
134134
QtQ.data<Type>(), n);
135135

136-
// Test code: print A
137-
std::cout << "A = " << std::endl;
138-
for (int i = 0; i < m; ++i) {
139-
for (int j = 0; j < n; ++j) {
140-
std::cout << A_input.to_device<DEVICE_CPU>().data<Type>()[i + j * m] << " ";
141-
}
142-
std::cout << std::endl;
143-
}
144-
// Test code: print QtQ
145-
std::cout << "QtQ = " << std::endl;
146-
for (int i = 0; i < n; ++i) {
147-
for (int j = 0; j < n; ++j) {
148-
std::cout << QtQ.data<Type>()[i + j * n] << " ";
149-
}
150-
std::cout << std::endl;
151-
}
136+
// To print value: first to_device CPU, then print
137+
// // Test code: print A
138+
// std::cout << "A = " << std::endl;
139+
// for (int i = 0; i < m; ++i) {
140+
// for (int j = 0; j < n; ++j) {
141+
// std::cout << A_input.to_device<DEVICE_CPU>().data<Type>()[i + j * m] << " ";
142+
// }
143+
// std::cout << std::endl;
144+
// }
145+
// // Test code: print Q
146+
// std::cout << "Q = " << std::endl;
147+
// for (int i = 0; i < m; ++i) {
148+
// for (int j = 0; j < n; ++j) {
149+
// std::cout << Q.data<Type>()[i + j * m] << " ";
150+
// }
151+
// std::cout << std::endl;
152+
// }
153+
// // Test code: print QtQ
154+
// std::cout << "QtQ = " << std::endl;
155+
// for (int i = 0; i < n; ++i) {
156+
// for (int j = 0; j < n; ++j) {
157+
// std::cout << QtQ.data<Type>()[i + j * n] << " ";
158+
// }
159+
// std::cout << std::endl;
160+
// }
152161

153162
// check QtQ
154163
for (int i = 0; i < n; ++i) {

0 commit comments

Comments
 (0)