Skip to content

MegEngine
MegEngine

MegEngine 天元 / MegEngine
1 年多 前同步成功

通知 404
Star 4705
Fork 582
MegEngine
MegEngine
切换分支/标签
  1. 11 9月, 2021 1 次提交
  3. 22 3月, 2021 1 次提交
    • M
      feat(cmake/arm/cuda): support cmake arm with cuda, · 914af286
      由 Megvii Engine Team 提交于 
      * support arm-host and x86-cross-arm
* fix trt myelib cmake build issue at 'trt copy env'(about 'copy env', Please refs cb92123f)

about x86-cross-arm CUDA env:
1: run ./scripts/cmake-build/create_cuda_build_libs.py to prepare
   cuda/cudnn/trt env(download deb package info, Please refs create_cuda_build_libs.py)
2: export TRT_ROOT_DIR=xxxxx which may create by step 1
3: export CUDNN_ROOT_DIR=xxxx which may create by step 1
4: export PATH=xxxx:$PATH  xxxx need create by step 1, which nvcc have
   relative path dir: ../targets/sbsa-linux/
GitOrigin-RevId: 440c76052aabe5b07a4b64d126e759f919c257a8
      914af286
  5. 02 2月, 2021 1 次提交
    • M
      fix(cmake/cuda): fix build at cuda `copy` env caused by b278a69e1 · 329306b0
      由 Megvii Engine Team 提交于 
      `copy` env vs `normal` env:
case a:`normal` env install CUDNN/TRT by package tool, eg, apt/yum, which
will put include file(cudnn.h/NvInfer.h) to /usr/include etc.

case b: self copy env after export CPATH with cudnn/trt include dir

case c: just copy libs, not export CPATH, only export
TRT_ROOT_DIR/CUDNN_ROOT_DIR

at case a and case b, BUILD system can find cudnn/trt headers, but
at case c, can not find. `FIX` it!

what`s more, recommend install CUDA env by case a and case c, NOT
recommend install by case b, which caused by CUDA SDK CPATH not
friendly to split c env, for example Android AOSP build env

GitOrigin-RevId: cb92123f9e58d730893fff45602d3e487b38d32c
      329306b0
  7. 27 11月, 2020 1 次提交
  9. 25 8月, 2020 1 次提交
  11. 25 3月, 2020 1 次提交