image-match/python/py/Lib/site-packages/torch/include/ATen/xpu/XPUUtils.h

#if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
#pragma once

#include <ATen/xpu/XPUContext.h>

namespace at::xpu {

// Check if every tensor in a list of tensors matches the current device.
inline bool check_device(ArrayRef<Tensor> ts) {
  if (ts.empty()) {
    return true;
  }
  Device curDevice = Device(kXPU, current_device());
  for (const Tensor& t : ts) {
    if (t.device() != curDevice) {
      return false;
    }
  }
  return true;
}

} // namespace at::xpu

#else
#error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
#endif  // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)