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

[jit] Replace get_executor() with call() in abstract Function interface. (#65969)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/65969

ghstack-source-id: 141759210

Test Plan: no behavior change.

Reviewed By: anjali411

Differential Revision: D31326151

fbshipit-source-id: 201f6dc4c23fdb2531f6b8c73d26127f9e212de4
This commit is contained in:
Zhengxu Chen
2021-10-28 13:10:13 -07:00
committed by Facebook GitHub Bot
parent 8363da3f92
commit 5ef62c88a9
8 changed files with 85 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
torch/csrc/jit/python/script_init.cpp
View File

@ -1442,12 +1442,16 @@ void initJitScriptBindings(PyObject* module) {
.def(
"get_debug_state",
[](const StrongFunctionPtr& self) {
return self.function_->get_executor().getDebugState();
return toGraphFunction(*self.function_)
.get_executor()
.getDebugState();
})
.def(
"_debug_flush_compilation_cache",
[](const StrongFunctionPtr& self) {
return self.function_->get_executor().debugFlushCompilationCache();
toGraphFunction(*self.function_)
.get_executor()
.debugFlushCompilationCache();
})
.def_property_readonly(
"name",