Skip to content

Commit f529bbe

Browse files
committed
[Quant] make x86 the default quantization backend (qengine)
1 parent c8546c9 commit f529bbe

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
@@ -307,7 +307,12 @@ at::QEngine Context::qEngine() const {
307307

308308
#ifdef USE_FBGEMM
309309
if (fbgemm::fbgemmSupportedCPU()) {
310-
qengine = at::kFBGEMM;
310+
/* X86 is enabled if and only if fbgemm is available.
311+
* It combines goodness of fbgemm and onednn by dispatching.
312+
* If onednn not available, always dispatch to fbgemm.
313+
* Make it default qengine for X86 CPU platforms.
314+
*/
315+
qengine = at::kX86;
311316
}
312317
#endif
313318
return qengine;

0 commit comments

Comments
 (0)