forked from OSchip/llvm-project
65 lines
1.4 KiB
C
65 lines
1.4 KiB
C
// RUN: %libomp-compile-and-run
|
|
//
|
|
// The test checks the teams construct pseudocode executed on host
|
|
//
|
|
|
|
#include <stdio.h>
|
|
#include <omp.h>
|
|
|
|
#ifndef N_TEAMS
|
|
#define N_TEAMS 4
|
|
#endif
|
|
#ifndef N_THR
|
|
#define N_THR 3
|
|
#endif
|
|
|
|
static int err = 0;
|
|
|
|
// Internal library staff to emulate compiler's code generation:
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
typedef struct {
|
|
int reserved_1;
|
|
int flags;
|
|
int reserved_2;
|
|
int reserved_3;
|
|
char *psource;
|
|
} ident_t;
|
|
|
|
static ident_t dummy_loc = {0, 2, 0, 0, ";dummyFile;dummyFunc;0;0;;"};
|
|
|
|
int __kmpc_global_thread_num(void*);
|
|
void __kmpc_push_num_teams(ident_t const*, int, int, int);
|
|
void __kmpc_fork_teams(ident_t const*, int argc, void *microtask, ...);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
// Outlined entry point:
|
|
void foo(int *gtid, int *tid, int *nt)
|
|
{ // start "serial" execution by master threads of each team
|
|
if ( nt ) {
|
|
printf(" team %d, param %d\n", omp_get_team_num(), *nt);
|
|
} else {
|
|
printf("ERROR: teams before parallel: gtid, tid: %d %d, bad pointer: %p\n", *gtid, *tid, nt);
|
|
err++;
|
|
return;
|
|
}
|
|
}
|
|
|
|
int main()
|
|
{
|
|
int nt = 4;
|
|
int th = __kmpc_global_thread_num(NULL); // registers initial thread
|
|
__kmpc_push_num_teams(&dummy_loc, th, N_TEAMS, N_THR);
|
|
__kmpc_fork_teams(&dummy_loc, 1, &foo, &nt); // pass 1 shared parameter "nt"
|
|
if (err)
|
|
printf("failed with %d errors\n",err);
|
|
else
|
|
printf("passed\n");
|
|
return err;
|
|
}
|