Revision 2c48ba87ac173341f641479cb711f6f755d48e93 authored by Volodymyr Kysenko on 08 December 2023, 21:58:37 UTC, committed by Volodymyr Kysenko on 08 December 2023, 21:58:37 UTC
1 parent d84e3a6
RemoveExternLoops.cpp
#include "RemoveExternLoops.h"
#include "IRMutator.h"
namespace Halide {
namespace Internal {
namespace {
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));
}
};
} // namespace
Stmt remove_extern_loops(const Stmt &s) {
return RemoveExternLoops().mutate(s);
}
} // namespace Internal
} // namespace Halide
Computing file changes ...