22 lines
466 B
C
22 lines
466 B
C
#include <gtest/gtest.h>
|
|
|
|
#include <c10/util/irange.h>
|
|
|
|
static inline void initHostData(int* hostData, int numel) {
|
|
for (const auto i : c10::irange(numel)) {
|
|
hostData[i] = i;
|
|
}
|
|
}
|
|
|
|
static inline void clearHostData(int* hostData, int numel) {
|
|
for (const auto i : c10::irange(numel)) {
|
|
hostData[i] = 0;
|
|
}
|
|
}
|
|
|
|
static inline void validateHostData(int* hostData, int numel) {
|
|
for (const auto i : c10::irange(numel)) {
|
|
EXPECT_EQ(hostData[i], i);
|
|
}
|
|
}
|