#pragma once#include <torch/extension.h>#define CHECK_CPU(x) AT_ASSERTM(x.device().is_cpu(), #x " must be CPU tensor")#define CHECK_INPUT(x) AT_ASSERTM(x, "Input mismatch")#define CHECK_CONTIGUOUS(x) \ AT_ASSERTM(x.is_contiguous(), #x " must be contiguous")
