gcc/testsuite/gcc.target/aarch64/sve/popcount_1_run.c

/* { dg-do run { target aarch64_sve_hw } } */
/* { dg-options "-O2 -ftree-vectorize" } */

#include "popcount_1.c"

extern void abort (void) __attribute__ ((noreturn));

unsigned int data[] = {
  0x11111100, 6,
  0xe0e0f0f0, 14,
  0x9900aab3, 13,
  0x00040003, 3,
  0x000e000c, 5,
  0x22227777, 16,
  0x12341234, 10,
  0x0, 0
};

int __attribute__ ((optimize (1)))
main (void)
{
  unsigned int count = sizeof (data) / sizeof (data[0]) / 2;

  uint32_t in32[count];
  unsigned int out32[count];
  for (unsigned int i = 0; i < count; ++i)
    {
      in32[i] = data[i * 2];
      asm volatile ("" ::: "memory");
    }
  popcount_32 (out32, in32, count);
  for (unsigned int i = 0; i < count; ++i)
    if (out32[i] != data[i * 2 + 1])
      abort ();

  count /= 2;
  uint64_t in64[count];
  unsigned int out64[count];
  for (unsigned int i = 0; i < count; ++i)
    {
      in64[i] = ((uint64_t) data[i * 4] << 32) | data[i * 4 + 2];
      asm volatile ("" ::: "memory");
    }
  popcount_64 (out64, in64, count);
  for (unsigned int i = 0; i < count; ++i)
    if (out64[i] != data[i * 4 + 1] + data[i * 4 + 3])
      abort ();

  return 0;
}