diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt index 5fddd520e1760888c8af518c16de1d6425a7680c..382f36d430cf7f36e8d72768f51c1e6c7d80026a 100644 --- a/src/CMakeLists.txt +++ b/src/CMakeLists.txt @@ -64,6 +64,7 @@ target_include_directories(megbrain PUBLIC $ PRIVATE ${PROJECT_SOURCE_DIR}/third_party/midout/src ) + foreach (INCPATH IN LISTS MGB_INC) target_include_directories(megbrain PUBLIC $ diff --git a/src/core/include/megbrain/comp_node.h b/src/core/include/megbrain/comp_node.h index 647f74662787ceec1c77e3c3b92dd6aa0917109b..0b2250a883109bf0399534a1ff4303d61a707bda 100644 --- a/src/core/include/megbrain/comp_node.h +++ b/src/core/include/megbrain/comp_node.h @@ -119,6 +119,7 @@ class CompNode { * * device == -1 means logical default device (maps to 0 by default, * and can be changed by set_device_map) + * */ int device = -1; diff --git a/src/core/include/megbrain/comp_node_env.h b/src/core/include/megbrain/comp_node_env.h index 2d04a1cc03947bdabac800553938e5dfdb2aa42d..3b9f1af8b786e03a4ca73fc3501e543e2c311d4d 100644 --- a/src/core/include/megbrain/comp_node_env.h +++ b/src/core/include/megbrain/comp_node_env.h @@ -20,6 +20,7 @@ #include "megdnn/handle.h" + #if MGB_CUDA #include #include diff --git a/src/core/test/comp_node.cpp b/src/core/test/comp_node.cpp index 663d80b6d715340eeb32f9eb8ef2c81da8a07eed..05bd4c658b0c86f3f905b17d733915b21968f124 100644 --- a/src/core/test/comp_node.cpp +++ b/src/core/test/comp_node.cpp @@ -481,7 +481,6 @@ void test_peer_copy_from_device(const char* comp_node) { opr::Sleep::sleep(cn_gpu, 0.1); dev_a.copy_from(dev_c); wait_event->record(); - cn_cpu.device_wait_event(*wait_event); dev_b.copy_from(dev_a);