#include <c10/util/Flags.h>
#include <stdexcept>
-C10_DEFINE_bool(torch_jit_enable_cpu_fusion, false, "enable cpu fusion");
-
namespace torch {
namespace jit {
namespace detail {
-// Note: CPU fusion is currently disabled due to test flakiness
-#if defined(FBCODE_CAFFE2)
+#ifdef TORCH_ENABLE_LLVM
bool cpu_fuser_enabled = true;
#else
bool cpu_fuser_enabled = false;
}
bool canFuseOnCPU() {
- return fuser::hasFusionBackend(DeviceType::CPU) &&
- (detail::cpu_fuser_enabled || FLAGS_torch_jit_enable_cpu_fusion);
+ return fuser::hasFusionBackend(DeviceType::CPU) && detail::cpu_fuser_enabled;
}
bool canFuseOnGPU() {
return !strict_fuser_check;
}
if ((*device).is_cpu()) {
- return canFuseOnCPU();
+ return canFuseOnCPULegacy();
} else if ((*device).is_cuda()) {
return canFuseOnGPU();
} else if ((*device).is_xpu()) {
} // anonymous namespace
+static bool cpu_fuser_enabled_legacy = false;
+
+bool canFuseOnCPULegacy() {
+ return cpu_fuser_enabled_legacy;
+}
+
+void overrideCanFuseOnCPULegacy(bool value) {
+ cpu_fuser_enabled_legacy = value;
+}
+
void FuseGraph(std::shared_ptr<Graph>& graph, bool strict_fuser_check) {
AliasDb db(graph);
GraphFuser(&db, graph->block(), strict_fuser_check).run();
namespace torch {
namespace jit {
+TORCH_API bool canFuseOnCPULegacy();
+TORCH_API void overideCanFuseOnCPULegacy(bool value);
+
// NB: Be sure to run DCE before fusion, because dead instructions
// can prevent fusion opportunities from being exploited.
// On Windows will noop, NYI
.def("_jit_override_can_fuse_on_gpu", &overrideCanFuseOnGPU)
.def("_jit_can_fuse_on_cpu", &canFuseOnCPU)
.def("_jit_can_fuse_on_gpu", &canFuseOnGPU)
+ .def("_jit_can_fuse_on_cpu_legacy", &canFuseOnCPULegacy)
+ .def("_jit_override_can_fuse_on_cpu_legacy", &canFuseOnCPULegacy)
.def(
"_jit_differentiate",
[](Graph& g) {