@@ -64,6 +64,7 @@ target_include_directories(megbrain | |||||
PUBLIC $<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}> | PUBLIC $<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}> | ||||
PRIVATE ${PROJECT_SOURCE_DIR}/third_party/midout/src | PRIVATE ${PROJECT_SOURCE_DIR}/third_party/midout/src | ||||
) | ) | ||||
foreach (INCPATH IN LISTS MGB_INC) | foreach (INCPATH IN LISTS MGB_INC) | ||||
target_include_directories(megbrain | target_include_directories(megbrain | ||||
PUBLIC $<BUILD_INTERFACE:${INCPATH}> | PUBLIC $<BUILD_INTERFACE:${INCPATH}> | ||||
@@ -119,6 +119,7 @@ class CompNode { | |||||
* | * | ||||
* device == -1 means logical default device (maps to 0 by default, | * device == -1 means logical default device (maps to 0 by default, | ||||
* and can be changed by set_device_map) | * and can be changed by set_device_map) | ||||
* | |||||
*/ | */ | ||||
int device = -1; | int device = -1; | ||||
@@ -20,6 +20,7 @@ | |||||
#include "megdnn/handle.h" | #include "megdnn/handle.h" | ||||
#if MGB_CUDA | #if MGB_CUDA | ||||
#include <cuda_runtime.h> | #include <cuda_runtime.h> | ||||
#include <cuda.h> | #include <cuda.h> | ||||
@@ -481,7 +481,6 @@ void test_peer_copy_from_device(const char* comp_node) { | |||||
opr::Sleep::sleep(cn_gpu, 0.1); | opr::Sleep::sleep(cn_gpu, 0.1); | ||||
dev_a.copy_from(dev_c); | dev_a.copy_from(dev_c); | ||||
wait_event->record(); | wait_event->record(); | ||||
cn_cpu.device_wait_event(*wait_event); | cn_cpu.device_wait_event(*wait_event); | ||||
dev_b.copy_from(dev_a); | dev_b.copy_from(dev_a); | ||||