xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix input with cell input problem

This commit is contained in:
liangzhibo 2024-01-31 15:27:27 +08:00 committed by r1chardf1d0
parent 85ef41529c
commit 39d9dc916f
2 changed files with 2 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/pipeline/jit/pi/graph_capture/abstract_object.cc
View File

@ -59,6 +59,7 @@ namespace pijit {
static const std::set<AObject::Type> kMsSupportedType = {
AObject::kTypeInt, AObject::kTypeBool, AObject::kTypeFloat,
AObject::kTypeNone, AObject::kTypeString, AObject::kTypeTensor,
AObject::kTypeFuncGraphOut,
};
MemPool<AbstractObjectBase> AbstractObjectBase::aobject_mem_pool_(__FILE__, __LINE__, "AObject");

11
mindspore/ccsrc/pipeline/jit/pi/graph_capture/graph_analyzer.cc
View File

@ -379,16 +379,7 @@ void MindGraphAnalyzer::Analyze() {
if (!support_ret) {
return;
}
PyCodeObject *co = graph_->GetCodeObj();
const FrameStates &enter_frame = graph_->GetFrame(0);
const auto &args = enter_frame.GetLocals();
int argc = co->co_argcount + co->co_kwonlyargcount;
// check all parameters is graph supported, but here not check variable arguments
auto end = args.begin() + argc;
auto iter = std::find_if(args.begin(), end, [](ValueNode *i) { return !ValidateGraphParameters(i); });
if (iter == end) {
need_interpret_ = false;
}
need_interpret_ = false;
}
bool MindGraphAnalyzer::AnalyzeAliveLocals(std::vector<ValueNode *> aliveNodes) {