halide 是否支持带有 neon 的 ARMv8(aarch64)？

Question

我想将 Halide 用于带有霓虹灯矢量化的 ARM A53(aarch64) 目标。

但我不知道如何创建 Target 对象。此外，我在 Target.h.

中找不到具有霓虹功能的 aarch64 目标

我测试过的以下代码在 A53 目标上运行，但生成的代码不包含 neon 指令。

Target target("arm-64-linux"); // is it right?
Buffer<uint16_t> input(640,480);

Var x,y;
Func brighter("brighter");
brighter(x,y) = input(x,y) + 100;
brighter.estimate(x, 0, 640).
         estimate(y, 0, 480);

Pipeline p(brighter);
p.auto_schedule(target);
p.compile_to_static_library("./lib_dummy", {input}, "", target);

Answer 1

arm-64 是 Halide 用于 aarch64 的，所以你的目标很好。要使用 neon 指令，您需要对某些内容进行矢量化。不确定自动调度程序是否正在这样做（应该是！）。尽量不要自动安排，而只是说：

brighter.vectorize(x, 8);

halide 是否支持带有 neon 的 ARMv8(aarch64)？

Does halide support ARMv8(aarch64) with neon?

arm64

neon

halide