Browse Source

feat(dnn): add memory manager for enflame

GitOrigin-RevId: 72a810d959
tags/v1.3.1
Megvii Engine Team 4 years ago
parent
commit
1310ad4926
4 changed files with 3 additions and 1 deletions
  1. +1
    -0
      src/CMakeLists.txt
  2. +1
    -0
      src/core/include/megbrain/comp_node.h
  3. +1
    -0
      src/core/include/megbrain/comp_node_env.h
  4. +0
    -1
      src/core/test/comp_node.cpp

+ 1
- 0
src/CMakeLists.txt View File

@@ -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}>


+ 1
- 0
src/core/include/megbrain/comp_node.h View File

@@ -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;



+ 1
- 0
src/core/include/megbrain/comp_node_env.h View File

@@ -20,6 +20,7 @@
#include "megdnn/handle.h"



#if MGB_CUDA
#include <cuda_runtime.h>
#include <cuda.h>


+ 0
- 1
src/core/test/comp_node.cpp View File

@@ -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);



Loading…
Cancel
Save