# Buggy code
def add_running_predecessor(self, op_key, worker):
    self._running_preds.add(op_key)
    self._pred_workers.add(worker)
    if len(self._pred_workers) > 1:
        # we do not push when multiple workers in input
        self._pred_workers = set()
        self._running_preds = set()
        return

    if self.state != OperandState.UNSCHEDULED:
        return

    if all(k in self._running_preds for k in self._pred_keys):
        try:
            if worker in self._assigned_workers:
                return
            serialized_exec_graph = self._graph_ref.get_executable_operand_dag(
                self._op_key
            )

            self._get_execution_ref(address=worker).enqueue_graph(
                self._session_id,
                self._op_key,
                serialized_exec_graph,
                self._io_meta,
                dict(),
                self._info["optimize"],
                succ_keys=self._succ_keys,
                pred_keys=self._pred_keys,
                _promise=True,
            ).then(functools.partial(self._handle_worker_accept, worker))
            self._assigned_workers.add(worker)
            logger.debug("Pre-push operand %s into worker %s.", self._op_key, worker)
        except:  # noqa: E722
            logger.exception("Failed to pre-push operand %s", self._op_key)
        finally:
            self._pred_workers = set()
            self._running_preds = set()


# Traceback (SystemError)
In [1]: from mars.deploy.local import new_cluster

In [2]: import mars.tensor as mt

In [3]: cluster = new_cluster(scheduler_n_process=2, worker_n_process=2)
/Users/travis/miniconda3/conda-bld/arrow-cpp_1540410566532/work/cpp/src/plasma/store.cc:971: Allowing the Plasma store to use up to 3.43597GB of memory.
/Users/travis/miniconda3/conda-bld/arrow-cpp_1540410566532/work/cpp/src/plasma/store.cc:1001: Starting object store with directory /tmp and huge page support disabled
/Users/travis/miniconda3/conda-bld/arrow-cpp_1540410566532/work/cpp/src/plasma/eviction_policy.cc:85: There is not enough space to create this object, so evicting 0 objects to free up 0 bytes. The number of bytes in use (before this eviction) is 2130706432.

In [4]: a = mt.ones((3, 3), chunk_size=2)

In [5]: cluster.session.run(a)
Out[5]:
array([[1., 1., 1.],
[1., 1., 1.],
[1., 1., 1.]])

In [6]: cluster.session.run(a.dot(a))
Failed to start graph execution.
Traceback (most recent call last):
File "/Users/hekaisheng/Documents/mars_dev/mars/mars/scheduler/graph.py", line 249, in execute_graph
self.create_operand_actors()
File "/Users/hekaisheng/Documents/mars_dev/mars/mars/scheduler/graph.py", line 679, in create_operand_actors
op_refs = dict((k, v.result()) for k, v in op_refs.items())
File "/Users/hekaisheng/Documents/mars_dev/mars/mars/scheduler/graph.py", line 679, in <genexpr>
op_refs = dict((k, v.result()) for k, v in op_refs.items())
File "src/gevent/event.py", line 457, in gevent._event.AsyncResult.result
File "src/gevent/event.py", line 381, in gevent._event.AsyncResult.get
File "src/gevent/event.py", line 409, in gevent._event.AsyncResult.get
File "src/gevent/event.py", line 399, in gevent._event.AsyncResult.get
File "src/gevent/event.py", line 379, in gevent._event.AsyncResult._raise_exception
File "/Users/hekaisheng/miniconda3/lib/python3.6/site-packages/gevent/_compat.py", line 47, in reraise
raise value.with_traceback(tb)
File "mars/actors/pool/gevent_pool.pyx", line 984, in mars.actors.pool.gevent_pool.Communicator._on_receive_create_actor
self.pool.create_actor(message.actor_cls, message.actor_ref.uid,
File "mars/actors/pool/gevent_pool.pyx", line 241, in mars.actors.pool.gevent_pool.LocalActorPool.create_actor
raise ActorAlreadyExist('Actor {0} already exist, cannot create'.format(actor.uid))
mars.actors.errors.ActorAlreadyExist: Actor s:operator$8a307641-78a8-43a9-a612-8edca02ca309$058707656a679b3a0ab03c10d40abb0e already exist, cannot create
Traceback (most recent call last):
File "src/gevent/greenlet.py", line 716, in gevent._greenlet.Greenlet.run
File "mars/actors/pool/gevent_pool.pyx", line 88, in mars.actors.pool.gevent_pool.ActorExecutionContext.fire_run
cpdef object fire_run(self):
File "mars/actors/pool/gevent_pool.pyx", line 91, in mars.actors.pool.gevent_pool.ActorExecutionContext.fire_run
with self.lock:
File "mars/actors/pool/gevent_pool.pyx", line 102, in mars.actors.pool.gevent_pool.ActorExecutionContext.fire_run
raise
File "mars/actors/pool/gevent_pool.pyx", line 96, in mars.actors.pool.gevent_pool.ActorExecutionContext.fire_run
res = actor.on_receive(message_ctx.message)
File "mars/actors/core.pyx", line 108, in mars.actors.core.FunctionActor.on_receive
File "mars/actors/core.pyx", line 110, in mars.actors.core.FunctionActor.on_receive
File "/Users/hekaisheng/Documents/mars_dev/mars/mars/scheduler/graph.py", line 249, in execute_graph
self.create_operand_actors()
File "/Users/hekaisheng/Documents/mars_dev/mars/mars/scheduler/graph.py", line 679, in create_operand_actors
op_refs = dict((k, v.result()) for k, v in op_refs.items())
File "/Users/hekaisheng/Documents/mars_dev/mars/mars/scheduler/graph.py", line 679, in <genexpr>
op_refs = dict((k, v.result()) for k, v in op_refs.items())
File "src/gevent/event.py", line 457, in gevent._event.AsyncResult.result
File "src/gevent/event.py", line 381, in gevent._event.AsyncResult.get
File "src/gevent/event.py", line 409, in gevent._event.AsyncResult.get
File "src/gevent/event.py", line 399, in gevent._event.AsyncResult.get
File "src/gevent/event.py", line 379, in gevent._event.AsyncResult._raise_exception
File "/Users/hekaisheng/miniconda3/lib/python3.6/site-packages/gevent/_compat.py", line 47, in reraise
raise value.with_traceback(tb)
File "mars/actors/pool/gevent_pool.pyx", line 984, in mars.actors.pool.gevent_pool.Communicator._on_receive_create_actor
self.pool.create_actor(message.actor_cls, message.actor_ref.uid,
File "mars/actors/pool/gevent_pool.pyx", line 241, in mars.actors.pool.gevent_pool.LocalActorPool.create_actor
raise ActorAlreadyExist('Actor {0} already exist, cannot create'.format(actor.uid))
mars.actors.errors.ActorAlreadyExist: Actor s:operator$8a307641-78a8-43a9-a612-8edca02ca309$058707656a679b3a0ab03c10d40abb0e already exist, cannot create
2019-01-24T12:05:09Z <Greenlet "Greenlet-0" at 0x119218448: <built-in method fire_run of mars.actors.pool.gevent_pool.ActorExecutionContext object at 0x319839688>> failed with ActorAlreadyExist

---------------------------------------------------------------------------
SystemError                               Traceback (most recent call last)
<ipython-input-6-d3363e4685e3> in <module>()
----> 1 cluster.session.run(a.dot(a))

~/Documents/mars_dev/mars/mars/session.py in run(self, *tensors, **kw)
115         # execute the non-executed tensors
116         if idx_to_run_tensors:
--> 117             execute_result = self._sess.run(*idx_to_run_tensors.values(), **kw)
118             if execute_result:
119                 # fetch is True

~/Documents/mars_dev/mars/mars/deploy/local/session.py in run(self, *tensors, **kw)
80             if graph_state == GraphState.FAILED:
81                 # TODO(qin): add traceback
---> 82                 raise SystemError('Graph execution failed with unknown reason')
83
84         if 0 < timeout < time.time() - exec_start_time:

SystemError: Graph execution failed with unknown reason

# Fixed code
def add_running_predecessor(self, op_key, worker):
    self._running_preds.add(op_key)
    self._pred_workers.add(worker)
    if len(self._pred_workers) > 1:
        # we do not push when multiple workers in input
        self._pred_workers = set()
        self._running_preds = set()
        return

    if self.state != OperandState.UNSCHEDULED:
        return

    if all(k in self._running_preds for k in self._pred_keys):
        try:
            if worker in self._assigned_workers:
                return
            serialized_exec_graph = self._graph_refs[0].get_executable_operand_dag(
                self._op_key
            )

            self._get_execution_ref(address=worker).enqueue_graph(
                self._session_id,
                self._op_key,
                serialized_exec_graph,
                self._io_meta,
                dict(),
                self._info["optimize"],
                succ_keys=self._succ_keys,
                pred_keys=self._pred_keys,
                _promise=True,
            ).then(functools.partial(self._handle_worker_accept, worker))
            self._assigned_workers.add(worker)
            logger.debug("Pre-push operand %s into worker %s.", self._op_key, worker)
        except:  # noqa: E722
            logger.exception("Failed to pre-push operand %s", self._op_key)
        finally:
            self._pred_workers = set()
            self._running_preds = set()