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

updated to init onednn attr (#19055)

Browse Source
This commit is contained in:
Eddy Kim 2023-08-10 16:38:32 +09:00 committed by GitHub
parent 32a6a31de2
commit 9deef1480a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/plugins/intel_gpu/src/graph/include/program_node.h
View File

@ -386,8 +386,16 @@ public:
std::vector<fused_primitive_desc>& get_fused_primitives() { return fused_prims; }
#ifdef ENABLE_ONEDNN_FOR_GPU
const std::shared_ptr<dnnl::primitive_attr>& get_onednn_primitive_attributes() const { return onednn_attrs; }
std::shared_ptr<dnnl::primitive_attr>& get_onednn_primitive_attributes() { return onednn_attrs; }
const std::shared_ptr<dnnl::primitive_attr>& get_onednn_primitive_attributes() const {
if (onednn_attrs == nullptr)
const_cast<program_node*>(this)->init_onednn_primitive_attributes();
return onednn_attrs;
}
std::shared_ptr<dnnl::primitive_attr>& get_onednn_primitive_attributes() {
if (onednn_attrs == nullptr)
init_onednn_primitive_attributes();
return onednn_attrs;
}
const std::vector<fused_primitive_desc_onednn>& get_fused_primitives_onednn() const { return fused_prims_onednn; }
std::vector<fused_primitive_desc_onednn>& get_fused_primitives_onednn() { return fused_prims_onednn; }