提交 1310ad49 编写于 作者: M Megvii Engine Team

feat(dnn): add memory manager for enflame

GitOrigin-RevId: 72a810d959777ac84275210861d525af8767e009
上级 2fd3fa88
...@@ -64,6 +64,7 @@ target_include_directories(megbrain ...@@ -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 { ...@@ -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 @@ ...@@ -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) { ...@@ -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);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册