IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bug in weight reorder. (#18224)

Browse Source 
The original memory was overwritten unexpectedly because it was chekcing shared_ptr instead of actual buffer address
This commit is contained in:
Taylor Yeonbok Lee 2023-06-24 00:35:07 -07:00 committed by GitHub
parent e0ad99081e
commit 22ef2f4e6a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/plugins/intel_gpu/src/graph/primitive_inst.cpp
View File

@ -1014,7 +1014,7 @@ event::ptr primitive_inst::update_weights() {
memory::ptr weights_memory = nullptr;
if (_reordered_weights_cache.is_full()) {
weights_memory = _reordered_weights_cache.get_lru_element().second;
can_reuse = weights_memory->size() <= expected_layout.bytes_count() && weights_memory != original_weights_memory;
can_reuse = weights_memory->size() <= expected_layout.bytes_count() && (weights_memory->buffer_ptr() != original_weights_memory->buffer_ptr());
}
if (can_reuse) {