Skip to content

T
tic
Commit 325c8274 by Tianqi Chen Committed by GitHub
Options

[NNVM] Fix check in layout parsing (#1502) 

* [NNVM] Fix check in layout parsing

* add one workload
parent 499ee952
Showing with 3 additions and 1 deletions
nnvm/include/nnvm/layout.h
...@@ -298,7 +298,7 @@ class Layout { ...@@ -298,7 +298,7 @@ class Layout {
std::ostringstream repr; std::ostringstream repr;
if (is_subdim(layout_simplified_[i])) { if (is_subdim(layout_simplified_[i])) {
auto factor = subsizeof(layout_simplified_[i]); auto factor = subsizeof(layout_simplified_[i]);
CHECK_LT(factor, 0); CHECK_GT(factor, 0);
repr << factor; repr << factor;
} }
repr << layout_simplified_[i]; repr << layout_simplified_[i];
......
vta/python/vta/top/arm_conv2d.py
...@@ -11,6 +11,7 @@ from topi.rasp import conv2d as _rasp_conv2d ...@@ -11,6 +11,7 @@ from topi.rasp import conv2d as _rasp_conv2d
from topi import generic from topi import generic
_WORKLOADS = [ _WORKLOADS = [
Workload('float32', 'float32', 224, 224, 3, 64, 7, 7, 3, 3, 2, 2),
Workload('int8', 'int32', 224, 224, 3, 64, 7, 7, 3, 3, 2, 2), Workload('int8', 'int32', 224, 224, 3, 64, 7, 7, 3, 3, 2, 2),
Workload('int8', 'int32', 56, 56, 64, 64, 3, 3, 1, 1, 1, 1), Workload('int8', 'int32', 56, 56, 64, 64, 3, 3, 1, 1, 1, 1),
Workload('int8', 'int32', 56, 56, 64, 64, 1, 1, 0, 0, 1, 1), Workload('int8', 'int32', 56, 56, 64, 64, 1, 1, 0, 0, 1, 1),
...@@ -27,6 +28,7 @@ _WORKLOADS = [ ...@@ -27,6 +28,7 @@ _WORKLOADS = [
_SCHEDULES = [ _SCHEDULES = [
# float32 imagenet # float32 imagenet
SpatialPack(1, 8, 4, 1, 4, True), SpatialPack(1, 8, 4, 1, 4, True),
SpatialPack(1, 8, 4, 1, 4, True),
SpatialPack(1, 7, 4, 2, 4, True), SpatialPack(1, 7, 4, 2, 4, True),
SpatialPack(1, 4, 8, 4, 1, True), SpatialPack(1, 4, 8, 4, 1, True),
SpatialPack(1, 4, 4, 1, 16, False), SpatialPack(1, 4, 4, 1, 16, False),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment