Skip to content

Commit

Permalink
[AMD] De-noise tf32 warnings (pytorch#144797)
Browse files Browse the repository at this point in the history
Summary: This is way too noisy especially during unit tests. So just log once.

Test Plan: OSS CI. Tested on a unit test and now I only see one line (hard to notice :) ).

Differential Revision: D68167633

Pull Request resolved: pytorch#144797
Approved by: https://github.com/jianyuh, https://github.com/leitian, https://github.com/yoyoyocmu
  • Loading branch information
xw285cornell authored and pytorchmergebot committed Jan 15, 2025
1 parent 69b883d commit 6ba53a5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions aten/src/ATen/Context.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -254,8 +254,8 @@ void Context::setAllowTF32CuBLAS(bool b) {
#ifdef USE_ROCM
const static auto allow_tf32 = c10::utils::check_env(hipblaslt_allow_tf32);
if (allow_tf32 != true) {
LOG(INFO) << "torch.backends.cuda.matmul.allow_tf32 is not supported on ROCm by default. "
<< "Please set environment variable HIPBLASLT_ALLOW_TF32=1 to enable it.";
C10_LOG_FIRST_N(INFO, 10) << "torch.backends.cuda.matmul.allow_tf32 is not supported on ROCm by default. "
<< "Please set environment variable HIPBLASLT_ALLOW_TF32=1 to enable it.";
return;
}
#endif
Expand Down

0 comments on commit 6ba53a5

Please sign in to comment.