https://github.com/halide/Halide
Raw File
Tip revision: 0eec1e95a0f7bec4a5b9e77f561c6211e2c8f4ef authored by Steven Johnson on 09 September 2022, 18:33:50 UTC
Merge branch 'main' into gpu_canon_fix
Tip revision: 0eec1e9
CanonicalizeGPUVars.h
#ifndef HALIDE_CANONICALIZE_GPU_VARS_H
#define HALIDE_CANONICALIZE_GPU_VARS_H

/** \file
 * Defines the lowering pass that canonicalize the GPU var names over.
 */

#include "Expr.h"

namespace Halide {
namespace Internal {

/** Canonicalize GPU var names into some pre-determined block/thread names
 * (i.e. __block_id_x, __thread_id_x, etc.). The x/y/z/w order is determined
 * by the nesting order: innermost is assigned to x and so on. */
Stmt canonicalize_gpu_vars(Stmt s);

}  // namespace Internal
}  // namespace Halide

#endif
back to top