1 #include "Argument.h"
2
3 namespace Halide {
4
operator ==(const ArgumentEstimates & rhs) const5 bool ArgumentEstimates::operator==(const ArgumentEstimates &rhs) const {
6 if (buffer_estimates.size() != rhs.buffer_estimates.size()) {
7 return false;
8 }
9 for (size_t i = 0; i < buffer_estimates.size(); ++i) {
10 if (!buffer_estimates[i].min.same_as(rhs.buffer_estimates[i].min) ||
11 !buffer_estimates[i].extent.same_as(rhs.buffer_estimates[i].extent)) {
12 return false;
13 }
14 }
15 return scalar_def.same_as(rhs.scalar_def) &&
16 scalar_min.same_as(rhs.scalar_min) &&
17 scalar_max.same_as(rhs.scalar_max) &&
18 scalar_estimate.same_as(rhs.scalar_estimate);
19 }
20
Argument(const std::string & _name,Kind _kind,const Type & _type,int _dimensions,const ArgumentEstimates & argument_estimates)21 Argument::Argument(const std::string &_name, Kind _kind, const Type &_type, int _dimensions, const ArgumentEstimates &argument_estimates)
22 : name(_name), kind(_kind), dimensions((uint8_t)_dimensions), type(_type), argument_estimates(argument_estimates) {
23 internal_assert(dimensions >= 0 && dimensions <= 255);
24 user_assert(!(is_scalar() && dimensions != 0)) << "Scalar Arguments must specify dimensions of 0";
25 user_assert(argument_estimates.buffer_estimates.empty() ||
26 argument_estimates.buffer_estimates.size() == dimensions)
27 << "buffer_estimates must match dimensionality for Argument " << name;
28 }
29
30 } // namespace Halide
31