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

feat(dnn): add memory manager for enflame

GitOrigin-RevId: 72a810d959777ac84275210861d525af8767e009
上级 2fd3fa88
......@@ -64,6 +64,7 @@ target_include_directories(megbrain
PUBLIC $<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>
PRIVATE ${PROJECT_SOURCE_DIR}/third_party/midout/src
)
foreach (INCPATH IN LISTS MGB_INC)
target_include_directories(megbrain
PUBLIC $<BUILD_INTERFACE:${INCPATH}>
......
......@@ -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;
......
......@@ -20,6 +20,7 @@
#include "megdnn/handle.h"
#if MGB_CUDA
#include <cuda_runtime.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);
dev_a.copy_from(dev_c);
wait_event->record();
cn_cpu.device_wait_event(*wait_event);
dev_b.copy_from(dev_a);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册