Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion cmake/modules/CUDA.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ if(USE_CUDA)
message(STATUS "Build cuda device runtime")

tvm_file_glob(GLOB RUNTIME_CUDA_SRCS src/backend/cuda/runtime/*.cc)
tvm_file_glob(GLOB VM_CUDA_BUILTIN_SRC_CC src/backend/cuda/runtime/vm/*.cc)
tvm_file_glob(GLOB VM_CUDA_BUILTIN_SRC_CC src/runtime/vm/cuda/*.cc)

add_library(tvm_runtime_cuda_objs OBJECT ${RUNTIME_CUDA_SRCS} ${VM_CUDA_BUILTIN_SRC_CC})
target_link_libraries(tvm_runtime_cuda_objs PUBLIC tvm_ffi_header)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
*/

/*!
* \file src/runtime/vm/cuda_graph_builtin.cc
* \file src/runtime/vm/cuda/cuda_graph_builtin.cc
* \brief The CUDA graph related builtin functions for Relax virtual machine.
*/

Expand All @@ -28,8 +28,8 @@
#include <tvm/ffi/reflection/registry.h>
#include <tvm/runtime/vm/vm.h>

#include "../../../../support/utils.h"
#include "../cuda_common.h"
#include "../../../backend/cuda/runtime/cuda_common.h"
#include "../../../support/utils.h"
namespace tvm {
namespace runtime {
namespace vm {
Expand Down
Loading