/* { dg-do compile } */
/* { dg-options "-O2 -ftree-vectorize -msve-vector-bits=scalable" } */

#include <stdint.h>

void consumer (void *);

#define TEST_LOOP(TYPE)				\
  void						\
  multi_loop_##TYPE (TYPE *x, TYPE val)		\
  {						\
    for (int i = 0; i < 9; ++i)			\
      x[i] += val;				\
    consumer (x);				\
    for (int i = 0; i < 9; ++i)			\
      x[i] += val;				\
    consumer (x);				\
    for (int i = 0; i < 9; ++i)			\
      x[i] += val;				\
    consumer (x);				\
  }

/* One iteration is enough.  */
TEST_LOOP (uint8_t);
/* Two iterations are enough.  We specialize the second two loops based
   on whether the first executes once or twice.  */
TEST_LOOP (uint16_t);
/* Three iterations are needed; ought to stay a loop.  */
TEST_LOOP (uint32_t);
/* Five iterations are needed; ought to stay a loop.  */
TEST_LOOP (uint64_t);

/* { dg-final { scan-assembler-times {\twhilelo\tp[0-9]\.b} 3 } } */
/* { dg-final { scan-assembler-times {\twhilelo\tp[0-9]\.h} 8 } } */
/* { dg-final { scan-assembler-times {\twhilelo\tp[0-9]\.s} 6 } } */
/* { dg-final { scan-assembler-times {\twhilelo\tp[0-9]\.d} 6 } } */
/* { dg-final { scan-assembler-not {\tldr\tz[0-9]} } } */
/* { dg-final { scan-assembler-not {\tstr\tz[0-9]} } } */
/* { dg-final { scan-assembler-not {\tldr\tp[0-9]} } } */
/* { dg-final { scan-assembler-not {\tstr\tp[0-9]} } } */