saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Do not explicitly set USE_FBGEMM in tools/setup_helpers/cmake.py (#23314)

Browse source 
Summary:
Instead, defer its default value to CMakeLists.txt

NO_FBGEMM has already been handled in tools/setup_helpers/env.py
(although deprecated)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/23314

Differential Revision: D16493580

Pulled By: ezyang

fbshipit-source-id: 7255eb1df5e8a6dd0362507d68da0986a9ed46e2
This commit is contained in:
Hong Xu 2019-07-25 07:08:23 -07:00 committed by Facebook Github Bot
parent ba6f65cf33
commit f91b19c2aa
2 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
CMakeLists.txt
View file

@ -101,7 +101,7 @@ option(CAFFE2_STATIC_LINK_CUDA "Statically link CUDA libraries" OFF)
cmake_dependent_option(
USE_CUDNN "Use cuDNN" ON
"USE_CUDA" OFF)
option(USE_FBGEMM "Use FBGEMM (quantized 8-bit server operators)" OFF)
option(USE_FBGEMM "Use FBGEMM (quantized 8-bit server operators)" ON)
option(USE_FFMPEG "Use ffmpeg" OFF)
option(USE_GFLAGS "Use GFLAGS" OFF)
option(USE_GLOG "Use GLOG" OFF)

2
tools/setup_helpers/cmake.py
View file

@ -260,8 +260,6 @@ class CMake:
'BUILD_TEST': build_test,
'USE_CUDA': USE_CUDA,
'USE_DISTRIBUTED': USE_DISTRIBUTED,
'USE_FBGEMM': not (check_env_flag('NO_FBGEMM') or
check_negative_env_flag('USE_FBGEMM')),
'USE_NUMPY': USE_NUMPY,
'USE_SYSTEM_EIGEN_INSTALL': 'OFF'
})