/* 

computearay.c 

compute a[i] = myfun(i); 
 */

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

#include <omp.h> 

#define NUM_THREADS 4


int myfun(int i)  {

  //or something else
  return i; 
}



int main(int argc, char** argv) {

  if (argc < 3) {

    printf("usage: %s n nthreads\n", argv[0]);
    exit(1); 
  }
  long n  =  atol(argv[1]); 
  int nthreads = atol(argv[2]); 


  int* a = malloc(n *sizeof(int));
  assert(a); 
  long  i; 
  double t1, t2; 

  t1 = omp_get_wtime(); 


  omp_set_num_threads(nthreads); 
#pragma omp parallel  private (i)
  { 
#pragma omp for  schedule(static,128*1024)
    for (i=0; i<n; i++) {
      a[i]  =  myfun(i); 
    }//for 
  } //end parallel  
				      
  t2 = omp_get_wtime();   
 
  printf("Done. time = %.1f\n",  t2-t1);
 
#if(0)
  for (i=0; i<n; i++) {
    printf("%d ", a[i]);    
  }
  printf("\n");
#endif
  return 0; 
}