Skip to content

Commit cb88f4f

Browse files
simplification of test_sycl_context_interface per PR feedback
1 parent fb0b8f0 commit cb88f4f

File tree

1 file changed

+8
-39
lines changed

1 file changed

+8
-39
lines changed

dpctl-capi/tests/test_sycl_context_interface.cpp

Lines changed: 8 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -43,16 +43,20 @@ DEFINE_SIMPLE_CONVERSION_FUNCTIONS(vector_class<DPCTLSyclDeviceRef>,
4343

4444
struct TestDPCTLContextInterface : public ::testing::TestWithParam<const char *>
4545
{
46-
DPCTLSyclDeviceSelectorRef DSRef = nullptr;
46+
DPCTLSyclDeviceRef DRef = nullptr;
4747

4848
TestDPCTLContextInterface()
4949
{
50-
EXPECT_NO_FATAL_FAILURE(DSRef = DPCTLFilterSelector_Create(GetParam()));
50+
auto DS = DPCTLFilterSelector_Create(GetParam());
51+
if (DS) {
52+
EXPECT_NO_FATAL_FAILURE(DRef = DPCTLDevice_CreateFromSelector(DS));
53+
}
54+
DPCTLDeviceSelector_Delete(DS);
5155
}
5256

5357
void SetUp()
5458
{
55-
if (!DSRef) {
59+
if (!DRef) {
5660
auto message = "Skipping as no device of type " +
5761
std::string(GetParam()) + ".";
5862
GTEST_SKIP_(message.c_str());
@@ -61,32 +65,23 @@ struct TestDPCTLContextInterface : public ::testing::TestWithParam<const char *>
6165

6266
~TestDPCTLContextInterface()
6367
{
64-
EXPECT_NO_FATAL_FAILURE(DPCTLDeviceSelector_Delete(DSRef));
68+
EXPECT_NO_FATAL_FAILURE(DPCTLDevice_Delete(DRef));
6569
}
6670
};
6771

6872
TEST_P(TestDPCTLContextInterface, Chk_Create)
6973
{
7074
DPCTLSyclContextRef CRef = nullptr;
71-
DPCTLSyclDeviceRef DRef = nullptr;
72-
EXPECT_NO_FATAL_FAILURE(DRef = DPCTLDevice_CreateFromSelector(DSRef));
73-
if (!DRef)
74-
GTEST_SKIP_("Device not found");
7575
EXPECT_NO_FATAL_FAILURE(CRef = DPCTLContext_Create(DRef, nullptr, 0));
7676
ASSERT_TRUE(CRef);
77-
EXPECT_NO_FATAL_FAILURE(DPCTLDevice_Delete(DRef));
7877
EXPECT_NO_FATAL_FAILURE(DPCTLContext_Delete(CRef));
7978
}
8079

8180
TEST_P(TestDPCTLContextInterface, Chk_CreateWithDevices)
8281
{
8382
size_t nCUs = 0;
8483
DPCTLSyclContextRef CRef = nullptr;
85-
DPCTLSyclDeviceRef DRef = nullptr;
8684
DPCTLDeviceVectorRef DVRef = nullptr;
87-
EXPECT_NO_FATAL_FAILURE(DRef = DPCTLDevice_CreateFromSelector(DSRef));
88-
if (!DRef)
89-
GTEST_SKIP_("Device not found");
9085

9186
/* TODO: Once we have wrappers for sub-device creation let us use those
9287
* functions.
@@ -108,7 +103,6 @@ TEST_P(TestDPCTLContextInterface, Chk_CreateWithDevices)
108103
GTEST_SKIP_("Skipping creating context for sub-devices");
109104
}
110105
}
111-
EXPECT_NO_FATAL_FAILURE(DPCTLDevice_Delete(DRef));
112106
EXPECT_NO_FATAL_FAILURE(DPCTLDeviceVector_Delete(DVRef));
113107
EXPECT_NO_FATAL_FAILURE(DPCTLContext_Delete(CRef));
114108
}
@@ -117,12 +111,8 @@ TEST_P(TestDPCTLContextInterface, Chk_CreateWithDevices_GetDevices)
117111
{
118112
size_t nCUs = 0;
119113
DPCTLSyclContextRef CRef = nullptr;
120-
DPCTLSyclDeviceRef DRef = nullptr;
121114
DPCTLDeviceVectorRef DVRef = nullptr;
122115
DPCTLDeviceVectorRef Res_DVRef = nullptr;
123-
EXPECT_NO_FATAL_FAILURE(DRef = DPCTLDevice_CreateFromSelector(DSRef));
124-
if (!DRef)
125-
GTEST_SKIP_("Device not found");
126116

127117
/* TODO: Once we have wrappers for sub-device creation let us use those
128118
* functions.
@@ -151,7 +141,6 @@ TEST_P(TestDPCTLContextInterface, Chk_CreateWithDevices_GetDevices)
151141
GTEST_SKIP_("Skipping creating context for sub-devices");
152142
}
153143
}
154-
EXPECT_NO_FATAL_FAILURE(DPCTLDevice_Delete(DRef));
155144
EXPECT_NO_FATAL_FAILURE(DPCTLDeviceVector_Delete(DVRef));
156145
EXPECT_NO_FATAL_FAILURE(DPCTLContext_Delete(CRef));
157146
EXPECT_NO_FATAL_FAILURE(DPCTLDeviceVector_Delete(Res_DVRef));
@@ -160,30 +149,21 @@ TEST_P(TestDPCTLContextInterface, Chk_CreateWithDevices_GetDevices)
160149
TEST_P(TestDPCTLContextInterface, Chk_GetDevices)
161150
{
162151
DPCTLSyclContextRef CRef = nullptr;
163-
DPCTLSyclDeviceRef DRef = nullptr;
164152
DPCTLDeviceVectorRef DVRef = nullptr;
165-
EXPECT_NO_FATAL_FAILURE(DRef = DPCTLDevice_CreateFromSelector(DSRef));
166-
if (!DRef)
167-
GTEST_SKIP_("Device not found");
168153
EXPECT_NO_FATAL_FAILURE(CRef = DPCTLContext_Create(DRef, nullptr, 0));
169154
ASSERT_TRUE(CRef);
170155
EXPECT_NO_FATAL_FAILURE(DVRef = DPCTLContext_GetDevices(CRef));
171156
ASSERT_TRUE(DVRef);
172157
EXPECT_TRUE(DPCTLDeviceVector_Size(DVRef) == 1);
173-
EXPECT_NO_FATAL_FAILURE(DPCTLDevice_Delete(DRef));
174158
EXPECT_NO_FATAL_FAILURE(DPCTLContext_Delete(CRef));
175159
EXPECT_NO_FATAL_FAILURE(DPCTLDeviceVector_Delete(DVRef));
176160
}
177161

178162
TEST_P(TestDPCTLContextInterface, Chk_AreEq)
179163
{
180164
DPCTLSyclContextRef CRef1 = nullptr, CRef2 = nullptr, CRef3 = nullptr;
181-
DPCTLSyclDeviceRef DRef = nullptr;
182165
bool are_eq = true, are_not_eq = false;
183166

184-
EXPECT_NO_FATAL_FAILURE(DRef = DPCTLDevice_CreateFromSelector(DSRef));
185-
if (!DRef)
186-
GTEST_SKIP_("Device not found");
187167
EXPECT_NO_FATAL_FAILURE(CRef1 = DPCTLContext_Create(DRef, nullptr, 0));
188168
EXPECT_NO_FATAL_FAILURE(CRef2 = DPCTLContext_Copy(CRef1));
189169
// TODO: This work till DPC++ does not have a default context per device,
@@ -198,7 +178,6 @@ TEST_P(TestDPCTLContextInterface, Chk_AreEq)
198178
EXPECT_TRUE(are_eq);
199179
EXPECT_FALSE(are_not_eq);
200180

201-
EXPECT_NO_FATAL_FAILURE(DPCTLDevice_Delete(DRef));
202181
EXPECT_NO_FATAL_FAILURE(DPCTLContext_Delete(CRef1));
203182
EXPECT_NO_FATAL_FAILURE(DPCTLContext_Delete(CRef2));
204183
EXPECT_NO_FATAL_FAILURE(DPCTLContext_Delete(CRef3));
@@ -207,41 +186,31 @@ TEST_P(TestDPCTLContextInterface, Chk_AreEq)
207186
TEST_P(TestDPCTLContextInterface, Chk_IsHost)
208187
{
209188
DPCTLSyclContextRef CRef = nullptr;
210-
DPCTLSyclDeviceRef DRef = nullptr;
211189
bool is_host_device = false, is_host_context = false;
212190

213-
EXPECT_NO_FATAL_FAILURE(DRef = DPCTLDevice_CreateFromSelector(DSRef));
214-
if (!DRef)
215-
GTEST_SKIP_("Device not found");
216191
EXPECT_NO_FATAL_FAILURE(CRef = DPCTLContext_Create(DRef, nullptr, 0));
217192
ASSERT_TRUE(CRef);
218193

219194
EXPECT_NO_FATAL_FAILURE(is_host_device = DPCTLDevice_IsHost(DRef));
220195
EXPECT_NO_FATAL_FAILURE(is_host_context = DPCTLContext_IsHost(CRef));
221196
EXPECT_TRUE(is_host_device == is_host_context);
222197

223-
EXPECT_NO_FATAL_FAILURE(DPCTLDevice_Delete(DRef));
224198
EXPECT_NO_FATAL_FAILURE(DPCTLContext_Delete(CRef));
225199
}
226200

227201
TEST_P(TestDPCTLContextInterface, Chk_GetBackend)
228202
{
229203
DPCTLSyclContextRef CRef = nullptr;
230-
DPCTLSyclDeviceRef DRef = nullptr;
231204
DPCTLSyclBackendType context_backend = DPCTL_UNKNOWN_BACKEND,
232205
device_backend = DPCTL_UNKNOWN_BACKEND;
233206

234-
EXPECT_NO_FATAL_FAILURE(DRef = DPCTLDevice_CreateFromSelector(DSRef));
235-
if (!DRef)
236-
GTEST_SKIP_("Device not found");
237207
EXPECT_NO_FATAL_FAILURE(CRef = DPCTLContext_Create(DRef, nullptr, 0));
238208
ASSERT_TRUE(CRef);
239209

240210
EXPECT_NO_FATAL_FAILURE(device_backend = DPCTLDevice_GetBackend(DRef));
241211
EXPECT_NO_FATAL_FAILURE(context_backend = DPCTLContext_GetBackend(CRef));
242212
EXPECT_TRUE(device_backend == context_backend);
243213

244-
EXPECT_NO_FATAL_FAILURE(DPCTLDevice_Delete(DRef));
245214
EXPECT_NO_FATAL_FAILURE(DPCTLContext_Delete(CRef));
246215
}
247216

0 commit comments

Comments
 (0)