Skip to content
Merged
Changes from 1 commit
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
83 changes: 83 additions & 0 deletions sycl/test-e2e/USM/fill_any_size.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,83 @@
// RUN: %{build} -o %t1.out
// RUN: %{run} %t1.out
// UNSUPPORTED: (opencl && cpu)
// clang-format off
// UNSUPPORTED-TRACKER: https://github.com/oneapi-src/unified-runtime/issues/2440
// clang-format on

/**
* Test of the queue::fill interface with a range of pattern sizes and values.
*
* Loops over pattern sizes from 1 to MaxPatternSize bytes and calls queue::fill
* with std::array<uint8_t,Size> for the pattern. Two pattern values are tested,
* all zeros and value=index+42. The output is copied back to host and
* validated.
*/

#include <array>
#include <cstdio>
#include <sycl/detail/core.hpp>
#include <sycl/usm.hpp>

constexpr size_t MaxPatternSize{32}; // Bytes.
constexpr size_t NumElements{10};
constexpr size_t NumRepeats{1};
constexpr bool verbose{false};

template <size_t PatternSize, bool SameValue>
int test(sycl::queue &q, uint8_t firstValue = 0) {
using T = std::array<uint8_t, PatternSize>;
T value{};
for (size_t i{0}; i < PatternSize; ++i) {
if constexpr (SameValue) {
value[i] = firstValue;
} else {
value[i] = firstValue + i;
}
}

T *dptr{sycl::malloc_device<T>(NumElements, q)};
for (size_t repeat{0}; repeat < NumRepeats; ++repeat) {
q.fill(dptr, value, NumElements).wait();
}

std::array<T, NumElements> host{};
q.copy<T>(dptr, host.data(), NumElements).wait();
bool pass{true};
for (size_t i{0}; i < NumElements; ++i) {
for (size_t j{0}; j < PatternSize; ++j) {
if (host[i][j] != value[j]) {
pass = false;
}
}
}
sycl::free(dptr, q);

if (!pass || verbose) {
printf("Pattern size %3zu bytes, %s values (initial %3u) %s\n", PatternSize,
(SameValue ? " equal" : "varied"), firstValue,
(pass ? "== PASS ==" : "== FAIL =="));
}

return !pass;
}

template <size_t Size> int testOneSize(sycl::queue &q) {
return test<Size, true>(q, 0) + test<Size, false>(q, 42);
}

template <size_t... Sizes>
int testSizes(sycl::queue &q, std::index_sequence<Sizes...>) {
return (testOneSize<1u + Sizes>(q) + ...);
}

int main() {
sycl::queue q{};
int failures = testSizes(q, std::make_index_sequence<MaxPatternSize>{});
if (failures > 0) {
printf("%d / %zu tests failed\n", failures, 2u * MaxPatternSize);
} else {
printf("All %zu tests passed\n", 2u * MaxPatternSize);
}
return failures;
}