[Relay][TOPI] Gluncv SSD support on the GPU (#2784)
authorLeyuan Wang <laurawly@gmail.com>
Mon, 29 Apr 2019 03:47:21 +0000 (20:47 -0700)
committerHaichen Shen <shenhaichen@gmail.com>
Mon, 29 Apr 2019 03:47:21 +0000 (20:47 -0700)
commita706ad16f83d810f4a6268cf604cece79c2c4791
treee6ffe68cca9c1afe54cd07877c189358b89f78d0
parent9d002e8eb2b1578020510aa6b770788bef95a67c
[Relay][TOPI] Gluncv SSD support on the GPU (#2784)

* ssd gluoncv gpu op updated

* ssd gluoncv gpu op updated

* tutorials and testes modified

* tutorials and testes modified

* fix lint

* fix lint

* address comment

* multibox bug fixed

* space line added

* use less threads per block

* use less threads per block

* less threads per block for get valid count

* less threads per block for get valid count

* merge with master

* Revert "less threads per block for get valid count"

This reverts commit 08896cfccc34b0b2a1646d01d01ea4cad73941c4.

* Revert "less threads per block for get valid count"

This reverts commit 08896cfccc34b0b2a1646d01d01ea4cad73941c4.

* typo fixed

* elem length made to a variable

* fix lint error

* fix lint error

* lint fixed

* bug fixed

* bug fixed

* lint fixed

* error fixed

* error fixed

* test ci

* test ci

* seperate argsort to be an independent op

* seperate argsort to be an independent op

* fix lint

* fix lint

* remove unsupported models

* typo fixed

* argsort added to realy

* solve conflicts with master

* fix lint

* fix lint

* test push

* Revert "test push"

This reverts commit 6db00883fab6cc06bddf564c926bb27c874397d8.

* fix lint error

* fix more lint

* cpu test_sort udpated

* debug ci

* nms fixed

* expose argsort to relay frontend

* test ci

* fix lint

* sort register error fixed

* fix nnvm

* nms type fixed

* adaptive pooling added to relay

* Revert "adaptive pooling added to relay"

This reverts commit 1119f1f2c055753e0cc5611627597749134c5c8c.

* fix lint

* expose argsort op

* fix lint

* fix lint

* fix lint

* sort test updated

* sort bug fixed

* nnvm error fixed

* fix argsort default data type returned to be float insteaf of int

* fix lint

* fix lint

* test fixed

* fix valid count

* fix titanx bug

* tutorial add both targets

* titanx error fixed

* try to fix CI old gpu error

* try to solve CI GPU error

* get_valid_count added

* reverse get_valid_count

* get valid count optimized

* address comments

* fix ci error

* remove unessesary block sync

* add back one sync

* address comments

* address more comments

* more comments

* move sort to be indepent algorithm

* typo fixed

* more typos

* comments addressed

* doc updated

* fix pylint

* address final comments

* apache license added
34 files changed:
docs/langref/relay_op.rst
include/tvm/relay/attrs/algorithm.h [new file with mode: 0644]
include/tvm/relay/attrs/vision.h
nnvm/include/nnvm/top/nn.h
nnvm/python/nnvm/top/vision.py
nnvm/tests/python/compiler/test_top_level4.py
python/tvm/relay/__init__.py
python/tvm/relay/frontend/mxnet.py
python/tvm/relay/op/__init__.py
python/tvm/relay/op/_algorithm.py [new file with mode: 0644]
python/tvm/relay/op/algorithm.py [new file with mode: 0644]
python/tvm/relay/op/tensor.py
python/tvm/relay/op/transform.py
python/tvm/relay/op/vision/_vision.py
python/tvm/relay/op/vision/nms.py
src/contrib/sort/sort.cc
src/relay/op/algorithm/sort.cc [new file with mode: 0644]
src/relay/op/vision/nms.cc
tests/python/contrib/test_sort.py
tests/python/relay/test_op_level5.py
tests/python/relay/test_op_level6.py [new file with mode: 0644]
topi/python/topi/__init__.py
topi/python/topi/cuda/nms.py
topi/python/topi/cuda/sort.py [new file with mode: 0644]
topi/python/topi/cuda/ssd/multibox.py
topi/python/topi/cuda/vision.py
topi/python/topi/generic/__init__.py
topi/python/topi/generic/sort.py [new file with mode: 0644]
topi/python/topi/sort.py [new file with mode: 0644]
topi/python/topi/vision/nms.py
topi/python/topi/vision/ssd/multibox.py
topi/tests/python/test_topi_sort.py [new file with mode: 0644]
topi/tests/python/test_topi_vision.py
tutorials/frontend/deploy_ssd_gluoncv.py