[RUNTIME] Implement TVMDSOOp(TensorFlow custom op) for TVM runtime (#4459)
authortobe <tobeg3oogle@gmail.com>
Tue, 7 Apr 2020 23:59:32 +0000 (07:59 +0800)
committerGitHub <noreply@github.com>
Tue, 7 Apr 2020 23:59:32 +0000 (16:59 -0700)
commit53a4ad354d435756927c2926d90b20fd30b431e6
tree13c449e36cf0add0a68439be690afc195ae9e5ab
parent4e00763242bd7242c37f64f62cf750e918790bf5
[RUNTIME] Implement TVMDSOOp(TensorFlow custom op) for TVM runtime (#4459)

* Add implementation of TVMDSOOp

* feat: Update cmake script to work with c++11 and in-repo build

* feat: Use libtvm as oplib dependency

* fix: Add missing link dependency to libtvm

* feat: Update tf tvmdso op by review comments

* fix: Update with pr comments

* fix: Fix lint

* feat: Add test script and fix gpu shape

* feat: Add test script and fix gpu shape

* fix: Conditional build tftvm op for gpu

* fix: Conditional build tftvm op for gpu

* fix: Fix pylint of tf_op module.py

* fix: Fix pylint of tf_op module.py

* feat: Conditional enable gpu test for tftvm op

* feat: Conditional enable gpu test for tftvm op

* feat: Add tf_tvmdsoop test script as an app test

* fix: Fix gpu/cpu enabled check on tvm in test script

* fix: Make tf tvmdso op test script runnable with pytest

* remove unused test script test_tfop_module.py

* fix: Remove pushd & popd in tfdsoop test script

* fix: Upgrade tftvmop use python3 to find TensorFlow

* fix: Upgrade tftvmop use python3 to find TensorFlow

* fix: Change target_link_options to target_link_libraries

* fix: Add tftvmop build script's c++ option

* fix: Add tvm library path to tf op test library path

* fix: Debug ci build for tftvm dso op

* fix: Fix cmake error and skip tfop test

* fix: Fix typo and indentation issues

* feat: Use TF list input op def

* fix: Fix style and unexpected changes

Co-authored-by: baoxinqi <baoxinqi@4paradigm.com>
Co-authored-by: Chen Dihao <chendihao@4paradigm.com>
Co-authored-by: wrongtest <wrongtest@4paradigm.com>
CMakeLists.txt
apps/tf_tvmdsoop/CMakeLists.txt [new file with mode: 0644]
apps/tf_tvmdsoop/prepare_and_test_tfop_module.sh [new file with mode: 0644]
apps/tf_tvmdsoop/tests/test_tfop_module.py [new file with mode: 0644]
cmake/config.cmake
cmake/modules/contrib/TF_TVMDSOOP.cmake [new file with mode: 0644]
python/tvm/contrib/tf_op/__init__.py [new file with mode: 0644]
python/tvm/contrib/tf_op/module.py [new file with mode: 0644]
src/contrib/tf_op/tvm_dso_op_kernels.cc [new file with mode: 0644]
src/contrib/tf_op/tvm_dso_ops.cc [new file with mode: 0644]
tests/scripts/task_python_integration.sh