Revision 288526cf7500f496c36eb0e74500e4fd4547b6fb authored by Dillon Sharlet on 28 January 2021, 17:52:16 UTC, committed by GitHub on 28 January 2021, 17:52:16 UTC
* Encapsulate more symbols.
1 parent f427ad1
FuzzFloatStores.cpp
#include "FuzzFloatStores.h"
#include "IRMutator.h"
#include "IROperator.h"
namespace Halide {
namespace Internal {
namespace {
class FuzzFloatStores : public IRMutator {
using IRMutator::visit;
Stmt visit(const Store *op) override {
Type t = op->value.type();
if (t.is_float()) {
// Drop the last bit of the mantissa.
Expr value = op->value;
Expr mask = make_one(t.with_code(Type::UInt));
value = reinterpret(mask.type(), value);
value = value & ~mask;
value = reinterpret(t, value);
return Store::make(op->name, value, op->index, op->param, op->predicate, op->alignment);
} else {
return IRMutator::visit(op);
}
}
};
} // namespace
Stmt fuzz_float_stores(const Stmt &s) {
return FuzzFloatStores().mutate(s);
}
} // namespace Internal
} // namespace Halide
Computing file changes ...