Skip to content

Commit

Permalink
[Runtime] Pipeline Executor Initial patch. (apache#8702)
Browse files Browse the repository at this point in the history
* [Runtime] Pipeline Executor Initial patch.

This patch is one of serial patch for PR 7892 splitting.this is the initial part
of the pipeline executor, this patch include the cmake change and python and C++
interface for pipeline executor.

* add pipeline config

* add config connect logic.

* fix build issue.

* set output index start from 0

* address review comments

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

* address review comments.

* address review comments.

* Update python/tvm/contrib/pipeline_executor.py

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

Update python/tvm/contrib/pipeline_executor.py

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

* address review comments.

* address review comments

* add topology sort

* add binding check logic.

* fix plint error.

* Update python/tvm/contrib/pipeline_executor.py

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

* address review comments.

* fix plint issue.

* address review comments.

* trigger build.

* Update python/tvm/contrib/pipeline_executor.py

Co-authored-by: Cody Yu <comaniac0422@gmail.com>

address review comments

* address review comments.

* polish doc and comments.

* polish doc and address review comments.

* address review comments.

* doc change.

* doc change.

* Trigger build.

* trigge build.

* address review comments.

* address review comments.

* address review comments.

* polish documents.

* Polish the document.

* address review comments.

Co-authored-by: Cody Yu <comaniac0422@gmail.com>
  • Loading branch information
huajsj and comaniac authored Sep 15, 2021
1 parent dd7d830 commit 9bc4dc0
Show file tree
Hide file tree
Showing 6 changed files with 903 additions and 0 deletions.
6 changes: 6 additions & 0 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -389,6 +389,12 @@ if(GTEST_INCLUDE_DIR AND GTEST_LIB)
include(GoogleTest)
endif()

if(USE_PIPELINE_EXECUTOR)
message(STATUS "Build with Pipeline Executor support...")
file(GLOB RUNTIME_PIPELINE_SRCS src/runtime/pipeline/*.cc)
list(APPEND RUNTIME_SRCS ${RUNTIME_PIPELINE_SRCS})
endif(USE_PIPELINE_EXECUTOR)

# Module rules
include(cmake/modules/VTA.cmake)
include(cmake/modules/StandaloneCrt.cmake)
Expand Down
3 changes: 3 additions & 0 deletions cmake/config.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -105,6 +105,9 @@ set(USE_GRAPH_EXECUTOR ON)
# Whether enable tiny graph executor with CUDA Graph
set(USE_GRAPH_EXECUTOR_CUDA_GRAPH OFF)

# Whether enable pipeline executor.
set(USE_PIPELINE_EXECUTOR OFF)

# Whether to enable the profiler for the graph executor and vm
set(USE_PROFILER ON)

Expand Down
Loading

0 comments on commit 9bc4dc0

Please sign in to comment.