This source file includes following definitions.
- visit
- fuzz_float_stores
#include "FuzzFloatStores.h"
#include "IRMutator.h"
#include "IROperator.h"
namespace Halide {
namespace Internal {
namespace {
class FuzzFloatStores : public IRMutator {
using IRMutator::visit;
void visit(const Store *op) {
Type t = op->value.type();
if (t.is_float()) {
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);
stmt = Store::make(op->name, value, op->index, op->param, op->predicate);
} else {
IRMutator::visit(op);
}
}
};
}
Stmt fuzz_float_stores(Stmt s) {
return FuzzFloatStores().mutate(s);
}
}
}