vta_test.cc 2.33 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
/*!
 *  Copyright (c) 2018 by Contributors
 * \file vta_test.cpp
 * \brief Simulation tests for the VTA design.
 */

#include <stdio.h>
#include <stdlib.h>
#include <iostream>

11 12
#include "../src/vta.h"
#include "../../../tests/hardware/common/test_lib.h"
13

14 15
int main(void) {
#if DEBUG == 1
16 17 18 19
    printParameters();
#endif

    // Micro op bound
20 21 22 23
    assert(VTA_UOP_GEM_2_1 < VTA_UOP_WIDTH);
    assert(VTA_UOP_ALU_1_1 < VTA_UOP_WIDTH);
    // Make sure there is no misaligment
    assert(VTA_INSN_GEM_9_1 < VTA_INSN_GEM_A_0);
24
    assert(VTA_INSN_MEM_7_1 < VTA_INSN_MEM_8_0);
25
    // Instruction bounds
26
    assert(VTA_INSN_MEM_E_1 < VTA_INS_WIDTH);
27 28
    assert(VTA_INSN_GEM_F_1 < VTA_INS_WIDTH);
    assert(VTA_INSN_ALU_G_1 < VTA_INS_WIDTH);
29 30 31 32

    int status = 0;

    // Run ALU test (vector-scalar operators)
33 34 35 36 37 38 39 40
    status |= alu_test(VTA_ALU_OPCODE_MIN, true, VTA_BLOCK_OUT, 128, true);
    status |= alu_test(VTA_ALU_OPCODE_MIN, true, VTA_BLOCK_OUT, 128, false);
    status |= alu_test(VTA_ALU_OPCODE_MAX, true, VTA_BLOCK_OUT, 128, true);
    status |= alu_test(VTA_ALU_OPCODE_MAX, true, VTA_BLOCK_OUT, 128, false);
    status |= alu_test(VTA_ALU_OPCODE_ADD, true, VTA_BLOCK_OUT, 128, true);
    status |= alu_test(VTA_ALU_OPCODE_ADD, true, VTA_BLOCK_OUT, 128, false);
    status |= alu_test(VTA_ALU_OPCODE_SHR, true, VTA_BLOCK_OUT, 128, true);
    status |= alu_test(VTA_ALU_OPCODE_SHR, true, VTA_BLOCK_OUT, 128, false);
41 42

    // Run ALU test (vector-vector operators)
43 44 45 46 47 48
    status |= alu_test(VTA_ALU_OPCODE_MIN, false, VTA_BLOCK_OUT, 128, true);
    status |= alu_test(VTA_ALU_OPCODE_MIN, false, VTA_BLOCK_OUT, 128, false);
    status |= alu_test(VTA_ALU_OPCODE_MAX, false, VTA_BLOCK_OUT, 128, true);
    status |= alu_test(VTA_ALU_OPCODE_MAX, false, VTA_BLOCK_OUT, 128, false);
    status |= alu_test(VTA_ALU_OPCODE_ADD, false, VTA_BLOCK_OUT, 128, true);
    status |= alu_test(VTA_ALU_OPCODE_ADD, false, VTA_BLOCK_OUT, 128, false);
49 50

    // Run blocked GEMM test
51 52 53 54
    status |= blocked_gemm_test(256, 256, VTA_BLOCK_OUT*4, true, 2);
    status |= blocked_gemm_test(256, 256, VTA_BLOCK_OUT*4, false, 2);
    status |= blocked_gemm_test(256, 256, VTA_BLOCK_OUT*4, true, 1);
    status |= blocked_gemm_test(256, 256, VTA_BLOCK_OUT*4, false, 1);
55

56 57 58
    // Simple GEMM unit test
    status |= gemm_test(64, 64, 64, true);

59
    return status;
60
}