xref: /aosp_15_r20/external/pytorch/c10/xpu/test/impl/XPUTest.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

#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);
  }
}