teststep.c

来自「shpf 1.9一个并行编译器」· C语言 代码 · 共 51 行

#include <stdio.h>

#include "ad++.h"
#include "admacros.h"


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

  const int n = 7 ;

  Procs2 p(2, 3) ;
  Range x(n, p.dim(0), STP) ;
  Range y(n, p.dim(1), STP) ;
  Array2<float> a(p, x, y), b(p, y, x) ;

  on(p) {

    where(x) {
      where(y) {
        a(x, y) = 10 * (1 + x) + 1 + y ;
      } erewh(y) ;
    } erewh(x) ;

    Array2<float> tmp1(x, y) ;
    where(x) {
      where(y) {
        tmp1(x, y) = a(x, y) + 2 ;
      } erewh(y) ;
    } erewh(x) ;

    remap(b, tmp1) ;

  } no(p) ;

  // output

  gprintf("a\n") ;
  aprintf("%7.2f%*N", a, y.size()) ;


  gprintf("b\n") ;
  aprintf("%7.2f%*N", b, x.size()) ;

  AdlibFinalize() ;
}