8000 [Intel GPU] trigger tf32 no-gpu warn only when setting true · pytorch/pytorch@e25f057 · GitHub
[go: up one dir, main page]

Skip to content

Commit e25f057

Browse files
committed
[Intel GPU] trigger tf32 no-gpu warn only when setting true
ghstack-source-id: 7f99dea Pull Request resolved: #149926
1 parent 86dcdf9 commit e25f057

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

aten/src/ATen/Context.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,9 @@ void Context::setAllowTF32OneDNN(bool b){
145145
#ifdef USE_XPU
146146
allow_tf32_onednn = b;
147147
#else
148-
TORCH_WARN("TF32 acceleration on top of oneDNN is available for Intel GPUs. The current Torch version does not have Intel GPU Support.");
148+
if (b) {
149+
TORCH_WARN("TF32 acceleration on top of oneDNN is available for Intel GPUs. The current Torch version does not have Intel GPU Support.");
150+
}
149151
#endif
150152
}
151153

0 commit comments

Comments
 (0)
0