Skip to content

MegEngine
MegEngine

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

通知 404
Star 4705
Fork 582
MegEngine
MegEngine
切换分支/标签
  1. 23 3月, 2021 3 次提交
  3. 22 3月, 2021 16 次提交
  5. 21 2月, 2021 1 次提交
  7. 05 2月, 2021 17 次提交
  9. 02 2月, 2021 3 次提交
    • M
      fix(dnn): fix bazel build issue for cambricon platform · 2b8150ab
      由 Megvii Engine Team 提交于 
      GitOrigin-RevId: a79e1b6f6aff177308b11f1032b8e954e632d3da
      2b8150ab
    • M
      fix(dnn/opencl): fix elemwise negative stride support · 0e8b81c2
      由 Megvii Engine Team 提交于 
      GitOrigin-RevId: 506d7e61043d84923a317722d295a7e3cf591341
      0e8b81c2
    • 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