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
|
#include "testsuite.h"
#include <blitz/array.h>
using namespace blitz;
using namespace blitz::tensor;
int main()
{
Array<float,2> A(4,4);
Array<float,1> B(4), C(4);
A = 0, 1, 2, 3,
30, 2, 4, 1,
3, 2, 5, 6,
12, 9, 0, 0;
B = sum(pow2(A),j);
C = 14, 921, 74, 225;
BZTEST(count(C == B) == 4);
// workaround for MIPSpro compiler bug
#if defined(__sgi)
TinyVector<int,1> N = maxIndex(B);
#else
TinyVector<int,1> N = maxIndex(sum(pow2(A),j));
#endif
BZTEST(N(0) == 1);
return 0;
}
|
