/* { dg-do run } */
/* { dg-additional-options "-foffload-memory=pinned" } */

/* { dg-skip-if "Pinning not implemented on this host" { ! *-*-linux-gnu* } } */

/* Test that -foffload-memory=pinned works.  */

#include <stdio.h>
#include <stdlib.h>

#ifdef __linux__
#include <sys/types.h>
#include <unistd.h>

#include <sys/mman.h>

int
get_pinned_mem ()
{
  int pid = getpid ();
  char buf[100];
  sprintf (buf, "/proc/%d/status", pid);

  FILE *proc = fopen (buf, "r");
  if (!proc)
    abort ();
  while (fgets (buf, 100, proc))
    {
      int val;
      if (sscanf (buf, "VmLck: %d", &val))
	{
	  fclose (proc);
	  return val;
	}
    }
  abort ();
}
#else
int
get_pinned_mem ()
{
  return 0;
}

#define mlockall(...) 0
#endif

#include <omp.h>

int
main ()
{
  // Sanity check
  if (get_pinned_mem () == 0)
    {
      /* -foffload-memory=pinned has failed, but maybe that's because
	 isufficient pinned memory was available.  */
      if (mlockall (MCL_CURRENT | MCL_FUTURE) == 0)
	abort ();
    }

  return 0;
}