Revision 9a94756d01d9071ff1610bfc4cb170bd47f701a8 authored by Alexander Root on 21 July 2022, 15:01:16 UTC, committed by GitHub on 21 July 2022, 15:01:16 UTC
* use pmaddubsw 8-bit horizontal widening adds * add SSE3 versions too * add pmaddubsw tests
1 parent 967c3bf
ConciseCasts.h
#ifndef HALIDE_CONCISE_CASTS_H
#define HALIDE_CONCISE_CASTS_H
#include "IROperator.h"
/** \file
*
* Defines concise cast and saturating cast operators to make it
* easier to read cast-heavy code. Think carefully about the
* readability implications before using these. They could make your
* code better or worse. Often it's better to add extra Funcs to your
* pipeline that do the upcasting and downcasting.
*/
namespace Halide {
namespace ConciseCasts {
inline Expr f64(Expr e) {
Type t = Float(64, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr f32(Expr e) {
Type t = Float(32, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr f16(Expr e) {
Type t = Float(16, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr bf16(Expr e) {
Type t = BFloat(16, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr i64(Expr e) {
Type t = Int(64, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr i32(Expr e) {
Type t = Int(32, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr i16(Expr e) {
Type t = Int(16, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr i8(Expr e) {
Type t = Int(8, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr u64(Expr e) {
Type t = UInt(64, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr u32(Expr e) {
Type t = UInt(32, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr u16(Expr e) {
Type t = UInt(16, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr u8(Expr e) {
Type t = UInt(8, e.type().lanes());
return cast(t, std::move(e));
}
inline Expr i8_sat(Expr e) {
Type t = Int(8, e.type().lanes());
return saturating_cast(t, std::move(e));
}
inline Expr u8_sat(Expr e) {
Type t = UInt(8, e.type().lanes());
return saturating_cast(t, std::move(e));
}
inline Expr i16_sat(Expr e) {
Type t = Int(16, e.type().lanes());
return saturating_cast(t, std::move(e));
}
inline Expr u16_sat(Expr e) {
Type t = UInt(16, e.type().lanes());
return saturating_cast(t, std::move(e));
}
inline Expr i32_sat(Expr e) {
Type t = Int(32, e.type().lanes());
return saturating_cast(t, std::move(e));
}
inline Expr u32_sat(Expr e) {
Type t = UInt(32, e.type().lanes());
return saturating_cast(t, std::move(e));
}
inline Expr i64_sat(Expr e) {
Type t = Int(64, e.type().lanes());
return saturating_cast(t, std::move(e));
}
inline Expr u64_sat(Expr e) {
Type t = UInt(64, e.type().lanes());
return saturating_cast(t, std::move(e));
}
}; // namespace ConciseCasts
}; // namespace Halide
#endif
Computing file changes ...