[UR][CTS] Add invalid size test to urPhysicalMemCreate

oneapi-src · Jun 29, 2023 · 9a3141a · 9a3141a
1 parent 6cde82d
commit 9a3141a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/test/conformance/testing/include/uur/fixtures.h b/test/conformance/testing/include/uur/fixtures.h
@@ -586,7 +586,7 @@ struct urPhysicalMemTest : urContextTest {
  ASSERT_NE(physical_mem, nullptr);
  }
 
- void TearDown() {
+ void TearDown() override {
  if (physical_mem) {
  EXPECT_SUCCESS(urPhysicalMemRelease(physical_mem));
  }

diff --git a/test/conformance/virtual_memory/urPhysicalMemCreate.cpp b/test/conformance/virtual_memory/urPhysicalMemCreate.cpp
@@ -8,7 +8,7 @@ using urPhysicalMemCreateTestParams = size_t;
 struct urPhysicalMemCreateTest
  : uur::urContextTestWithParam<urPhysicalMemCreateTestParams> {
 
- void SetUp() {
+ void SetUp() override {
  UUR_RETURN_ON_FATAL_FAILURE(uur::urContextTestWithParam<
  urPhysicalMemCreateTestParams>::SetUp());
  ASSERT_SUCCESS(urVirtualMemGranularityGetInfo(
@@ -54,3 +54,15 @@ TEST_P(urPhysicalMemCreateTest, InvalidNullPointerPhysicalMem) {
  urPhysicalMemCreate(context, device, size, nullptr, nullptr),
  UR_RESULT_ERROR_INVALID_NULL_POINTER);
 }
+
+TEST_P(urPhysicalMemCreateTest, InvalidSize) {
+ if (granularity == 1) {
+ GTEST_SKIP()
+ << "A granularity of 1 means that any size will be accepted.";
+ }
+ ur_physical_mem_handle_t physical_mem = nullptr;
+ size_t invalid_size = size - 1;
+ ASSERT_EQ_RESULT(urPhysicalMemCreate(context, device, invalid_size, nullptr,
+ &physical_mem),
+ UR_RESULT_ERROR_INVALID_SIZE);
+}