Skip to content

Commit d359eb0

Browse files
Private mem size may be 0 on CPU devices
1 parent 7ea6b96 commit d359eb0

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

libsyclinterface/tests/test_sycl_kernel_interface.cpp

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,8 +137,14 @@ TEST_P(TestDPCTLSyclKernelInterface, CheckGetPrivateMemSize)
137137
EXPECT_NO_FATAL_FAILURE(axpy_private_mem_sz =
138138
DPCTLKernel_GetPrivateMemSize(AxpyKRef));
139139

140-
ASSERT_TRUE(add_private_mem_sz != 0);
141-
ASSERT_TRUE(axpy_private_mem_sz != 0);
140+
if (DPCTLDevice_IsGPU(DRef)) {
141+
ASSERT_TRUE(add_private_mem_sz != 0);
142+
ASSERT_TRUE(axpy_private_mem_sz != 0);
143+
}
144+
else {
145+
ASSERT_TRUE(add_private_mem_sz >= 0);
146+
ASSERT_TRUE(axpy_private_mem_sz >= 0);
147+
}
142148
}
143149

144150
TEST_P(TestDPCTLSyclKernelInterface, CheckGetMaxNumSubGroups)

0 commit comments

Comments
 (0)