Skip to content

Commit dcc6af5

Browse files
authored
[RUNTIME] Separate runtime related contrib into runtime/contrib (#4207)
1 parent 13b2856 commit dcc6af5

31 files changed

Lines changed: 51 additions & 62 deletions

cmake/modules/CUDA.cmake

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,14 +40,14 @@ if(USE_CUDA)
4040

4141
if(USE_CUDNN)
4242
message(STATUS "Build with cuDNN support")
43-
file(GLOB CONTRIB_CUDNN_SRCS src/contrib/cudnn/*.cc)
43+
file(GLOB CONTRIB_CUDNN_SRCS src/runtime/contrib/cudnn/*.cc)
4444
list(APPEND RUNTIME_SRCS ${CONTRIB_CUDNN_SRCS})
4545
list(APPEND TVM_RUNTIME_LINKER_LIBS ${CUDA_CUDNN_LIBRARY})
4646
endif(USE_CUDNN)
4747

4848
if(USE_CUBLAS)
4949
message(STATUS "Build with cuBLAS support")
50-
file(GLOB CONTRIB_CUBLAS_SRCS src/contrib/cublas/*.cc)
50+
file(GLOB CONTRIB_CUBLAS_SRCS src/runtime/contrib/cublas/*.cc)
5151
list(APPEND RUNTIME_SRCS ${CONTRIB_CUBLAS_SRCS})
5252
list(APPEND TVM_RUNTIME_LINKER_LIBS ${CUDA_CUBLAS_LIBRARY})
5353
endif(USE_CUBLAS)

cmake/modules/Metal.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ if(USE_METAL)
2424
list(APPEND RUNTIME_SRCS ${RUNTIME_METAL_SRCS})
2525

2626
if(USE_MPS)
27-
file(GLOB MPS_CONTRIB_SRC src/contrib/mps/*.mm)
27+
file(GLOB MPS_CONTRIB_SRC src/runtime/contrib/mps/*.mm)
2828
list(APPEND RUNTIME_SRCS ${MPS_CONTRIB_SRC})
2929
find_library(MPS_CONTRIB_LIB MetalPerformanceShaders)
3030
list(APPEND TVM_RUNTIME_LINKER_LIBS ${MPS_CONTRIB_LIB})

cmake/modules/ROCM.cmake

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,14 +37,14 @@ if(USE_ROCM)
3737

3838
if(USE_MIOPEN)
3939
message(STATUS "Build with MIOpen support")
40-
file(GLOB MIOPEN_CONTRIB_SRCS src/contrib/miopen/*.cc)
40+
file(GLOB MIOPEN_CONTRIB_SRCS src/runtime/contrib/miopen/*.cc)
4141
list(APPEND RUNTIME_SRCS ${MIOPEN_CONTRIB_SRCS})
4242
list(APPEND TVM_RUNTIME_LINKER_LIBS ${ROCM_MIOPEN_LIBRARY})
4343
endif(USE_MIOPEN)
4444

4545
if(USE_ROCBLAS)
4646
message(STATUS "Build with RocBLAS support")
47-
file(GLOB ROCBLAS_CONTRIB_SRCS src/contrib/rocblas/*.cc)
47+
file(GLOB ROCBLAS_CONTRIB_SRCS src/runtime/contrib/rocblas/*.cc)
4848
list(APPEND RUNTIME_SRCS ${ROCBLAS_CONTRIB_SRCS})
4949
list(APPEND TVM_RUNTIME_LINKER_LIBS ${ROCM_ROCBLAS_LIBRARY})
5050
endif(USE_ROCBLAS)

cmake/modules/contrib/BLAS.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
# under the License.
1717

1818
# Plugin rules for cblas
19-
file(GLOB CBLAS_CONTRIB_SRC src/contrib/cblas/*.cc)
19+
file(GLOB CBLAS_CONTRIB_SRC src/runtime/contrib/cblas/*.cc)
2020

2121
if(USE_BLAS STREQUAL "openblas")
2222
find_library(BLAS_LIBRARY openblas)

cmake/modules/contrib/NNPack.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ if(USE_NNPACK)
2020
set(NNPACK_PATH ${CMAKE_CURRENT_SOURCE_DIR}/NNPack)
2121
endif()
2222
set(PTHREAD_POOL_PATH ${NNPACK_PATH}/deps/pthreadpool)
23-
file(GLOB NNPACK_CONTRIB_SRC src/contrib/nnpack/*.cc)
23+
file(GLOB NNPACK_CONTRIB_SRC src/runtime/contrib/nnpack/*.cc)
2424
list(APPEND RUNTIME_SRCS ${NNPACK_CONTRIB_SRC})
2525
include_directories(${NNPACK_PATH}/include)
2626
include_directories(${PTHREAD_POOL_PATH}/include)

cmake/modules/contrib/Random.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,6 @@
1717

1818
if(USE_RANDOM)
1919
message(STATUS "Build with contrib.random")
20-
file(GLOB RANDOM_CONTRIB_SRC src/contrib/random/random.cc)
20+
file(GLOB RANDOM_CONTRIB_SRC src/runtime/contrib/random/random.cc)
2121
list(APPEND RUNTIME_SRCS ${RANDOM_CONTRIB_SRC})
2222
endif(USE_RANDOM)

cmake/modules/contrib/Sort.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,6 @@
1717

1818
if(USE_SORT)
1919
message(STATUS "Build with contrib.sort")
20-
file(GLOB SORT_CONTRIB_SRC src/contrib/sort/*.cc)
20+
file(GLOB SORT_CONTRIB_SRC src/runtime/contrib/sort/*.cc)
2121
list(APPEND RUNTIME_SRCS ${SORT_CONTRIB_SRC})
2222
endif(USE_SORT)
Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
*/
1919

2020
/*!
21-
* Copyright (c) 2017 by Contributors
2221
* \file Use external cblas library call.
2322
*/
2423
#include <dmlc/logging.h>
Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
*/
1919

2020
/*!
21-
* Copyright (c) 2018 by Contributors
2221
* \file tvm/contrib/gemm.h
2322
* \brief Shared implementation of gemm
2423
*/
Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,9 @@
66
* to you under the Apache License, Version 2.0 (the
77
* "License"); you may not use this file except in compliance
88
* with the License. You may obtain a copy of the License at
9-
*
9+
*
1010
* http://www.apache.org/licenses/LICENSE-2.0
11-
*
11+
*
1212
* Unless required by applicable law or agreed to in writing,
1313
* software distributed under the License is distributed on an
1414
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
@@ -18,7 +18,6 @@
1818
*/
1919

2020
/*!
21-
* Copyright (c) 2018 by Contributors
2221
* \file Use external cblas library call.
2322
*/
2423
#include <tvm/runtime/registry.h>

0 commit comments

Comments
 (0)