Halide 19.0.0
Halide compiler and libraries
Loading...
Searching...
No Matches
EliminateBoolVectors.h
Go to the documentation of this file.
1#ifndef HALIDE_IR_ELIMINATE_BOOL_VECTORS_H
2#define HALIDE_IR_ELIMINATE_BOOL_VECTORS_H
3
4/** \file
5 * Method to eliminate vectors of booleans from IR.
6 */
7
8#include "Expr.h"
9
10namespace Halide {
11namespace Internal {
12
13/** Some targets treat vectors of bools as integers of the same type that the
14 * boolean operation is being used to operate on. For example, instead of
15 * select(i1x8, u16x8, u16x8), the target would prefer to see select(u16x8,
16 * u16x8, u16x8), where the first argument is a vector of integers representing
17 * a mask. This pass converts vectors of bools to vectors of integers to meet
18 * this requirement. This is done by injecting intrinsics to convert bools to
19 * architecture-specific masks, and using a select_mask intrinsic instead of a
20 * Select node. This also converts any intrinsics that operate on vectorized
21 * conditions to a *_mask equivalent (if_then_else, require). Because the masks
22 * are architecture specific, they may not be stored or loaded. On Stores, the
23 * masks are converted to UInt(8) with a value of 0 or 1, which is our canonical
24 * in-memory representation of a bool. */
25///@{
28///@}
29
30/** If a type is a boolean vector, find the type that it has been
31 * changed to by eliminate_bool_vectors. */
32inline Type eliminated_bool_type(Type bool_type, Type other_type) {
33 if (bool_type.is_vector() && bool_type.bits() == 1) {
34 bool_type = bool_type.with_code(Type::Int).with_bits(other_type.bits());
35 }
36 return bool_type;
37}
38
39} // namespace Internal
40} // namespace Halide
41
42#endif
Base classes for Halide expressions (Halide::Expr) and statements (Halide::Internal::Stmt)
Type eliminated_bool_type(Type bool_type, Type other_type)
If a type is a boolean vector, find the type that it has been changed to by eliminate_bool_vectors.
Stmt eliminate_bool_vectors(const Stmt &s)
Some targets treat vectors of bools as integers of the same type that the boolean operation is being ...
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
@ Internal
Not visible externally, similar to 'static' linkage in C.
A fragment of Halide syntax.
Definition Expr.h:258
A reference-counted handle to a statement node.
Definition Expr.h:427
Types in the halide type system.
Definition Type.h:283
static const halide_type_code_t Int
Aliases for halide_type_code_t values for legacy compatibility and to match the Halide internal C++ s...
Definition Type.h:291
Type with_bits(int new_bits) const
Return Type with same type code and lanes, but new_bits for the number of bits.
Definition Type.h:366
HALIDE_ALWAYS_INLINE int bits() const
Return the bit size of a single element of this type.
Definition Type.h:349
HALIDE_ALWAYS_INLINE bool is_vector() const
Is this type a vector type? (lanes() != 1).
Definition Type.h:410
Type with_code(halide_type_code_t new_code) const
Return Type with same number of bits and lanes, but new_code for a type code.
Definition Type.h:360