Skip to content

Commit

Permalink
Add sram wmem tests
Browse files Browse the repository at this point in the history
  • Loading branch information
lukamac committed Feb 28, 2024
1 parent 9ed5075 commit c283557
Show file tree
Hide file tree
Showing 4 changed files with 69 additions and 6 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
- support 32bit scale
- cmake support
- const qualifier to `<acc>_dev_t` function arguments
- support for N-EUREKA's dedicated weight memory

### Changed

Expand Down
14 changes: 8 additions & 6 deletions test/conf.toml
Original file line number Diff line number Diff line change
Expand Up @@ -22,20 +22,20 @@
# Ne16TestClasses.py:Ne16TestConf().check_valid()

# Input dimensions
in_height = 4
in_height = 19
in_width = 3
in_channel = 8
in_channel = 43

# Output dimensions
# - height and width calculated from input dimensions
out_channel = 8
out_channel = 39

# If depthwise, group == in_channel
depthwise = false

# Kernel shape
kernel_shape.height = 3
kernel_shape.width = 3
kernel_shape.height = 1
kernel_shape.width = 1

# Stride
stride.height = 1
Expand All @@ -58,4 +58,6 @@ bias_type = "int32"
has_norm_quant = true
has_bias = true
has_relu = true
wmem = "sram"

# Weight memory source
wmem = "tcdm"
30 changes: 30 additions & 0 deletions test/tests/test_119/conf.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
{
"in_height": 4,
"in_width": 20,
"in_channel": 80,
"in_type": "uint8",
"out_channel": 25,
"out_type": "int8",
"depthwise": false,
"kernel_shape": {
"height": 3,
"width": 3
},
"weight_type": "int8",
"scale_type": "uint8",
"bias_type": "int32",
"stride": {
"height": 1,
"width": 1
},
"padding": {
"top": 0,
"bottom": 1,
"left": 1,
"right": 1
},
"has_norm_quant": true,
"has_bias": true,
"has_relu": false,
"wmem": "sram"
}
30 changes: 30 additions & 0 deletions test/tests/test_120/conf.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
{
"in_height": 19,
"in_width": 3,
"in_channel": 43,
"out_channel": 39,
"padding": {
"top": 0,
"bottom": 0,
"left": 0,
"right": 0
},
"kernel_shape": {
"height": 1,
"width": 1
},
"depthwise": false,
"stride": {
"height": 1,
"width": 1
},
"in_type": "uint8",
"out_type": "uint8",
"weight_type": "int8",
"scale_type": "uint8",
"bias_type": "int32",
"has_norm_quant": true,
"has_bias": true,
"has_relu": true,
"wmem": "sram"
}

0 comments on commit c283557

Please sign in to comment.