Skip to content

Instantly share code, notes, and snippets.

@mebjas

mebjas/halide_vectorisation_comparision.cc

Created July 11, 2022 07:43
Show Gist options
  • Save mebjas/cd68f04023175461803d3a75116d3d81 to your computer and use it in GitHub Desktop.
Save mebjas/cd68f04023175461803d3a75116d3d81 to your computer and use it in GitHub Desktop.
Download ZIP
Example of schedule with vectorisation vs without
Raw
halide_vectorisation_comparision.cc
// Algorithm
output_(x, y) = u8(u16(x + y) % 256);
// Schedule 1 without vectorisation
output_.compute_root().vectorise(x).reorder(x, y);
// Schedule 2 with vectorisation
output_.compute_root().reorder(x, y);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment