|
|
|
@@ -131,7 +131,6 @@ py::function PrimitivePy::GetComputeFunction() const { |
|
|
|
MS_LOG(INFO) << name() << ": get_vm_impl_fn"; |
|
|
|
py::function get_fn = parse::python_adapter::GetPyFn(vm_module, get_vm_impl_fn); |
|
|
|
py::function vm_fn = get_fn(python_obj_); |
|
|
|
|
|
|
|
if (py::isinstance<py::none>(vm_fn)) { |
|
|
|
MS_LOG(WARNING) << "Cannot find " << python_obj_.attr("__class__").attr("__name__").cast<std::string>(); |
|
|
|
vm_fn = mindspore::GetComputeFunction(Primitive::name()); |
|
|
|
|