m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '410907f624'
${ noResults }
Paddle/cmake/anakin_subgraph.cmake

34 lines
917 B
Raw Normal View History Unescape

1. Add ANAKIN_ROOT compile option 2. refine trt code test=develop
6 years ago
if(NOT WITH_GPU)
return()
endif()
set(ANAKIN_ROOT "/usr" CACHE PATH "ANAKIN ROOT")
find_path(ANAKIN_INCLUDE_DIR anakin_config.h
PATHS ${ANAKIN_ROOT} ${ANAKIN_ROOT}/include
$ENV{ANAKIN_ROOT} $ENV{ANAKIN_ROOT}/include
NO_DEFAULT_PATH
)
find_library(ANAKIN_LIBRARY NAMES libanakin_saber_common.so libanakin.so
PATHS ${ANAKIN_ROOT}
$ENV{ANAKIN_ROOT} $ENV{ANAKIN_ROOT}/lib
NO_DEFAULT_PATH
DOC "Path to ANAKIN library.")
if(ANAKIN_INCLUDE_DIR AND ANAKIN_LIBRARY)
if(WITH_DSO)
set(ANAKIN_FOUND ON)
endif(WITH_DSO)
else()
set(ANAKIN_FOUND OFF)
endif()
if(ANAKIN_FOUND)
message(STATUS "Current ANAKIN header is ${ANAKIN_INCLUDE_DIR}/anakin_config.h. ")
Cherry-pick benchmark related changes from release/1.4 (#17156) * cherry-pick commit from 8877054 * cherry-pick commit from 3f0b97d * cherry-pick from 16691:Anakin subgraph support yolo_v3 and faster-rcnn (cherry picked from commit 8643dbc233f12f829b64cc0ee6926e41fb891ddf) * Cherry-Pick from 16662 : Anakin subgraph cpu support (cherry picked from commit 7ad182e16cbd099523dd274d3b4051b3734c9adf) * Cherry-pick from 1662, 16797.. : add anakin int8 support (cherry picked from commit e14ab180fe76b97aa33c0089f98d1cfa771905e9) * Cherry-pick from 16813 : change singleton to graph RegistBlock test=release/1.4 (cherry picked from commit 4b9fa42307aeb90c8e2710ad07d02e286a4620aa) * Cherry Pick : 16837 Support ShuffleNet and MobileNet-v2 Support ShuffleNet and MobileNet-v2, test=release/1.4 (cherry picked from commit a6fb066f90d1009ab32e981cf8e7d47d55bbc9e6) * Cherry-pick : anakin subgraph add opt config layout argument #16846 test=release/1.4 (cherry picked from commit 8121b3eccbd57a41e448ae0e1e716d634abd338d) * 1. add shuffle_channel_detect (cherry picked from commit 6efdea8997cda4d737777f9be89cc9991120df64) * update shuffle_channel op convert, test=release/1.4 (cherry picked from commit e4726a066fd6b7790745cb02a7601531c556793b) * Modify symbol export rules test=develop
6 years ago
include_directories(${ANAKIN_ROOT})
1. Add ANAKIN_ROOT compile option 2. refine trt code test=develop
6 years ago
include_directories(${ANAKIN_ROOT}/include)
Cherry-pick benchmark related changes from release/1.4 (#17156) * cherry-pick commit from 8877054 * cherry-pick commit from 3f0b97d * cherry-pick from 16691:Anakin subgraph support yolo_v3 and faster-rcnn (cherry picked from commit 8643dbc233f12f829b64cc0ee6926e41fb891ddf) * Cherry-Pick from 16662 : Anakin subgraph cpu support (cherry picked from commit 7ad182e16cbd099523dd274d3b4051b3734c9adf) * Cherry-pick from 1662, 16797.. : add anakin int8 support (cherry picked from commit e14ab180fe76b97aa33c0089f98d1cfa771905e9) * Cherry-pick from 16813 : change singleton to graph RegistBlock test=release/1.4 (cherry picked from commit 4b9fa42307aeb90c8e2710ad07d02e286a4620aa) * Cherry Pick : 16837 Support ShuffleNet and MobileNet-v2 Support ShuffleNet and MobileNet-v2, test=release/1.4 (cherry picked from commit a6fb066f90d1009ab32e981cf8e7d47d55bbc9e6) * Cherry-pick : anakin subgraph add opt config layout argument #16846 test=release/1.4 (cherry picked from commit 8121b3eccbd57a41e448ae0e1e716d634abd338d) * 1. add shuffle_channel_detect (cherry picked from commit 6efdea8997cda4d737777f9be89cc9991120df64) * update shuffle_channel op convert, test=release/1.4 (cherry picked from commit e4726a066fd6b7790745cb02a7601531c556793b) * Modify symbol export rules test=develop
6 years ago
include_directories(${ANAKIN_ROOT}/saber)
1. Add ANAKIN_ROOT compile option 2. refine trt code test=develop
6 years ago
link_directories(${ANAKIN_ROOT})
add_definitions(-DPADDLE_WITH_ANAKIN)
endif()