Skip to content

Commit 29605c0

Browse files
committed
Fix typo and warning
1 parent 1a9243a commit 29605c0

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/ATen/native/xpu/mkl/BatchLinearAlgebra.cpp

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,11 +43,9 @@ void error_handle(int32_t* infos, const oneapi::mkl::lapack::batch_error& be) {
4343
auto ids = be.ids();
4444

4545
if (!errs.size()) {
46-
std::cout << "Cathed lapack exception:"
47-
<< "\nWhat: " << be.what() << "\nInfo: " << be.info()
48-
<< std::endl;
46+
TORCH_WARN("Caught lapack exception:\nWhat: ", be.what(), "\nInfo: ", be.info());
4947
for (auto& i : ids) {
50-
std::cout << "Error in martix #" << i << std::endl;
48+
TORCH_WARN("Error in matrix #", i);
5149
infos[i] = 1;
5250
}
5351
return;

0 commit comments

Comments
 (0)