USE_FBGEMM=True by default

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/14868

Differential Revision: D13383390

Pulled By: jamesr66a

fbshipit-source-id: 1880c07dfd239e19153bd4fde2ab2c8d0604f956

diff --git a/tools/build_pytorch_libs.bat b/tools/build_pytorch_libs.bat
index aed0ffd..6af203f 100755 (executable)
--- a/tools/build_pytorch_libs.bat
+++ b/tools/build_pytorch_libs.bat
@@ -27,7 +27,7 @@ if not exist torch\lib\tmp_install mkdir torch\lib\tmp_install
 
 : Variable defaults
 set /a USE_CUDA=0
-set /a USE_FBGEMM=0
+set /a USE_FBGEMM=1
 set /a USE_ROCM=0
 set /a USE_NNPACK=0
 set /a USE_QNNPACK=0

diff --git a/tools/build_pytorch_libs.sh b/tools/build_pytorch_libs.sh
index 63f169d..d164bdc 100755 (executable)
--- a/tools/build_pytorch_libs.sh
+++ b/tools/build_pytorch_libs.sh
@@ -43,7 +43,7 @@ fi
 
 # Options for building only a subset of the libraries
 USE_CUDA=0
-USE_FBGEMM=0
+USE_FBGEMM=1
 USE_ROCM=0
 USE_NNPACK=0
 USE_MKLDNN=0

diff --git a/tools/setup_helpers/fbgemm.py b/tools/setup_helpers/fbgemm.py
index 8599398..bd30bcc 100644 (file)
--- a/tools/setup_helpers/fbgemm.py
+++ b/tools/setup_helpers/fbgemm.py
@@ -1,6 +1,6 @@
 from .env import check_env_flag
 
-USE_FBGEMM = False
+USE_FBGEMM = True
 
 if check_env_flag('NO_FBGEMM'):
     USE_FBGEMM = False