Revision 6f5dc6bec58bf0a6305113166bfdcfeb80af3c5f authored by Steven Johnson on 07 December 2020, 19:27:17 UTC, committed by Steven Johnson on 07 December 2020, 19:27:17 UTC
Currently, it can contain garbage after parsing
1 parent 4f554a7
RemoveExternLoops.cpp
#include "RemoveExternLoops.h"
#include "IRMutator.h"
namespace Halide {
namespace Internal {
class RemoveExternLoops : public IRMutator {
private:
using IRMutator::visit;
Stmt visit(const For *op) override {
if (op->for_type != ForType::Extern) {
return IRMutator::visit(op);
}
// Replace the for with its first iteration (implemented with a let).
return LetStmt::make(op->name, op->min, mutate(op->body));
}
};
Stmt remove_extern_loops(const Stmt &s) {
return RemoveExternLoops().mutate(s);
}
} // namespace Internal
} // namespace Halide
Computing file changes ...
