#ifndef HALIDE_INFER_ARGUMENTS_H
#define HALIDE_INFER_ARGUMENTS_H
#include <vector>
#include "Argument.h"
#include "Buffer.h"
#include "Expr.h"
#include "Parameter.h"
namespace Halide {
namespace Internal {
struct InferredArgument {
Argument arg;
Parameter param;
Buffer<> buffer;
bool operator<(const InferredArgument &other) const {
if (arg.is_buffer() && !other.arg.is_buffer()) {
return true;
} else if (other.arg.is_buffer() && !arg.is_buffer()) {
return false;
} else {
return arg.name < other.arg.name;
}
}
};
class Function;
std::vector<InferredArgument> infer_arguments(Stmt body, const std::vector<Function> &outputs);
}
}
#endif