Commit 325c8274 by Tianqi Chen Committed by GitHub

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

* [NNVM] Fix check in layout parsing

* add one workload
parent 499ee952
......@@ -298,7 +298,7 @@ class Layout {
std::ostringstream repr;
if (is_subdim(layout_simplified_[i])) {
auto factor = subsizeof(layout_simplified_[i]);
CHECK_LT(factor, 0);
CHECK_GT(factor, 0);
repr << factor;
}
repr << layout_simplified_[i];
......
......@@ -11,6 +11,7 @@ from topi.rasp import conv2d as _rasp_conv2d
from topi import generic
_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', 56, 56, 64, 64, 3, 3, 1, 1, 1, 1),
Workload('int8', 'int32', 56, 56, 64, 64, 1, 1, 0, 0, 1, 1),
......@@ -27,6 +28,7 @@ _WORKLOADS = [
_SCHEDULES = [
# float32 imagenet
SpatialPack(1, 8, 4, 1, 4, True),
SpatialPack(1, 8, 4, 1, 4, True),
SpatialPack(1, 7, 4, 2, 4, True),
SpatialPack(1, 4, 8, 4, 1, True),
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