Skip to content

Commit 8c19f62

Browse files
committed
fixed FEDP emulation MACROs
1 parent ebb12a6 commit 8c19f62

File tree

4 files changed

+7
-7
lines changed

4 files changed

+7
-7
lines changed

ci/regression.sh.in

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -494,7 +494,7 @@ tensor()
494494
make -C hw/unittest/tcu_fedp clean && CONFIGS="-DTCU_TYPE_DPI" NUM_REGS=1 LATENCY=4 make -C hw/unittest/tcu_fedp && hw/unittest/tcu_fedp/tcu_fedp --fmt=1 --no-fused
495495
make -C hw/unittest/tcu_fedp clean && CONFIGS="-DTCU_TYPE_BHF" NUM_REGS=1 LATENCY=10 make -C hw/unittest/tcu_fedp && hw/unittest/tcu_fedp/tcu_fedp --fmt=1 --no-fused
496496
make -C hw/unittest/tcu_fedp clean && CONFIGS="-DTCU_TYPE_DSP" NUM_REGS=1 LATENCY=31 make -C hw/unittest/tcu_fedp && hw/unittest/tcu_fedp/tcu_fedp --fmt=1 --no-fused --ulp=3
497-
make -C hw/unittest/tcu_fedp clean && CONFIGS="-DTCU_TYPE_DRL -DUSE_FEDP" NUM_REGS=2 LATENCY=4 make -C hw/unittest/tcu_fedp && hw/unittest/tcu_fedp/tcu_fedp --fmt=1 --ulp=2
497+
make -C hw/unittest/tcu_fedp clean && CONFIGS="-DTCU_TYPE_DRL -DUSE_FEDP" NUM_REGS=2 LATENCY=4 make -C hw/unittest/tcu_fedp && hw/unittest/tcu_fedp/tcu_fedp --fmt=1 --no-zeros --no-subnormals --no-infinities --no-nans
498498

499499
# test bf16
500500
make -C hw/unittest/tcu_fedp clean && CONFIGS="-DTCU_TYPE_DPI" NUM_REGS=1 LATENCY=4 make -C hw/unittest/tcu_fedp && hw/unittest/tcu_fedp/tcu_fedp --fmt=2 --no-fused

hw/rtl/tcu/drl/VX_tcu_fedp_drl.sv

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
module VX_tcu_fedp_drl #(
1717
parameter LATENCY = 0,
1818
parameter N = 2,
19-
parameter W = 53
19+
parameter W = 25
2020
) (
2121
input wire clk,
2222
input wire reset,

hw/unittest/tcu_fedp/fedp.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
#include <unordered_map>
2323
#include <vector>
2424

25-
#if FEDP_TRACE
25+
#ifdef FEDP_TRACE
2626
#include <cstdio>
2727
#define LOG(...) std::fprintf(stderr, __VA_ARGS__);
2828
#else

hw/unittest/tcu_fedp/main.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
#include <bitmanip.h>
4444
#include "softfloat_ext.h"
4545

46-
#ifdef FEDP_EMUL
46+
#ifdef USE_FEDP
4747
#include "fedp.h"
4848
#endif
4949

@@ -280,7 +280,7 @@ static void pack_elements(const std::vector<uint32_t> &elements, int element_bit
280280
}
281281
}
282282

283-
#ifndef FEDP_EMUL
283+
#ifndef USE_FEDP
284284
// Calculate expected fp dot product
285285
static float dot_product(const uint32_t* A, const uint32_t* B, uint32_t C, int n, int eb, int sb, bool fused) {
286286
auto to_float = [&](uint32_t x, int ebits, int sbits) -> long double {
@@ -679,7 +679,7 @@ class Testbench {
679679
const uint32_t NF = features_to_test.size();
680680
const uint32_t tests_per_feature = (NT + NF - 1) / NF;
681681

682-
#ifdef FEDP_EMUL
682+
#ifdef USE_FEDP
683683
FEDP fedp(config_.exp_bits, config_.sig_bits, NUM_REGS * 2, (int)config_.frm, config_.W, config_.renorm);
684684
#endif
685685

@@ -746,7 +746,7 @@ class Testbench {
746746
std::memcpy(&dut_result, &dut_result_bits, sizeof(float));
747747

748748
// Calculate expected result
749-
#ifdef FEDP_EMUL
749+
#ifdef USE_FEDP
750750
float expected = fedp(a_packed.data(), b_packed.data(), c_value_float, NUM_REGS);
751751
#else
752752
float expected = dot_product(a_value_hex.data(), b_value_hex.data(),

0 commit comments

Comments
 (0)