Use internal engine if selected for immediate list

This ensures internal for instance that the list used for the
page-fault manager uses the internal engine, rather than one
used by the application.

Change-Id: I531d5c9d31c75ea9de14ac0439a4da01c688e91b
Signed-off-by: Jaime Arteaga <[email protected]>

Author: Jaime Arteaga

level_zero/core/source/cmdlist/cmdlist_imp.cpp

@@ -69,7 +69,12 @@ CommandList *CommandList::createImmediate(uint32_t productFamily, Device *device
                                           bool internalUsage, bool isCopyOnly) {
 
     NEO::CommandStreamReceiver *csr = nullptr;
-    device->getCsrForOrdinalAndIndex(&csr, desc->ordinal, desc->index);
+    auto deviceImp = static_cast<DeviceImp *>(device);
+    if (internalUsage) {
+        csr = deviceImp->neoDevice->getInternalEngine().commandStreamReceiver;
+    } else {
+        device->getCsrForOrdinalAndIndex(&csr, desc->ordinal, desc->index);
+    }
 
     auto commandQueue = CommandQueue::create(productFamily, device, csr, desc, isCopyOnly);
     if (!commandQueue) {

level_zero/core/test/unit_tests/sources/cmdlist/test_cmdlist_1.cpp

@@ -14,6 +14,7 @@
 #include "test.h"
 
 #include "level_zero/core/source/builtin/builtin_functions_lib_impl.h"
+#include "level_zero/core/source/cmdqueue/cmdqueue_imp.h"
 #include "level_zero/core/source/context/context.h"
 #include "level_zero/core/source/driver/driver_handle_imp.h"
 #include "level_zero/core/source/image/image_hw.h"
@@ -159,6 +160,33 @@ TEST_F(CommandListCreate, givenValidPtrThenAppendMemoryPrefetchReturnsSuccess) {
     ASSERT_EQ(res, ZE_RESULT_SUCCESS);
 }
 
+TEST_F(CommandListCreate, givenImmediateCommandListThenInternalEngineIsUsedIfRequested) {
+    const ze_command_queue_desc_t desc = {};
+    bool internalEngine = true;
+
+    std::unique_ptr<L0::CommandList> commandList0(CommandList::createImmediate(productFamily,
+                                                                               device,
+                                                                               &desc,
+                                                                               internalEngine,
+                                                                               false));
+    ASSERT_NE(nullptr, commandList0);
+
+    CommandQueueImp *cmdQueue = reinterpret_cast<CommandQueueImp *>(commandList0->cmdQImmediate);
+    EXPECT_EQ(cmdQueue->getCsr(), neoDevice->getInternalEngine().commandStreamReceiver);
+
+    internalEngine = false;
+
+    std::unique_ptr<L0::CommandList> commandList1(CommandList::createImmediate(productFamily,
+                                                                               device,
+                                                                               &desc,
+                                                                               internalEngine,
+                                                                               false));
+    ASSERT_NE(nullptr, commandList1);
+
+    cmdQueue = reinterpret_cast<CommandQueueImp *>(commandList1->cmdQImmediate);
+    EXPECT_NE(cmdQueue->getCsr(), neoDevice->getInternalEngine().commandStreamReceiver);
+}
+
 TEST_F(CommandListCreate, givenImmediateCommandListThenCustomNumIddPerBlockUsed) {
     const ze_command_queue_desc_t desc = {};