DidaWiki

Questa pagina è in sola lettura. Puoi visualizzare il sorgente, ma non puoi modificarlo. Contatta l'amministratore se pensi che ci sia un errore.
<code c++ sbil_ff.cpp>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>

#include <omp.h>
#include "ff/farm.hpp"

using namespace std;
using namespace ff; 

float **a; 
int     n; 

float calc(float x) {
  long iter = (long) x; 
  for(long i=0; i<iter; i++) 
    x = sin(x); 
  return(x);
}

class Worker : public ff_node {
  void * svc(void * t) {

    float * task = (float *) t;
    float * sum = new float;

    for(int i=0; i<n; i++) {
      task[i] = calc(task[i]);
      *sum += task[i]; 
    }
    return ((void *) sum); 
  }
};

class Emitter : public ff_node {
  void * svc(void * t) {
    a = new float * [n];
    for(int i=0; i<n; i++) 
      a[i] = new float[n];
    
    for(int i=0; i<n; i++) 
      for(int j=0; j<n; j++) 
	a[i][j] = i* 100 + j * 10; 
    
    for(int i=0; i<n; i++) 
      ff_send_out((void *) a[i]);
    
    return(EOS); 
  }
};


class Collector : public ff_node {
private: 
  float sum; 
public: 

  int svc_init() {
    sum = 0.0; 
    return(0); 
  }

  void svc_end() {
    cout << "sum is " << sum << endl; 
    return; 
  }

  void * svc(void * t) {
    float * x = ((float *) t) ;
    sum += *x; 
    return GO_ON; 
  }
};

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

  if(argc == 1) {
    cout << "Usage is:\n" << argv[0] << " veclen numthread " << endl; 
    return(0);
  }

  n   = atoi(argv[1]); 
  int nw; 
  float tc1; 

  if(argc == 3) {
    nw  = atoi(argv[2]); 
  } 
  if(argc == 4) {
    nw = atoi(argv[3]); 
    tc1 = atof(argv[2]); 
  }
  

  float sum = 0.0;
  ff_farm<> farm; 

  vector<ff_node *> workers;
  for(int i=0;i<nw;++i) 
    workers.push_back(new Worker);
  farm.add_workers(workers); 
  farm.add_collector(new Collector); 
  farm.add_emitter(new Emitter); 

  float ts0 = omp_get_wtime();
  farm.run_and_wait_end();
  float ts1 = omp_get_wtime();
  float tc = ts1 - ts0; 

  cout << tc << " " ;
  if(argc==4) {
    cout << " sp = " << tc1 / tc;
  } 
  cout << endl; 
  return(0); 

}
</code>