source: doc/theses/mike_brooks_MMath/programs/array-boundcheck-removal-stdvec.cpp @ df2e00f

/*
// traditional "naiive" loops
forall( [M], [N], [P] )
void matmul( array(float, M, P) & src1,
                         array(float, P, N) & src2, 
                         array(float, M, N) & tgt ) {
        for (i; M) for (j; N) {
                tgt[i][j] = 0;
                for (k; P)
                        tgt[i][j] += src1[i][k] * src2[k][j];
        }
}
*/

#if defined BADBOUNDS
#define BOUND(X) 17
#else
#define BOUND(X) X
#endif

#include <vector>
using namespace std;


#ifndef TESTCASE
#define TESTCASE 1
#endif

#if TESTCASE==1

float f( vector<float> & a ) {
        float result = 0;
        for( int i = 0; i < BOUND(a.size()); i++ ) {
                result += a.at(i);
                // hillarious that, while writing THIS VERY DEMO, on first go, I actaully wrote it a[i]
        }
        return result;
}

#ifdef WITHHARNESS
#include <iostream>
int main( int argc, char ** argv ) {
        vector<float> v(5);
        v.at(0) = 3.14;
        v.at(1) = 3.14;
        v.at(2) = 3.14;
        v.at(3) = 3.14;
        v.at(4) = 3.14;

        float answer = f(v);

        cout << "answer: " << answer << endl;

}
#endif 

// g++ array-boundcheck-removal-stdvec.cpp -DWITHHARNESS -DBADBOUNDS
// ./a.out
// Aborted: terminate called after throwing an instance of 'std::out_of_range'


// g++ array-boundcheck-removal-stdvec.cpp -DWITHHARNESS
// ./a.out
// answer: 15.7

// g++ array-boundcheck-removal-stdvec.cpp -S -O2 -DBADBOUNDS
// loop has cmp-jmp
// jmp target has call _ZSt24__throw_out_of_range_fmtPKcz@PLT

// g++ array-boundcheck-removal-stdvec.cpp -S -O2
// loop is clean


#elif TESTCASE==2

//#include <cassert>
#define assert(prop) if (!prop) return

typedef vector<vector<float> > mat;

void matmul( mat & a, mat & b, mat & rslt ) {
        size_t m = rslt.size();
        assert( m == a.size() );
        size_t p = b.size();
        for ( int i = 0; i < BOUND(m); i++ ) {
                assert( p == a.at(i).size() );
                size_t n = rslt.at(i).size();
                for ( int j = 0; j < BOUND(n); j++ ) {
                        rslt.at(i).at(j) = 0.0;
                        for ( int k = 0; k < BOUND(p); k++ ) {
                                assert(b.at(k).size() == n); // asking to check it too often
                                rslt.at(i).at(j) += a.at(i).at(k) * b.at(k).at(j);
                        }
                }
        }
}

#ifdef WITHHARNESS
#include <iostream>
int main( int argc, char ** argv ) {
        mat a(5, vector<float>(6));
        mat b(6, vector<float>(7));
        mat r(5, vector<float>(7));
        matmul(a, b, r);
}
#endif 

// (modify the declarations in main to be off by one, each of the 6 values at a time)
// g++ array-boundcheck-removal-stdvec.cpp -DWITHHARNESS -DTESTCASE=2
// ./a.out
// (see an assertion failure)

// g++ array-boundcheck-removal-stdvec.cpp -DWITHHARNESS -DTESTCASE=2
// ./a.out
// (runs fine)

// g++ array-boundcheck-removal-stdvec.cpp -S -O2 -DTESTCASE=2
// hypothesis: loop bound checks are clean because the assertions took care of it
// actual so far:  both assertions and bound checks survived

#endif