source: trunk/lib_ir/gen/test_llc.pytemplate.cpp @ 4143

/******************
 * Python template to auto gen llc test file.
 * sudo easy_install Jinja2 to install the Jinja2 template system
 * {{HeaderComment}}
 */

#include "utility.h"
#include <ctime>
#include <iostream>
#include <cstdlib>
using namespace std;
{% set Names = FunctionNamesI1 + FunctionNamesI2 + FunctionNamesI4 + FunctionNamesI8 %}
{% set INS_EXT_VEC_ELT_FWs = [2, 4] %}

extern "C" {
{% for name in Names %}
  SIMD_type {{ name.c }}(SIMD_type a, SIMD_type b);
{% endfor %}

{% for fw in INS_EXT_VEC_ELT_FWs %}
  SIMD_type insertelement_{{ fw }}(SIMD_type a, short elt, int idx);
  short extractelement_{{ fw }}(SIMD_type a, int idx);
{% endfor %}

//long add/shift
SIMD_type add_128(SIMD_type a, SIMD_type b);
SIMD_type sll_128(SIMD_type a, SIMD_type b);

SIMD_type add_with_carry_ir(SIMD_type a, SIMD_type b, SIMD_type carry_in, SIMD_type *carry_out, SIMD_type *sum);

{% for fw in [128]%}
void uadd_with_overflow_i{{ fw }}(bitblock{{fw}}_t a, bitblock{{fw}}_t b, bitblock{{fw}}_t *sum, short *obit);
{% endfor %}
}

void fill_random(SIMD_type &a, SIMD_type &b)
{
  a = mvmd<8>::fill((int)rand() % (1 << 8));
  b = mvmd<8>::fill((int)rand() % (1 << 8));
}

int test_long_stream_addition(const SIMD_type &a, const SIMD_type &b)
{
    SIMD_type c, d, e, f;
    short x, y;

    //test uadd.with.overflow.i128
    c = mvmd<32>::fill(0);
    SIMD_type sum0, obit0, sum1, sum2, obit2;
    short obit1;
    adc(a, b, c, obit0, sum0);
    uadd_with_overflow_i128(a, b, &sum1, &obit1);
    x = 0;
    if (Store2String(sum0,1) != Store2String(sum1,1)) {
      cout << "uadd_with_overflow_i128, sum error" << endl;
      x = 1;
    }
    if (Store2String(mvmd<128>::fill(obit1),1) != Store2String(obit0,1)) {
      cout << "uadd_with_overflow_i128 failed, obit error" << endl;
      x = 1;
    }
    if (x) {
      cout << "a =      " << Store2String(a, 1) << endl;
      cout << "b =      " << Store2String(b, 1) << endl;
      cout << "sum0 =   " << Store2String(sum0, 1) << endl;
      cout << "sum1 =   " << Store2String(sum1, 1) << endl;
      cout << "obit0 =  " << Store2String(obit0, 1) << endl;
      cout << "obit1 =  " << obit1 << endl;
      return 1;
    }

    //test add_with_carry in pure IR
    y = rand() % 2;
    c = mvmd<128>::fill(y);
    adc(a, b, c, obit0, sum0);
    add_with_carry_ir(a, b, c, &obit2, &sum2);
    if (Store2String(sum0,1) != Store2String(sum2,1) ||
        Store2String(obit0,1) != Store2String(obit2,1)) {
      cout << "add_with_carry_ir failed." << endl;
      cout << "a =      " << Store2String(a, 1) << endl;
      cout << "b =      " << Store2String(b, 1) << endl;
      cout << "sum0 =   " << Store2String(sum0, 1) << endl;
      cout << "sum2 =   " << Store2String(sum2, 1) << endl;
      cout << "obit0 =  " << Store2String(obit0, 1) << endl;
      cout << "obit2 =  " << Store2String(obit2, 1) << endl;
      return 1;
    }

    return 0;
}

int main() {
  SIMD_type a, b, c, d, e, f;
  short x, y;
  srand(time(0));

  for (unsigned i = 0; i < 500; ++i) {
    fill_random(a, b);

    //All the binary ops
    {% for name in Names %}
    {% if name.flag == "" %}
      c = {{ name.c }}(a, b);
      d = {{ name.cxx }}(a, b);
      if (Store2String(c,1) != Store2String(d,1)) {
        cout << "{{ name.c }} faild." << endl;
        cout << "A = " << Store2String(a, 1) << endl;
        cout << "B = " << Store2String(b, 1) << endl;
        cout << "c = " << Store2String(c, 1) << endl;
        cout << "cxx=" << Store2String(d, 1) << endl;

        return 1;
      }

    {% endif %}
    {% endfor %}

    //Shifting tests need special treatment
    {% for name in Names %}
    {% if name.flag == "shifting" %}
    {% for x in range(0, name.fw) %}
      c = {{ name.cxx }}<{{ x }}>(a);
      f = mvmd<{{ name.fw }}>::fill({{ x }});
      d = {{ name.c }}(a, f);
      if (Store2String(c,1) != Store2String(d,1)) {
        cout << "{{ name.c }} faild." << endl;
        return 1;
      }

    {% endfor %}
    {% endif %}
    {% endfor %}

    //insert_vector_elt and extract_
    {% for fw in INS_EXT_VEC_ELT_FWs %}
    {% for idx in range(0, 128 // fw) %}
      x = extractelement_{{ fw }}(a, {{ idx }});
      y = mvmd<{{ fw }}>::extract<{{ idx }}>(a);
      if (x != y) {
        cout << "extractelement_{{fw}} failed." << endl;
        return 1;
      }

      x = rand() % 100;
      c = insertelement_{{ fw }}(a, x, {{ idx }});
      d = mvmd<{{ fw }}>::insert<{{ idx }}>(a, x);
      if (Store2String(c,1) != Store2String(d,1)) {
        cout << "insertelement_{{fw}} failed." << endl;
        return 1;
      }
    {% endfor %}
    {% endfor %}

    //try i128 add and shift
    c = simd<128>::add(a, b);
    d = add_128(a, b);
    if (Store2String(c,1) != Store2String(d,1)) {
      cout << "add_128 failed." << endl;
      return 1;
    }

    {% set imm = range(0,128) | random %}
    c = simd<128>::slli<{{ imm }}>(a);
    f = mvmd<128>::fill({{ imm }});
    d = sll_128(a, f);
    if (Store2String(c,1) != Store2String(d,1)) {
      cout << "sll_128 failed." << endl;
      return 1;
    }

    if (test_long_stream_addition(a, b))
      return 1;
  }

  return 0;
}