refactor: improve xe mock engine testing

Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:
Bartosz Dunajski 2024-08-08 11:47:56 +00:00 committed by Compute-Runtime-Automation
parent e434c5595f
commit c42d0fbe49
2 changed files with 8 additions and 2 deletions

View File

@ -50,7 +50,7 @@ struct DrmMockXe : public DrmMockCustom {
StackVec<drm_xe_sync, 1> syncInputs;
StackVec<drm_xe_ext_set_property, 1> execQueueProperties;
drm_xe_exec_queue_create latestExecQueueCreate = {};
drm_xe_engine_class_instance latestQueueEngineClassInstance = {};
std::vector<drm_xe_engine_class_instance> latestQueueEngineClassInstances;
int waitUserFenceReturn = 0;
int execQueueBanPropertyReturn = 0;

View File

@ -152,7 +152,13 @@ int DrmMockXe::ioctl(DrmIoctl request, void *arg) {
case DrmIoctl::gemContextCreateExt: {
auto queueCreate = static_cast<drm_xe_exec_queue_create *>(arg);
latestExecQueueCreate = *queueCreate;
latestQueueEngineClassInstance = reinterpret_cast<drm_xe_engine_class_instance *>(queueCreate->instances)[0];
latestQueueEngineClassInstances.clear();
auto instances = reinterpret_cast<drm_xe_engine_class_instance *>(queueCreate->instances);
for (uint16_t i = 0; i < queueCreate->num_placements; i++) {
latestQueueEngineClassInstances.push_back(instances[i]);
}
auto extension = queueCreate->extensions;
while (extension) {