[intel-npu] use new extension support verification for cmdQueue and B…

…atching also, for uniformity
csoka · Oct 18, 2024 · 5f69603 · 5f69603
1 parent de85a87
commit 5f69603
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/plugins/intel_npu/src/backend/src/zero_backend.cpp b/src/plugins/intel_npu/src/backend/src/zero_backend.cpp
@@ -30,11 +30,11 @@ uint32_t ZeroEngineBackend::getGraphExtVersion() const {
 }
 
 bool ZeroEngineBackend::isBatchingSupported() const {
-    return _instance->getGraphDdiTable().version() >= ZE_GRAPH_EXT_VERSION_1_6;
+    return _instance->isExtensionSupported(std::string(ZE_GRAPH_EXT_NAME_1_6));
 }
 
 bool ZeroEngineBackend::isCommandQueueExtSupported() const {
-    return _instance->getCommandQueueDdiTable().version() >= ZE_COMMAND_QUEUE_NPU_EXT_VERSION_1_0;
+    return _instance->isExtensionSupported(std::string(ZE_COMMAND_QUEUE_NPU_EXT_NAME));
 }
 
 bool ZeroEngineBackend::isLUIDExtSupported() const {