OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity

Revert D27652484: [nnc] Enable CPU fusion inside Facebook

Browse Source 
Test Plan: revert-hammer

Differential Revision:
D27652484 (ac04cc775b)

Original commit changeset: a82681455dae

fbshipit-source-id: ecfef3ee1e7197148b172234691e72faf4b95cf8
This commit is contained in:
Bert Maher 2021-05-14 16:39:57 -07:00 committed by Facebook GitHub Bot
parent a4075fca9a
commit c4c2039fb2
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
torch/csrc/jit/codegen/fuser/interface.cpp
View File

@ -8,13 +8,8 @@
#include <c10/util/Flags.h>
#include <stdexcept>
#ifdef FBCODE_CAFFE2
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
C10_DEFINE_bool(torch_jit_enable_cpu_fusion, true, "enable cpu fusion");
#else
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
C10_DEFINE_bool(torch_jit_enable_cpu_fusion, false, "enable cpu fusion");
#endif
namespace torch {
namespace jit {
@ -51,7 +46,6 @@ bool canFuseOnGPU() {
void overrideCanFuseOnCPU(bool value) {
detail::cpu_fuser_enabled = value;
FLAGS_torch_jit_enable_cpu_fusion = value;
}
void overrideCanFuseOnGPU(bool value) {