Function set_worker_pids

imperative/python/src/utils.cpp:267–287

}

static void set_worker_pids(PyObject* id, PyObject* tuple_item) {
    int64_t key = unpackLong(id);
    if (worker_pids.find(key) != worker_pids.end()) {
        throw std::runtime_error(
                "_set_worker_pids should be called only once for each "
                "DataLoaderIter.");
    }

    if (!PyTuple_Check(tuple_item)) {
        throw std::runtime_error("_set_worker_pids expects a tuple for child_pids");
    }

    std::set<pid_t> pids_set = {};
    auto size = PyTuple_GET_SIZE(tuple_item);
    for (Py_ssize_t idx = 0; idx < size; ++idx) {
        PyObject* obj = PyTuple_GET_ITEM(tuple_item, idx);
        pids_set.insert(static_cast<pid_t>(unpackLong(obj)));
    }

    worker_pids[key] = pids_set;
}

static void reset_worker_pids(PyObject* loader_id) {
    int64_t key = unpackLong(loader_id);