File: omp_pinned.c

package info (click to toggle)
llvm-toolchain-17 1%3A17.0.6-22
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,799,624 kB
  • sloc: cpp: 6,428,607; ansic: 1,383,196; asm: 793,408; python: 223,504; objc: 75,364; f90: 60,502; lisp: 33,869; pascal: 15,282; sh: 9,684; perl: 7,453; ml: 4,937; awk: 3,523; makefile: 2,889; javascript: 2,149; xml: 888; fortran: 619; cs: 573
file content (29 lines) | stat: -rw-r--r-- 791 bytes parent folder | download | duplicates (12) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
// RUN: %libomp-compile-and-run
// RUN: env OMP_ALLOCATOR=omp_default_mem_space:pinned=true %libomp-run

#include <omp.h>

int main() {
  omp_alloctrait_t pinned_trait[1] = {{omp_atk_pinned, omp_atv_true}};
  omp_allocator_handle_t pinned_alloc =
      omp_init_allocator(omp_default_mem_space, 1, pinned_trait);
  omp_allocator_handle_t default_alloc = omp_get_default_allocator();
  double *a = (double *)omp_alloc(10 * sizeof(double), pinned_alloc);
  double *b = (double *)omp_alloc(10 * sizeof(double), default_alloc);

  if (!a || !b) return 1;

#pragma omp parallel for
  for (int i = 0; i < 10; i++) {
    a[i] = 0;
    b[i] = 1;
  }

  for (int i = 0; i < 10; i++)
    if (a[i] != 0 || b[i] != 1) return 1;

  omp_free(a, pinned_alloc);
  omp_free(b, default_alloc);

  return 0;
}