@@ -343,8 +343,8 @@ TEST_P(urEnqueueKernelLaunchIncrementMultiDeviceMultiThreadTest, Success) {
343
343
constexpr size_t n_dimensions = 1 ;
344
344
345
345
auto queue = queuePerThread ? queues[i] : queues.back ();
346
- auto kernel = queuePerThread ? kernels[i] : kernels. back () ;
347
- auto sharedPtr = queuePerThread ? SharedMem[i] : SharedMem. back () ;
346
+ auto kernel = kernels[i];
347
+ auto sharedPtr = SharedMem[i];
348
348
349
349
std::vector<uur::raii::Event> Events (numOpsPerThread + 1 );
350
350
for (size_t j = 0 ; j < numOpsPerThread; j++) {
@@ -381,29 +381,12 @@ TEST_P(urEnqueueKernelLaunchIncrementMultiDeviceMultiThreadTest, Success) {
381
381
size_t ExpectedValue = InitialValue;
382
382
ExpectedValue += numOpsPerThread;
383
383
for (uint32_t j = 0 ; j < ArraySize; ++j) {
384
- if (queuePerThread) {
385
- ASSERT_EQ (data[j], ExpectedValue);
386
- } else {
387
- // All threads write to the same memory, so the value might be greater
388
- ASSERT_GE (data[j], ExpectedValue);
389
- }
384
+ ASSERT_EQ (data[j], ExpectedValue);
390
385
}
391
386
});
392
387
}
393
388
394
389
for (auto &thread : threads) {
395
390
thread.join ();
396
391
}
397
-
398
- if (!queuePerThread) {
399
- size_t ExpectedValue = InitialValue;
400
- ExpectedValue += numOpsPerThread * numThreads;
401
-
402
- for (size_t i = 0 ; i < devices.size (); i++) {
403
- for (uint32_t j = 0 ; j < ArraySize; ++j) {
404
- ASSERT_EQ (reinterpret_cast <uint32_t *>(SharedMem.back ())[j],
405
- ExpectedValue);
406
- }
407
- }
408
- }
409
392
}
0 commit comments