MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / foreach

Method foreach

src/core/impl/comp_node/rocm/comp_node.cpp:657–672  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

655}
656
657void ROCmCompNode::foreach (thin_function<void(CompNode)> callback) {
658 auto sd = ROCmCompNodeImpl::sd;
659 if (!sd)
660 return;
661
662 for (int i = 0;; ++i) {
663 CompNode cur;
664 {
665 MGB_LOCK_GUARD(sd->mtx);
666 if (i >= sd->nr_node)
667 return;
668 cur = make_comp_node_from_impl(&sd->node[i]);
669 }
670 callback(cur);
671 }
672}
673
674size_t ROCmCompNode::get_device_count() {
675 static int cnt = -1;

Callers

nothing calls this directly

Calls 1

callbackFunction · 0.50

Tested by

no test coverage detected