starpu-max/gcc-plugin/tests/acquire.c

/* GCC-StarPU
   Copyright (C) 2011 INRIA

   GCC-StarPU is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   GCC-StarPU is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with GCC-StarPU.  If not, see <http://www.gnu.org/licenses/>.  */

/* Test whether `#pragma starpu acquire ...' generates the right code.  */

#undef NDEBUG

#include <mocks.h>

static void
foo (char *x, int foo)
{
  expected_acquire_arguments.pointer = x;
#pragma starpu acquire x
}


int
main (int argc, char *argv[])
{
#pragma starpu initialize

  int x[123];
  static char z[345];

  expected_register_arguments.pointer = x;
  expected_register_arguments.elements = 123;
  expected_register_arguments.element_size = sizeof x[0];
#pragma starpu register x

  expected_acquire_arguments.pointer = x;
#pragma starpu acquire x

  expected_register_arguments.pointer = z;
  expected_register_arguments.elements = sizeof z;
  expected_register_arguments.element_size = sizeof z[0];
#pragma starpu register z

  expected_acquire_arguments.pointer = z;
#pragma starpu acquire z

  foo (z, 345);

  assert (data_register_calls == 2);
  assert (data_acquire_calls == 3);

  return EXIT_SUCCESS;
}