frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-27 17:26:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[Re-landing 68111] Add JIT graph fuser for oneDNN Graph API (Preview4.1)"

Browse Source 
This reverts commit 8b11d810583ab1aac16b211efcc131c85d17c502.

Reverted https://github.com/pytorch/pytorch/pull/74596 on behalf of https://github.com/janeyx99
This commit is contained in:
PyTorch MergeBot
2022-04-29 15:40:17 +00:00
parent 0708630d9f
commit 3dcd67a1b3
38 changed files with 71 additions and 3238 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
torch/csrc/jit/python/init.cpp
View File

@ -8,9 +8,6 @@
#include <torch/csrc/jit/codegen/cuda/python_frontend/python_bindings.h>
#include <torch/csrc/jit/codegen/fuser/interface.h>
#include <torch/csrc/jit/codegen/fuser/kernel_cache.h>
#if (!defined(FBCODE_CAFFE2) && defined(BUILD_ONEDNN_GRAPH))
#include <torch/csrc/jit/codegen/onednn/interface.h>
#endif
#include <torch/csrc/jit/frontend/ir_emitter.h>
#include <torch/csrc/jit/frontend/tracer.h>
#include <torch/csrc/jit/ir/irparser.h>
@ -666,10 +663,6 @@ void initJITBindings(PyObject* module) {
return oldState;
})
.def("_jit_nvfuser_enabled", &RegisterCudaFuseGraph::isRegistered)
#if (!defined(FBCODE_CAFFE2) && defined(BUILD_ONEDNN_GRAPH))
.def("_jit_set_llga_enabled", &RegisterLlgaFuseGraph::setEnabled)
.def("_jit_llga_enabled", &RegisterLlgaFuseGraph::isEnabled)
#endif
.def(
"_jit_nvfuser_set_comparison_callback",
[](bool run_fallback, py::function fn) {