Stencil_catapult.h

#ifndef STENCIL_H
#define STENCIL_H

#include <stddef.h>
#include <stdint.h>
#include <assert.h>
#include "ac_channel.h"

///Forward declaration
template <typename T, size_t EXTENT_0, size_t EXTENT_1, size_t EXTENT_2, size_t EXTENT_3> struct Stencil;
template <typename T, size_t EXTENT_0, size_t EXTENT_1, size_t EXTENT_2, size_t EXTENT_3> struct PackedStencil;
template <typename T, size_t EXTENT_0, size_t EXTENT_1, size_t EXTENT_2, size_t EXTENT_3> struct AxiPackedStencil;


#ifndef AC_INT_MAX_W
//#define AP_INT_MAX_W 32768
#define AC_INT_MAX_W 4554
//#define AP_INT_MAX_W 17280
//#define AP_INT_MAX_W 2048
#endif

#undef true
#undef false
#include "ac_int.h"

union single_cast {
  float f;
  uint32_t i;
};

union double_cast {
  float f;
  uint64_t i;
};

template<typename T>
inline T bitcast_to_uint(T in) {
  return in;
}

static inline ac_int<32, false> bitcast_to_uint(float in) {
  union single_cast temp;
  temp.f = in;
  return temp.i;
}

static inline ac_int<64, false> bitcast_to_uint(double in) {
  union double_cast temp;
  temp.f = in;
  return temp.i;
}

template<typename T, int N>
  inline void bitcast_to_type(ac_int<N, false>& in, T& out) {
  out = in;
}

static inline void bitcast_to_type(ac_int<32, false>& in, float& out) {
  union single_cast temp;
  temp.i = in;
  out = temp.f;
}

static inline void bitcast_to_type(ac_int<64, false>& in, double& out) {
  union double_cast temp;
  temp.i = in;
  out = temp.f;
}

template <typename T, size_t EXTENT_0, size_t EXTENT_1 = 1, size_t EXTENT_2 = 1, size_t EXTENT_3 = 1>
  struct PackedStencil {
    ac_int<8*sizeof(T)*EXTENT_3*EXTENT_2*EXTENT_1*EXTENT_0, false> value;

    /** writer
     */
    inline ac_int<8*sizeof(T), false>
      operator()(ac_int<sizeof(T)*8, false> set_val, size_t index_0, size_t index_1, size_t index_2, size_t index_3) {

#ifndef __SYNTHESIS__
      assert(index_0 < EXTENT_0 && index_1 < EXTENT_1 && index_2 < EXTENT_2 && index_3 < EXTENT_3);
#endif

      const size_t word_length = sizeof(T) * 8; // in bits
        const size_t lo = index_0 * word_length +
                          index_1 * EXTENT_0 * word_length +
                          index_2 * EXTENT_0 * EXTENT_1 * word_length +
	  index_3 * EXTENT_0 * EXTENT_1 * EXTENT_2 * word_length;
	value.set_slc((unsigned int)lo, set_val);
        return set_val;
    }

    /** reader
     */
    inline ac_int<8*sizeof(T), false>
      operator()(size_t index_0, size_t index_1 = 0, size_t index_2 = 0, size_t index_3 = 0) const {

#ifndef __SYNTHESIS__
      assert(index_0 < EXTENT_0 && index_1 < EXTENT_1 && index_2 < EXTENT_2 && index_3 < EXTENT_3);
#endif

      const size_t word_length = sizeof(T) * 8; // in bits
        const size_t lo = index_0 * word_length +
                          index_1 * EXTENT_0 * word_length +
                          index_2 * EXTENT_0 * EXTENT_1 * word_length +
      index_3 * EXTENT_0 * EXTENT_1 * EXTENT_2 * word_length;
      return value.template slc<word_length>((unsigned int)lo);
    }

    // convert to Stencil
    operator Stencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3>() {
      Stencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> res;
#pragma hls_unroll yes
      for(size_t idx_3 = 0; idx_3 < EXTENT_3; idx_3++)
#pragma hls_unroll yes      
        for(size_t idx_2 = 0; idx_2 < EXTENT_2; idx_2++)
#pragma hls_unroll yes        
	  for(size_t idx_1 = 0; idx_1 < EXTENT_1; idx_1++)
#pragma hls_unroll yes	  
	    for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
	      ac_int<sizeof(T) * 8, false> temp = operator()(idx_0, idx_1, idx_2, idx_3);
	      bitcast_to_type(temp, res.value[idx_3][idx_2][idx_1][idx_0]);
	    }
      return res;
    }

    // convert to AxiPackedStencil
    operator AxiPackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3>() {
      AxiPackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> res;
      res.value = value;
      return res;
    }

    // set 1st dim
   void set_dim(PackedStencil<T, EXTENT_0> set_val, size_t index_1, size_t index_2, size_t index_3) {
      const size_t word_length = sizeof(T) * 8; // in bits      
 #pragma hls_unroll yes	  
      for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
          const size_t lo = idx_0 * word_length +
                index_1 * EXTENT_0 * word_length +
                index_2 * EXTENT_0 * EXTENT_1 * word_length +
              index_3 * EXTENT_0 * EXTENT_1 * EXTENT_2 * word_length;

 	  ac_int<sizeof(T) * 8, false> temp = set_val(idx_0);
          value.set_slc((unsigned int)lo, temp);
      }
   } 

     // set 1st and 2nd dim
   void set_dim(PackedStencil<T, EXTENT_0, EXTENT_1> set_val, size_t index_2, size_t index_3) {
      const size_t word_length = sizeof(T) * 8; // in bits      
 #pragma hls_unroll yes	  
      for(size_t idx_1 = 0; idx_1 < EXTENT_1; idx_1++) 
 #pragma hls_unroll yes	  
      for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
          const size_t lo = idx_0 * word_length +
                idx_1 * EXTENT_0 * word_length +
                index_2 * EXTENT_0 * EXTENT_1 * word_length +
              index_3 * EXTENT_0 * EXTENT_1 * EXTENT_2 * word_length;

 	  ac_int<sizeof(T) * 8, false> temp = set_val(idx_0, idx_1);
          value.set_slc((unsigned int)lo, temp);
      }
   } 

 

   // get 1st dim
   PackedStencil<T, EXTENT_0> get_dim(size_t index_1, size_t index_2, size_t index_3) {
      PackedStencil<T, EXTENT_0> res;
      const size_t word_length = sizeof(T) * 8; // in bits      
 #pragma hls_unroll yes	  
      for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
        const size_t lo = idx_0 * word_length;

 	    ac_int<sizeof(T) * 8, false> temp = operator()(idx_0, index_1, index_2, index_3);
        res.value.set_slc((unsigned int)lo, temp);
      }
      return res; 
   } 


   // get 1st and 2nd dim
   PackedStencil<T, EXTENT_0, EXTENT_1> get_dim(size_t index_2, size_t index_3) {
      PackedStencil<T, EXTENT_0, EXTENT_1> res;
      const size_t word_length = sizeof(T) * 8; // in bits
 #pragma hls_unroll yes	  
      for(size_t idx_1 = 0; idx_1 < EXTENT_1; idx_1++) 
 #pragma hls_unroll yes	  
        for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
          const size_t lo = idx_0 * word_length +
                  idx_1 * EXTENT_0 * word_length;

 	  ac_int<sizeof(T) * 8, false> temp = operator()(idx_0, idx_1, index_2, index_3);
          res.value.set_slc((unsigned int)lo, temp);
      }
      return res; 
   } 

   void add(PackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> set_stencil, bool flag) {
      const size_t word_length = sizeof(T) * 8; // in bits       
#pragma hls_unroll yes
//#pragma hls_pipeline_init_interval 1
      for(size_t idx_3 = 0; idx_3 < EXTENT_3; idx_3++)
#pragma hls_unroll yes      
      for(size_t idx_2 = 0; idx_2 < EXTENT_2; idx_2++)
#pragma hls_unroll yes        
	  for(size_t idx_1 = 0; idx_1 < EXTENT_1; idx_1++)
#pragma hls_unroll yes	  
	    for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
          const size_t lo = idx_0 * word_length +
                idx_1 * EXTENT_0 * word_length +
                idx_2 * EXTENT_0 * EXTENT_1 * word_length +
                idx_3 * EXTENT_0 * EXTENT_1 * EXTENT_2 * word_length;	        
	      ac_int<sizeof(T) * 8, false> temp = operator()(idx_0, idx_1, idx_2, idx_3);
	      if (flag == false) {temp = set_stencil(idx_0, idx_1, idx_2, idx_3);}
	      else {
	        temp += set_stencil(idx_0, idx_1, idx_2, idx_3);
	      }
	      value.set_slc((unsigned int)lo, temp); 
	    }  
   }

  };

template <typename T, size_t EXTENT_0, size_t EXTENT_1 = 1, size_t EXTENT_2 = 1, size_t EXTENT_3 = 1>
  struct AxiPackedStencil {
    ac_int<8*sizeof(T)*EXTENT_3*EXTENT_2*EXTENT_1*EXTENT_0> value;
    ac_int<1, false> last;

    // convert to PackedStencil
    operator PackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3>() {
      PackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> res;
      res.value = value;
      return res;
    }

    // convert to Stencil
    operator Stencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3>() {
      PackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> res = *this;
      return (Stencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3>)res;
    }
  };

/** multi-dimension (up-to 4 dimensions) stencil struct
 */
template <typename T, size_t EXTENT_0, size_t EXTENT_1 = 1, size_t EXTENT_2 = 1, size_t EXTENT_3 = 1>
  struct Stencil {
  public:
    T value[EXTENT_3][EXTENT_2][EXTENT_1][EXTENT_0];

    /** writer
     */
    inline T& operator()(T set_val, size_t index_0, size_t index_1, size_t index_2, size_t index_3) {
        #ifndef __SYNTHESIS__
      assert(index_0 < EXTENT_0 && index_1 < EXTENT_1 && index_2 < EXTENT_2 && index_3 < EXTENT_3);
        #endif
      value[index_3][index_2][index_1][index_0] = set_val;
      return set_val;
    }

    /** reader
     */
    inline const T& operator()(size_t index_0, size_t index_1 = 0, size_t index_2 = 0, size_t index_3 = 0) const {
        #ifndef __SYNTHESIS__
      assert(index_0 < EXTENT_0 && index_1 < EXTENT_1 && index_2 < EXTENT_2 && index_3 < EXTENT_3);
        #endif
      return value[index_3][index_2][index_1][index_0];
    }

    // convert to PackedStencil
    operator PackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3>() {
      PackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> res;
      const size_t word_length = sizeof(T) * 8; // in bits
#pragma hls_unroll yes
      for(size_t idx_3 = 0; idx_3 < EXTENT_3; idx_3++)
#pragma hls_unroll yes      
        for(size_t idx_2 = 0; idx_2 < EXTENT_2; idx_2++)
#pragma hls_unroll yes        
	  for(size_t idx_1 = 0; idx_1 < EXTENT_1; idx_1++)
#pragma hls_unroll yes	  
	    for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
            const size_t lo = idx_0 * word_length +
                idx_1 * EXTENT_0 * word_length +
                idx_2 * EXTENT_0 * EXTENT_1 * word_length +
	      idx_3 * EXTENT_0 * EXTENT_1 * EXTENT_2 * word_length;
            ac_int<word_length, false> temp = bitcast_to_uint(value[idx_3][idx_2][idx_1][idx_0]);
            res.value.set_slc((unsigned int)lo, temp);
	    }
      return res;
    }

    // convert to AxiPackedStencil
    operator AxiPackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3>() {
      PackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> res = *this;
      return (AxiPackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3>)res;
    }

    //set 1st dim
    void set_dim(Stencil<T, EXTENT_0> set_val, size_t index_1, size_t index_2, size_t index_3) {
      const size_t word_length = sizeof(T) * 8; // in bits 
#pragma hls_unroll yes	      
      for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
          ac_int<word_length, false> temp = bitcast_to_uint(set_val[0][0][0][idx_0]);
          bitcast_to_type(temp, value[index_3][index_2][index_1][idx_0]);
      }
    }

    //set 1st and 2nd dim
    void set_dim(Stencil<T, EXTENT_0, EXTENT_1> set_val, size_t index_2, size_t index_3) {
      const size_t word_length = sizeof(T) * 8; // in bits 
#pragma hls_unroll yes	      
      for(size_t idx_1 = 0; idx_1 < EXTENT_0; idx_1++) 
#pragma hls_unroll yes	      
      for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
          ac_int<word_length, false> temp = bitcast_to_uint(set_val[0][0][idx_1][idx_0]);
          bitcast_to_type(temp, value[index_3][index_2][idx_1][idx_0]);
      }
    }


    //get 1st dim
    Stencil<T, EXTENT_0> get_dim(size_t index_1, size_t index_2, size_t index_3) {
      Stencil<T, EXTENT_0> res;
      const size_t word_length = sizeof(T) * 8; // in bits 
#pragma hls_unroll yes	      
      for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
          ac_int<word_length, false> temp = bitcast_to_uint(value[index_3][index_2][index_1][idx_0]);
          bitcast_to_type(temp, res.value[0][0][0][idx_0]);
      }
      return res;
    }

    //get 1st and 2nd dim
    Stencil<T, EXTENT_0, EXTENT_1> get_dim(size_t index_2, size_t index_3) {
      Stencil<T, EXTENT_0, EXTENT_1> res;
      const size_t word_length = sizeof(T) * 8; // in bits
#pragma hls_unroll yes	      
          for(size_t idx_1 = 0; idx_1 < EXTENT_1; idx_1++) 
#pragma hls_unroll yes	          
            for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
              ac_int<word_length, false> temp = bitcast_to_uint(value[index_3][index_2][idx_1][idx_0]);
              bitcast_to_type(temp, res.value[0][0][idx_1][idx_0]);
            }
      return res;
    }
  };

#ifndef HALIDE_ATTRIBUTE_ALIGN
  #ifdef _MSC_VER
#define HALIDE_ATTRIBUTE_ALIGN(x) __declspec(align(x))
  #else
#define HALIDE_ATTRIBUTE_ALIGN(x) __attribute__((aligned(x)))
  #endif
#endif
#ifndef BUFFER_T_DEFINED
#define BUFFER_T_DEFINED
#include <stdbool.h>
#include <stdint.h>
typedef struct buffer_t {
  uint64_t dev;
  uint8_t* host;
  int32_t extent[4];
  int32_t stride[4];
  int32_t min[4];
  int32_t elem_size;
  HALIDE_ATTRIBUTE_ALIGN(1) bool host_dirty;
  HALIDE_ATTRIBUTE_ALIGN(1) bool dev_dirty;
  HALIDE_ATTRIBUTE_ALIGN(1) uint8_t _padding[10 - sizeof(void *)];
} buffer_t;
#endif


template <typename T, size_t EXTENT_0, size_t EXTENT_1, size_t EXTENT_2, size_t EXTENT_3>
  void buffer_to_stencil(const buffer_t *buffer,
			 Stencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> &stencil) {
  assert(EXTENT_0 == buffer->extent[0]);
  assert((EXTENT_1 == buffer->extent[1]) || (EXTENT_1 == 1 && buffer->extent[1] == 0));
  assert((EXTENT_2 == buffer->extent[2]) || (EXTENT_2 == 1 && buffer->extent[2] == 0));
  assert((EXTENT_3 == buffer->extent[3]) || (EXTENT_3 == 1 && buffer->extent[3] == 0));
  assert(sizeof(T) == buffer->elem_size);

  for(size_t idx_3 = 0; idx_3 < EXTENT_3; idx_3++)
    for(size_t idx_2 = 0; idx_2 < EXTENT_2; idx_2++)
      for(size_t idx_1 = 0; idx_1 < EXTENT_1; idx_1++)
	for(size_t idx_0 = 0; idx_0 < EXTENT_0; idx_0++) {
	  const uint8_t *ptr = buffer->host;
        size_t offset = idx_0 * buffer->stride[0] +
            idx_1 * buffer->stride[1] +
            idx_2 * buffer->stride[2] +
	  idx_3 * buffer->stride[3];
        const T *address =  (T *)(ptr + offset * buffer->elem_size);
        stencil(*address, idx_0, idx_1, idx_2, idx_3);
	}
}

template <typename T, size_t EXTENT_0, size_t EXTENT_1, size_t EXTENT_2, size_t EXTENT_3>
  void subimage_to_stream(const struct buffer_t *buf_noop,
			  ac_channel<AxiPackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> > &stream,
			  void *subimage,
			  int stride_0, int subimage_extent_0,
			  int stride_1 = 1, int subimage_extent_1 = 1,
			  int stride_2 = 1, int subimage_extent_2 = 1,
			  int stride_3 = 1, int subimage_extent_3 = 1) {
  assert(subimage_extent_0 % EXTENT_0 == 0);
  assert(subimage_extent_1 % EXTENT_1 == 0);
  assert(subimage_extent_2 % EXTENT_2 == 0);
  assert(subimage_extent_3 % EXTENT_3 == 0);
  (void) buf_noop;  // avoid unused warnning
  for(size_t idx_3 = 0; idx_3 < subimage_extent_3; idx_3 += EXTENT_3)
    for(size_t idx_2 = 0; idx_2 < subimage_extent_2; idx_2 += EXTENT_2)
      for(size_t idx_1 = 0; idx_1 < subimage_extent_1; idx_1 += EXTENT_1)
	for(size_t idx_0 = 0; idx_0 < subimage_extent_0; idx_0 += EXTENT_0) {
	  Stencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> stencil;
	  for(size_t st_idx_3 = 0; st_idx_3 < EXTENT_3; st_idx_3++)
	    for(size_t st_idx_2 = 0; st_idx_2 < EXTENT_2; st_idx_2++)
	      for(size_t st_idx_1 = 0; st_idx_1 < EXTENT_1; st_idx_1++)
		for(size_t st_idx_0 = 0; st_idx_0 < EXTENT_0; st_idx_0++) {
		  int offset = (idx_0 + st_idx_0) * stride_0 +
		    (idx_1 + st_idx_1) * stride_1 +
		    (idx_2 + st_idx_2) * stride_2 +
		    (idx_3 + st_idx_3) * stride_3;
		  stencil(*((T *)subimage + offset), st_idx_0, st_idx_1, st_idx_2, st_idx_3);
		}
	  stream.write(stencil);
	}
}

template <typename T, size_t EXTENT_0, size_t EXTENT_1, size_t EXTENT_2, size_t EXTENT_3>
  void stream_to_subimage(const struct buffer_t *buf_noop,
			  ac_channel<AxiPackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> > &stream,
			  void *subimage,
			  int stride_0, int subimage_extent_0,
			  int stride_1 = 1, int subimage_extent_1 = 1,
			  int stride_2 = 1, int subimage_extent_2 = 1,
			  int stride_3 = 1, int subimage_extent_3 = 1) {
  assert(subimage_extent_0 % EXTENT_0 == 0);
  assert(subimage_extent_1 % EXTENT_1 == 0);
  assert(subimage_extent_2 % EXTENT_2 == 0);
  assert(subimage_extent_3 % EXTENT_3 == 0);
  (void) buf_noop;  // avoid unused warnning
  for(size_t idx_3 = 0; idx_3 < subimage_extent_3; idx_3 += EXTENT_3)
    for(size_t idx_2 = 0; idx_2 < subimage_extent_2; idx_2 += EXTENT_2)
      for(size_t idx_1 = 0; idx_1 < subimage_extent_1; idx_1 += EXTENT_1)
	for(size_t idx_0 = 0; idx_0 < subimage_extent_0; idx_0 += EXTENT_0) {
	  AxiPackedStencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> axi_stencil = stream.read();
	  Stencil<T, EXTENT_0, EXTENT_1, EXTENT_2, EXTENT_3> stencil = axi_stencil;
	  for(size_t st_idx_3 = 0; st_idx_3 < EXTENT_3; st_idx_3++)
	    for(size_t st_idx_2 = 0; st_idx_2 < EXTENT_2; st_idx_2++)
	      for(size_t st_idx_1 = 0; st_idx_1 < EXTENT_1; st_idx_1++)
		for(size_t st_idx_0 = 0; st_idx_0 < EXTENT_0; st_idx_0++) {
		  int offset = (idx_0 + st_idx_0) * stride_0 +
		    (idx_1 + st_idx_1) * stride_1 +
		    (idx_2 + st_idx_2) * stride_2 +
		    (idx_3 + st_idx_3) * stride_3;
		  *((T *)subimage + offset) = stencil(st_idx_0, st_idx_1, st_idx_2, st_idx_3);
		}
	  // check TLAST
	  if (idx_3 == subimage_extent_3 - EXTENT_3 &&
            idx_2 == subimage_extent_2 - EXTENT_2 &&
            idx_1 == subimage_extent_1 - EXTENT_1 &&
	      idx_0 == subimage_extent_0 - EXTENT_0) {
            if(axi_stencil.last != 1) {
	      printf("TLAS check failed.\n");
            }
	  } else {
            if(axi_stencil.last != 0) {
	      printf("TLAS check failed.\n");
            }


	  }
	}
}


#endif