Skip to content

Commit 755cad9

Browse files
committed
[Quant] make x86 the default quantization backend (qengine)
1 parent 136dadd commit 755cad9

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

aten/src/ATen/Context.cpp

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,12 @@ at::QEngine Context::qEngine() const {
304304

305305
#ifdef USE_FBGEMM
306306
if (fbgemm::fbgemmSupportedCPU()) {
307-
qengine = at::kFBGEMM;
307+
/* X86 is enabled if and only if fbgemm is available.
308+
* It combines goodness of fbgemm and onednn by dispatching.
309+
* If onednn not available, always dispatch to fbgemm.
310+
* Make it default qengine for X86 CPU platforms.
311+
*/
312+
qengine = at::kX86;
308313
}
309314
#endif
310315
return qengine;

0 commit comments

Comments
 (0)