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

Fixed compilation with gcc-7 (#15876)

Browse Source 
* Fixed compilation with gcc-7

* Update src/core/reference/include/ngraph/runtime/reference/eye.hpp

Co-authored-by: Katarzyna Mitrus <katarzyna.mitrus@intel.com>

* returned f16 and bf16

---------

Co-authored-by: Katarzyna Mitrus <katarzyna.mitrus@intel.com>
This commit is contained in:
Ilya Lavrenov 2023-03-01 23:08:10 +04:00 committed by GitHub
parent 3c67509fc8
commit 5422242e86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/core/reference/include/ngraph/runtime/reference/eye.hpp
View File

@ -19,15 +19,8 @@ void eye(T* data, const Shape& out_shape, const int64_t diagonal_index) {
const int64_t num_columns = out_shape[out_shape.size() - 1];
const int64_t matrix_size = num_rows * num_columns;
#if defined(__GNUC__) && !defined(__clang__)
# pragma GCC diagnostic push
# pragma GCC diagnostic ignored "-Wclass-memaccess" // to avoid warnings for memset of ov::float16, ov::bfloat16
#endif
// fill tensor by zero
std::memset(data, 0, num_matrices * num_columns * num_rows * sizeof(T));
#if defined(__GNUC__) && !defined(__clang__)
# pragma GCC diagnostic pop
#endif
std::fill(data, data + num_matrices * matrix_size, T(0));
// set ones on diagonal
const int64_t shift_by_columns = std::max(diagonal_index, int64_t(0));