| 80 | } |
| 81 | |
| 82 | CompNode as_comp_node(const std::string& name) { |
| 83 | thread_local struct { |
| 84 | std::string name; |
| 85 | CompNode cn; |
| 86 | } dlpack_cncached; |
| 87 | if (dlpack_cncached.name != name) { |
| 88 | dlpack_cncached.name = name; |
| 89 | dlpack_cncached.cn = CompNode::load(name); |
| 90 | } |
| 91 | return dlpack_cncached.cn; |
| 92 | } |
| 93 | |
| 94 | CompNode mgb::imperative::get_tensor_device(const DLDevice& ctx, int stream) { |
| 95 | int id = ctx.device_id; |
no test coverage detected