Skip to content

Commit 3510a8e

Browse files
committed
Review comments
1 parent 62e668a commit 3510a8e

8 files changed

+13
-18
lines changed

src/plugins/intel_gpu/src/graph/impls/ocl/border.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ struct border_impl : typed_primitive_impl_ocl<border> {
150150
return args;
151151
}
152152

153-
std::vector<BufferDescriptor> get_internal_buffer_descs_impl(const kernel_impl_params&) const override {
153+
std::vector<BufferDescriptor> get_internal_buffer_descs(const kernel_impl_params&) const override {
154154
const auto& prim_params = static_cast<const kernel_selector::border_params&>(*_kernel_data.params);
155155
std::vector<BufferDescriptor> internal_buffers;
156156

src/plugins/intel_gpu/src/graph/impls/ocl/multi_stage_primitive.hpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ struct multi_stage_primitive : public typed_primitive_impl<PType> {
157157
return _kernels;
158158
}
159159

160-
std::vector<BufferDescriptor> get_internal_buffer_descs_impl(const kernel_impl_params&) const override {
160+
std::vector<BufferDescriptor> get_internal_buffer_descs(const kernel_impl_params&) const override {
161161
std::vector<BufferDescriptor> internal_buffers;
162162
for (auto& kd : _kernels_data) {
163163
if (kd.internalBuffers.empty())

src/plugins/intel_gpu/src/graph/impls/ocl/paged_attention.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ struct paged_attention_impl : multi_stage_primitive<paged_attention> {
141141
ob << make_data(&use_micro_sdpa, sizeof(bool));
142142
}
143143

144-
std::vector<BufferDescriptor> get_internal_buffer_descs_impl(const kernel_impl_params&) const override {
144+
std::vector<BufferDescriptor> get_internal_buffer_descs(const kernel_impl_params&) const override {
145145
/*
146146
* Internal buffers allocation owners and users:
147147
* +--------------------------------------+--------------------+--------------------+

src/plugins/intel_gpu/src/graph/impls/ocl/primitive_base.hpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ struct typed_primitive_impl_ocl : public typed_primitive_impl<PType> {
184184
return _kernels;
185185
}
186186

187-
std::vector<BufferDescriptor> get_internal_buffer_descs_impl(const kernel_impl_params&) const override {
187+
std::vector<BufferDescriptor> get_internal_buffer_descs(const kernel_impl_params&) const override {
188188
if (_kernel_data.internalBuffers.empty())
189189
return {};
190190

src/plugins/intel_gpu/src/graph/impls/ocl/scaled_dot_product_attention.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ struct scaled_dot_product_attention_impl : multi_stage_primitive<scaled_dot_prod
5656
}
5757

5858
protected:
59-
std::vector<BufferDescriptor> get_internal_buffer_descs_impl(const kernel_impl_params&) const override {
59+
std::vector<BufferDescriptor> get_internal_buffer_descs(const kernel_impl_params&) const override {
6060
// Look for the first sdpa_opt kernel entry. Currently, it can be used as default sdpa, indirect sdpa, or for both default
6161
// and indirect cases. All of sdpa_opt kernels use the same internal buffers, so we can find the first sdpa_opt and
6262
// use its` internal buffers configuration. The following scenarios are possible:

src/plugins/intel_gpu/src/graph/impls/onednn/primitive_onednn_base.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -572,7 +572,7 @@ struct typed_primitive_onednn_impl : public typed_primitive_impl<PType> {
572572
return event;
573573
}
574574

575-
std::vector<BufferDescriptor> get_internal_buffer_descs_impl(const kernel_impl_params&) const override {
575+
std::vector<BufferDescriptor> get_internal_buffer_descs(const kernel_impl_params&) const override {
576576
if (_scratchpad_md.get_size() == 0)
577577
return {};
578578
return {BufferDescriptor({ov::PartialShape{static_cast<int64_t>(_scratchpad_md.get_size())},

src/plugins/intel_gpu/src/graph/impls/sycl/primitive_sycl_base.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ struct typed_primitive_sycl_impl : public typed_primitive_impl<PType> {
5151
}
5252
}
5353

54-
std::vector<BufferDescriptor> get_internal_buffer_descs_impl(const kernel_impl_params&) const override {
54+
std::vector<BufferDescriptor> get_internal_buffer_descs(const kernel_impl_params&) const override {
5555
return {};
5656
}
5757
};

src/plugins/intel_gpu/src/graph/include/primitive_inst.h

+6-11
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,12 @@ template <class PType>
4242
class typed_primitive_inst;
4343

4444
struct ImplementationManager;
45-
struct BufferDescriptor;
45+
46+
struct BufferDescriptor {
47+
explicit BufferDescriptor(const layout& l, bool lockable = false) : m_lockable(lockable), m_layout(l) {}
48+
bool m_lockable = false;
49+
layout m_layout;
50+
};
4651

4752
/*
4853
Base class for all implementations.
@@ -148,12 +153,6 @@ struct ImplementationsFactory {
148153
bool has(impl_types impl_type) const;
149154
};
150155

151-
struct BufferDescriptor {
152-
explicit BufferDescriptor(const layout& l, bool lockable = false) : m_lockable(lockable), m_layout(l) {}
153-
bool m_lockable = false;
154-
layout m_layout;
155-
};
156-
157156
/*
158157
Base class for all primitive instances.
159158
It's main responsibility is to allocate memory required to run single, specified in ctor,
@@ -522,10 +521,6 @@ struct typed_primitive_impl : public primitive_impl {
522521
}
523522

524523
std::vector<BufferDescriptor> get_internal_buffer_descs(const kernel_impl_params& impl_params) const override {
525-
return get_internal_buffer_descs_impl(impl_params);
526-
}
527-
528-
virtual std::vector<BufferDescriptor> get_internal_buffer_descs_impl(const kernel_impl_params&) const {
529524
return {};
530525
}
531526

0 commit comments

Comments
 (0)