RFR: 8264409: AArch64: generate better code for Vector API allTrue

Pengfei Li pli at openjdk.java.net
Fri Apr 2 02:30:24 UTC 2021


On Thu, 1 Apr 2021 07:58:07 GMT, Ningsheng Jian <njian at openjdk.org> wrote:

> In Vector API NEON implementation, we use a vector register to represent vector mask, where an element value of -1 is a true mask and an element value of 0 is a false mask. The allTrue() api is used to check whether all the elements of current mask are set.
> 
> Currently, the AArch64 NEON allTrue implementation looks like:
> 
>   andr  $tmp, T16B $src1, $src2\t# src2 is maskAllTrue
>   notr  $tmp, T16B, $tmp
>   addv  $tmp, T16B, $tmp
>   umov  $dst, $tmp, B, 0
>   cmp   $dst, 0
>   cset  $dst
> 
> where $src2 is a preset all true (-1) constant. We can optimize it to the code sequence like below, to check whether all bits are set:
> 
>   uminv $tmp, T16B, $src1
>   umov  $dst, $tmp, B, 0
>   cmp   $dst, 0xff
>   cset  $dst
> 
> With this codegen improvement, we can see about 8%~70% performance uplift on different machines for Alibaba's Vector API bigdata benchmarks [1][2].
> 
> Tested with tier1 and vector api jtreg tests.
> 
> [1] https://github.com/openjdk/panama-vector/blob/vectorIntrinsics/test/jdk/jdk/incubator/vector/benchmark/src/main/java/benchmark/bigdata/BooleanArrayCheck.java#L61
> [2] https://github.com/openjdk/panama-vector/blob/vectorIntrinsics/test/jdk/jdk/incubator/vector/benchmark/src/main/java/benchmark/bigdata/ValueRangeCheckAndCastL2I.java#L93

Overall looks good to me. (not a reviewer)

src/hotspot/cpu/aarch64/aarch64_neon.ad line 3571:

> 3569:   format %{ "uminv $tmp, T8B, $src1\n\t"
> 3570:             "umov  $dst, $tmp, B, 0\n\t"
> 3571:             "cmp   $dst, 0xff\n\t"

I think we should write "#0xff" here. But it looks that all other immediates in format field of aarch64_neon.ad lose the number sign as well.

-------------

Marked as reviewed by pli (Committer).

PR: https://git.openjdk.java.net/jdk/pull/3302


More information about the hotspot-compiler-dev mailing list