kparzysz-quic commented on a change in pull request #9769:
URL: https://github.com/apache/tvm/pull/9769#discussion_r772655068
##########
File path: src/runtime/hexagon/hexagon/hexagon_buffer.cc
##########
@@ -187,66 +188,55 @@ void HexagonBuffer::SetStorageScope(Optional<String>
scope) {
}
}
-void HexagonBuffer::CopyTo(void* data, size_t nbytes) {
- CHECK(nbytes_ == nbytes);
+void HexagonBuffer::CopyTo(void* data, size_t nbytes) const {
+ CHECK_LE(nbytes, nbytes_);
size_t offset = 0;
for (size_t i = 0; i < nallocs_; ++i) {
- CHECK(nbytes / nallocs_ == managed_allocations_[i]->nbytes_);
+ size_t bytes_to_copy = std::min(nbytes - offset,
managed_allocations_[i]->nbytes_);
+ if (bytes_to_copy == 0) break;
memcpy(static_cast<char*>(data) + offset,
- static_cast<const char*>(managed_allocations_[i]->data_),
- managed_allocations_[i]->nbytes_);
+ static_cast<const char*>(managed_allocations_[i]->data_),
bytes_to_copy);
- offset += managed_allocations_[i]->nbytes_;
+ offset += bytes_to_copy;
}
}
void HexagonBuffer::CopyFrom(void* data, size_t nbytes) {
- CHECK(nbytes_ == nbytes);
+ CHECK_LE(nbytes, nbytes_);
size_t offset = 0;
for (size_t i = 0; i < nallocs_; ++i) {
- CHECK(nbytes / nallocs_ == managed_allocations_[i]->nbytes_);
+ size_t bytes_to_copy = std::min(nbytes - offset,
managed_allocations_[i]->nbytes_);
+ if (bytes_to_copy == 0) break;
memcpy(static_cast<char*>(managed_allocations_[i]->data_),
- static_cast<const char*>(data) + offset,
managed_allocations_[i]->nbytes_);
+ static_cast<const char*>(data) + offset, bytes_to_copy);
- offset += managed_allocations_[i]->nbytes_;
+ offset += bytes_to_copy;
}
}
-void HexagonBuffer::CopyFrom(const HexagonBuffer& other) {
- CHECK(nbytes_ == other.nbytes_);
+void HexagonBuffer::CopyFrom(const HexagonBuffer& other, size_t nbytes) {
+ CHECK_LE(nbytes, nbytes_);
+ CHECK_LE(nbytes, other.nbytes_);
if (nallocs_ == other.nallocs_) {
+ size_t copied = 0;
for (size_t i = 0; i < nallocs_; ++i) {
- CHECK(managed_allocations_[i]->nbytes_ ==
other.managed_allocations_[i]->nbytes_);
+ size_t bytes_to_copy = std::min(nbytes - copied,
managed_allocations_[i]->nbytes_);
+ if (bytes_to_copy == 0) break;
- memcpy(static_cast<char*>(managed_allocations_[i]->data_),
- static_cast<const char*>(other.managed_allocations_[i]->data_),
- managed_allocations_[i]->nbytes_);
- }
- } else if (nallocs_ == 1) {
- size_t offset = 0;
- for (size_t i = 0; i < other.nallocs_; ++i) {
- CHECK(nbytes_ / other.nallocs_ ==
other.managed_allocations_[i]->nbytes_);
-
- memcpy(static_cast<char*>(managed_allocations_[0]->data_) + offset,
- static_cast<const char*>(other.managed_allocations_[i]->data_),
- other.managed_allocations_[i]->nbytes_);
-
- offset += other.managed_allocations_[i]->nbytes_;
- }
- } else if (other.nallocs_ == 1) {
- size_t offset = 0;
- for (size_t i = 0; i < nallocs_; ++i) {
- CHECK(other.nbytes_ / nallocs_ == managed_allocations_[i]->nbytes_);
+ CHECK_LE(other.managed_allocations_[i]->nbytes_,
managed_allocations_[i]->nbytes_);
memcpy(static_cast<char*>(managed_allocations_[i]->data_),
- static_cast<const char*>(other.managed_allocations_[0]->data_) +
offset,
- managed_allocations_[i]->nbytes_);
+ static_cast<const char*>(other.managed_allocations_[i]->data_),
bytes_to_copy);
- offset += managed_allocations_[i]->nbytes_;
+ copied += bytes_to_copy;
Review comment:
Apparently this PR still needs approval from someone else, so I just
added another commit that renames `offset` to `copied`.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
