Skip to content

Commit 60d5898

Browse files
Investigate refactoring opportunities for batch management in Plugin and Compiler - clang
1 parent 599367e commit 60d5898

File tree

5 files changed

+25
-30
lines changed

5 files changed

+25
-30
lines changed

src/plugins/intel_npu/src/common/include/intel_npu/common/icompiler_adapter.hpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -42,12 +42,11 @@ class ICompilerAdapter {
4242
* @return A wrapper over the corresponding L0 graph handles (multiple only if "initBlobs" has been provided). This
4343
* wrapper further details the compiled model and brings it in a state closer to execution.
4444
*/
45-
virtual std::shared_ptr<IGraph> parse(
46-
ov::Tensor mainBlob,
47-
const Config& config,
48-
std::optional<std::vector<ov::Tensor>> initBlobs = std::nullopt,
49-
const std::optional<std::shared_ptr<const ov::Model>>& model = std::nullopt,
50-
std::optional<int64_t> batchSize = std::nullopt) const = 0;
45+
virtual std::shared_ptr<IGraph> parse(ov::Tensor mainBlob,
46+
const Config& config,
47+
std::optional<std::vector<ov::Tensor>> initBlobs = std::nullopt,
48+
const std::optional<std::shared_ptr<const ov::Model>>& model = std::nullopt,
49+
std::optional<int64_t> batchSize = std::nullopt) const = 0;
5150

5251
virtual ov::SupportedOpsMap query(const std::shared_ptr<const ov::Model>& model, const Config& config) const = 0;
5352
virtual uint32_t get_version() const = 0;

src/plugins/intel_npu/src/compiler_adapter/include/driver_compiler_adapter.hpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,11 @@ class DriverCompilerAdapter final : public ICompilerAdapter {
2222

2323
std::shared_ptr<IGraph> compileWS(const std::shared_ptr<ov::Model>& model, const Config& config) const override;
2424

25-
std::shared_ptr<IGraph> parse(
26-
ov::Tensor mainBlob,
27-
const Config& config,
28-
std::optional<std::vector<ov::Tensor>> initBlobs = std::nullopt,
29-
const std::optional<std::shared_ptr<const ov::Model>>& model = std::nullopt,
30-
std::optional<int64_t> batchSize = std::nullopt) const override;
25+
std::shared_ptr<IGraph> parse(ov::Tensor mainBlob,
26+
const Config& config,
27+
std::optional<std::vector<ov::Tensor>> initBlobs = std::nullopt,
28+
const std::optional<std::shared_ptr<const ov::Model>>& model = std::nullopt,
29+
std::optional<int64_t> batchSize = std::nullopt) const override;
3130

3231
ov::SupportedOpsMap query(const std::shared_ptr<const ov::Model>& model, const Config& config) const override;
3332

src/plugins/intel_npu/src/compiler_adapter/include/plugin_compiler_adapter.hpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,12 +23,11 @@ class PluginCompilerAdapter final : public ICompilerAdapter {
2323

2424
std::shared_ptr<IGraph> compileWS(const std::shared_ptr<ov::Model>& model, const Config& config) const override;
2525

26-
std::shared_ptr<IGraph> parse(
27-
ov::Tensor mainBlob,
28-
const Config& config,
29-
std::optional<std::vector<ov::Tensor>> initBlobs = std::nullopt,
30-
const std::optional<std::shared_ptr<const ov::Model>>& model = std::nullopt,
31-
std::optional<int64_t> batchSize = std::nullopt) const override;
26+
std::shared_ptr<IGraph> parse(ov::Tensor mainBlob,
27+
const Config& config,
28+
std::optional<std::vector<ov::Tensor>> initBlobs = std::nullopt,
29+
const std::optional<std::shared_ptr<const ov::Model>>& model = std::nullopt,
30+
std::optional<int64_t> batchSize = std::nullopt) const override;
3231

3332
ov::SupportedOpsMap query(const std::shared_ptr<const ov::Model>& model, const Config& config) const override;
3433

src/plugins/intel_npu/src/compiler_adapter/src/driver_compiler_adapter.cpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -231,12 +231,11 @@ std::shared_ptr<IGraph> DriverCompilerAdapter::compileWS(const std::shared_ptr<o
231231
config);
232232
}
233233

234-
std::shared_ptr<IGraph> DriverCompilerAdapter::parse(
235-
ov::Tensor mainBlob,
236-
const Config& config,
237-
std::optional<std::vector<ov::Tensor>> initBlobs,
238-
const std::optional<std::shared_ptr<const ov::Model>>& model,
239-
std::optional<int64_t> batchSize) const {
234+
std::shared_ptr<IGraph> DriverCompilerAdapter::parse(ov::Tensor mainBlob,
235+
const Config& config,
236+
std::optional<std::vector<ov::Tensor>> initBlobs,
237+
const std::optional<std::shared_ptr<const ov::Model>>& model,
238+
std::optional<int64_t> batchSize) const {
240239
OV_ITT_TASK_CHAIN(PARSE_BLOB, itt::domains::NPUPlugin, "DriverCompilerAdapter", "parse");
241240

242241
_logger.debug("parse start");

src/plugins/intel_npu/src/compiler_adapter/src/plugin_compiler_adapter.cpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -255,12 +255,11 @@ std::shared_ptr<IGraph> PluginCompilerAdapter::compileWS(const std::shared_ptr<o
255255
_compiler);
256256
}
257257

258-
std::shared_ptr<IGraph> PluginCompilerAdapter::parse(
259-
ov::Tensor mainBlob,
260-
const Config& config,
261-
std::optional<std::vector<ov::Tensor>> initBlobs,
262-
const std::optional<std::shared_ptr<const ov::Model>>& model,
263-
std::optional<int64_t> batchSize) const {
258+
std::shared_ptr<IGraph> PluginCompilerAdapter::parse(ov::Tensor mainBlob,
259+
const Config& config,
260+
std::optional<std::vector<ov::Tensor>> initBlobs,
261+
const std::optional<std::shared_ptr<const ov::Model>>& model,
262+
std::optional<int64_t> batchSize) const {
264263
OV_ITT_TASK_CHAIN(PARSE_BLOB, itt::domains::NPUPlugin, "PluginCompilerAdapter", "parse");
265264

266265
_logger.debug("parse start");

0 commit comments

Comments
 (0)