change USE_MKLDNN default from ON (from #13303) to OFF for ppc64le (#…

…13759)

Summary:
MKLDNN is not supported on ppc64le change USE_MKLDNN to OFF for ppc64le
Pull Request resolved: #13759

Differential Revision: D12993121

Pulled By: soumith

fbshipit-source-id: 539d5cfcff2c03b59fa71e10b52fac333a64c381

setup.py

@@ -192,9 +192,15 @@ def hotpatch_var(var, prefix='USE_'):
 IS_WINDOWS = (platform.system() == 'Windows')
 IS_DARWIN = (platform.system() == 'Darwin')
 IS_LINUX = (platform.system() == 'Linux')
+IS_PPC = (platform.machine() == 'ppc64le')
 
 BUILD_PYTORCH = check_env_flag('BUILD_PYTORCH')
-USE_MKLDNN = check_env_flag('USE_MKLDNN', 'ON')
+# ppc64le does not support MKLDNN
+if IS_PPC:
+    USE_MKLDNN = check_env_flag('USE_MKLDNN', 'OFF')
+else:
+    USE_MKLDNN = check_env_flag('USE_MKLDNN', 'ON')
+
 USE_CUDA_STATIC_LINK = check_env_flag('USE_CUDA_STATIC_LINK')
 RERUN_CMAKE = True