| 7 | namespace imperative { |
| 8 | |
| 9 | TensorChecksumCalc::ChecksumResult TensorChecksumCalc::calc(TensorPtr ptr) { |
| 10 | auto&& dt = ptr->dev_tensor(); |
| 11 | if (!dt.layout().total_nr_elems()) { |
| 12 | static ChecksumResult empty_checksum; |
| 13 | return empty_checksum; |
| 14 | } |
| 15 | |
| 16 | auto span = dt.layout().span(); |
| 17 | megdnn::TensorND tensor; |
| 18 | tensor.reset_ptr(dt.raw_ptr() + span.low_byte); |
| 19 | tensor.layout.init_contiguous_stride({span.dist_byte()}); |
| 20 | tensor.layout.dtype = dtype::Byte(); |
| 21 | |
| 22 | DeviceTensorStorage* workspace; |
| 23 | { |
| 24 | MGB_LOCK_GUARD(m_workspace_mtx); |
| 25 | workspace = &m_workspace[std::this_thread::get_id()].storage[ptr->comp_node()]; |
| 26 | } |
| 27 | auto comp_node = ptr->comp_node(); |
| 28 | comp_node.activate(); |
| 29 | auto opr = opr::intl::get_megdnn_global_opr<megdnn::Checksum>(comp_node); |
| 30 | auto workspace_reqsize = opr->get_workspace_in_bytes(tensor.layout); |
| 31 | workspace->comp_node(ptr->comp_node()).ensure_size(workspace_reqsize); |
| 32 | |
| 33 | megdnn::Workspace mwk; |
| 34 | if (workspace_reqsize) |
| 35 | mwk = {workspace->ptr(), workspace_reqsize}; |
| 36 | |
| 37 | return opr->exec(tensor, mwk); |
| 38 | } |
| 39 | |
| 40 | class TensorSanityCheckImpl { |
| 41 | public: |
no test coverage detected