Skip to content

Commit c10ad96

Browse files
huajsjcomaniac
authored andcommitted
[Runtime] Pipeline Executor Initial patch. (apache#8702)
* [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>
1 parent e559065 commit c10ad96

File tree

6 files changed

+903
-0
lines changed

6 files changed

+903
-0
lines changed

CMakeLists.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -406,6 +406,12 @@ if(GTEST_INCLUDE_DIR AND GTEST_LIB)
406406
include(GoogleTest)
407407
endif()
408408

409+
if(USE_PIPELINE_EXECUTOR)
410+
message(STATUS "Build with Pipeline Executor support...")
411+
file(GLOB RUNTIME_PIPELINE_SRCS src/runtime/pipeline/*.cc)
412+
list(APPEND RUNTIME_SRCS ${RUNTIME_PIPELINE_SRCS})
413+
endif(USE_PIPELINE_EXECUTOR)
414+
409415
# Module rules
410416
include(cmake/modules/VTA.cmake)
411417
include(cmake/modules/StandaloneCrt.cmake)

cmake/config.cmake

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,9 @@ set(USE_GRAPH_EXECUTOR ON)
105105
# Whether enable tiny graph executor with CUDA Graph
106106
set(USE_GRAPH_EXECUTOR_CUDA_GRAPH OFF)
107107

108+
# Whether enable pipeline executor.
109+
set(USE_PIPELINE_EXECUTOR OFF)
110+
108111
# Whether to enable the profiler for the graph executor and vm
109112
set(USE_PROFILER ON)
110113

0 commit comments

Comments
 (0)