Function check_tensor

examples/cuda_cpp/rmsnorm/main.cpp:7–11

Helper function to check common tensor properties

// Helper function to check common tensor properties
void check_tensor(const torch::Tensor& tensor, const std::string& name) {
    TORCH_CHECK(tensor.is_cuda(), name, " must be a CUDA tensor");
    TORCH_CHECK(tensor.dtype() == torch::kBFloat16, name, " must have bfloat16 dtype");
    TORCH_CHECK(tensor.is_contiguous(), name, " must be contiguous");
}

/**
 * @brief Python-bindable 'run' function for RMSNorm.