Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
125 changes: 66 additions & 59 deletions conformance_tests/core/test_copy/src/test_copy.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -29,29 +29,6 @@ using lzt::to_int;
using lzt::to_u32;
using lzt::to_u8;

void get_copy_and_compute_ordinals(
const std::vector<ze_command_queue_group_properties_t>
&cmd_queue_group_props,
int &compute_ordinal, int &copy_ordinal) {
for (uint32_t i = 0; i < cmd_queue_group_props.size(); i++) {
if (cmd_queue_group_props[i].flags &
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE &&
compute_ordinal < 0) {
compute_ordinal = to_int(i);
}
if (cmd_queue_group_props[i].flags &
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COPY &&
!(cmd_queue_group_props[i].flags &
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE) &&
copy_ordinal < 0) {
copy_ordinal = to_int(i);
}
if (compute_ordinal >= 0 && copy_ordinal >= 0) {
break;
}
}
}

class zeCommandListAppendMemoryFillTests : public ::testing::Test {
protected:
void RunMaxMemoryFillTest(bool is_immediate, bool is_shared_system);
Expand All @@ -69,15 +46,20 @@ void zeCommandListAppendMemoryFillTests::
auto cmd_queue_group_props = get_command_queue_group_properties(
zeDevice::get_instance()->get_device());

int compute_ordinal = -1, copy_ordinal = -1;
get_copy_and_compute_ordinals(cmd_queue_group_props, compute_ordinal,
copy_ordinal);
ASSERT_TRUE((use_copy_engine && copy_ordinal >= 0) ||
(!use_copy_engine && compute_ordinal >= 0));
auto compute_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE |
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COOPERATIVE_KERNELS,
0);
auto copy_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props, ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COPY,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE);
ASSERT_TRUE((use_copy_engine && copy_ordinal) ||
(!use_copy_engine && compute_ordinal));

auto cmd_bundle = lzt::create_command_bundle(
lzt::get_default_context(), zeDevice::get_instance()->get_device(), 0,
to_u32(use_copy_engine ? copy_ordinal : compute_ordinal), is_immediate);
use_copy_engine ? *copy_ordinal : *compute_ordinal, is_immediate);
const size_t size = 4096;
void *memory = lzt::allocate_device_memory_with_allocator_selector(
size, is_shared_system);
Expand Down Expand Up @@ -212,15 +194,20 @@ void zeCommandListAppendMemoryFillTests::
auto cmd_queue_group_props = get_command_queue_group_properties(
zeDevice::get_instance()->get_device());

int compute_ordinal = -1, copy_ordinal = -1;
get_copy_and_compute_ordinals(cmd_queue_group_props, compute_ordinal,
copy_ordinal);
ASSERT_TRUE((use_copy_engine && copy_ordinal >= 0) ||
(!use_copy_engine && compute_ordinal >= 0));
auto compute_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE |
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COOPERATIVE_KERNELS,
0);
auto copy_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props, ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COPY,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE);
ASSERT_TRUE((use_copy_engine && copy_ordinal) ||
(!use_copy_engine && compute_ordinal));

auto cmd_bundle = lzt::create_command_bundle(
lzt::get_default_context(), zeDevice::get_instance()->get_device(), 0,
to_u32(use_copy_engine ? copy_ordinal : compute_ordinal), is_immediate);
use_copy_engine ? *copy_ordinal : *compute_ordinal, is_immediate);
const size_t size = 4096;
void *memory = lzt::allocate_device_memory_with_allocator_selector(
size, is_shared_system);
Expand Down Expand Up @@ -307,15 +294,20 @@ void zeCommandListAppendMemoryFillTests::
auto cmd_queue_group_props = get_command_queue_group_properties(
zeDevice::get_instance()->get_device());

int compute_ordinal = -1, copy_ordinal = -1;
get_copy_and_compute_ordinals(cmd_queue_group_props, compute_ordinal,
copy_ordinal);
ASSERT_TRUE((use_copy_engine && copy_ordinal >= 0) ||
(!use_copy_engine && compute_ordinal >= 0));
auto compute_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE |
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COOPERATIVE_KERNELS,
0);
auto copy_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props, ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COPY,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE);
ASSERT_TRUE((use_copy_engine && copy_ordinal) ||
(!use_copy_engine && compute_ordinal));

auto cmd_bundle = lzt::create_command_bundle(
lzt::get_default_context(), zeDevice::get_instance()->get_device(), 0,
to_u32(use_copy_engine ? copy_ordinal : compute_ordinal), is_immediate);
use_copy_engine ? *copy_ordinal : *compute_ordinal, is_immediate);
const size_t size = 4096;
void *memory = lzt::allocate_device_memory_with_allocator_selector(
size, is_shared_system);
Expand Down Expand Up @@ -1367,15 +1359,20 @@ void zeCommandListAppendMemoryCopyTests::
auto cmd_queue_group_props = get_command_queue_group_properties(
zeDevice::get_instance()->get_device());

int compute_ordinal = -1, copy_ordinal = -1;
get_copy_and_compute_ordinals(cmd_queue_group_props, compute_ordinal,
copy_ordinal);
ASSERT_TRUE((use_copy_engine && copy_ordinal >= 0) ||
(!use_copy_engine && compute_ordinal >= 0));
auto compute_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE |
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COOPERATIVE_KERNELS,
0);
auto copy_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props, ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COPY,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE);
ASSERT_TRUE((use_copy_engine && copy_ordinal) ||
(!use_copy_engine && compute_ordinal));

auto cmd_bundle = lzt::create_command_bundle(
lzt::get_default_context(), zeDevice::get_instance()->get_device(), 0,
to_u32(use_copy_engine ? copy_ordinal : compute_ordinal), is_immediate);
use_copy_engine ? *copy_ordinal : *compute_ordinal, is_immediate);

const size_t size = 16;
const std::vector<char> host_memory(size, 123);
Expand Down Expand Up @@ -1449,15 +1446,20 @@ void zeCommandListAppendMemoryCopyTests::
auto cmd_queue_group_props = get_command_queue_group_properties(
zeDevice::get_instance()->get_device());

int compute_ordinal = -1, copy_ordinal = -1;
get_copy_and_compute_ordinals(cmd_queue_group_props, compute_ordinal,
copy_ordinal);
ASSERT_TRUE((use_copy_engine && copy_ordinal >= 0) ||
(!use_copy_engine && compute_ordinal >= 0));
auto compute_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE |
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COOPERATIVE_KERNELS,
0);
auto copy_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props, ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COPY,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE);
ASSERT_TRUE((use_copy_engine && copy_ordinal) ||
(!use_copy_engine && compute_ordinal));

auto cmd_bundle = lzt::create_command_bundle(
lzt::get_default_context(), zeDevice::get_instance()->get_device(), 0,
to_u32(use_copy_engine ? copy_ordinal : compute_ordinal), is_immediate);
use_copy_engine ? *copy_ordinal : *compute_ordinal, is_immediate);

lzt::zeEventPool ep;
const size_t size = 16;
Expand Down Expand Up @@ -1543,15 +1545,20 @@ void zeCommandListAppendMemoryCopyTests::
auto cmd_queue_group_props = get_command_queue_group_properties(
zeDevice::get_instance()->get_device());

int compute_ordinal = -1, copy_ordinal = -1;
get_copy_and_compute_ordinals(cmd_queue_group_props, compute_ordinal,
copy_ordinal);
ASSERT_TRUE((use_copy_engine && copy_ordinal >= 0) ||
(!use_copy_engine && compute_ordinal >= 0));
auto compute_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE |
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COOPERATIVE_KERNELS,
0);
auto copy_ordinal = lzt::get_queue_ordinal(
cmd_queue_group_props, ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COPY,
ZE_COMMAND_QUEUE_GROUP_PROPERTY_FLAG_COMPUTE);
ASSERT_TRUE((use_copy_engine && copy_ordinal) ||
(!use_copy_engine && compute_ordinal));

auto cmd_bundle = lzt::create_command_bundle(
lzt::get_default_context(), zeDevice::get_instance()->get_device(), 0,
to_u32(use_copy_engine ? copy_ordinal : compute_ordinal), is_immediate);
use_copy_engine ? *copy_ordinal : *compute_ordinal, is_immediate);

lzt::zeEventPool ep;
const size_t size = 16;
Expand Down
Loading
Loading