https://github.com/halide/Halide
Raw File
Tip revision: 7d7dca76d0674d31d12c1ad4ea31ea0fb32b2fd4 authored by Steven Johnson on 14 September 2022, 17:42:01 UTC
Revert "[HVX] Simplify constant factor before distributing (#7009)"
Tip revision: 7d7dca7
LowerWarpShuffles.h
#ifndef HALIDE_LOWER_WARP_SHUFFLES_H
#define HALIDE_LOWER_WARP_SHUFFLES_H

/** \file
 * Defines the lowering pass that injects CUDA warp shuffle
 * instructions to access storage outside of a GPULane loop.
 */

#include "Expr.h"

namespace Halide {
namespace Internal {

/** Rewrite access to things stored outside the loop over GPU lanes to
 * use nvidia's warp shuffle instructions. */
Stmt lower_warp_shuffles(Stmt s, const Target &t);

}  // namespace Internal
}  // namespace Halide

#endif
back to top