Skip to content

MegEngine
MegEngine

陈六生 / MegEngine
与 Fork 源项目一致

Fork自 MegEngine 天元 / MegEngine

通知 1
Star 0
Fork 0
MegEngine
MegEngine

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

切换分支/标签
  1. 23 3月, 2021 2 次提交
  3. 22 3月, 2021 12 次提交
  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(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
    • M
      chore(mge): update copyright years · 364afec0
      由 Megvii Engine Team 提交于 
      GitOrigin-RevId: 3c0690bcc130e60568ee0bbfafdd1c19dc073273
      364afec0
  11. 23 1月, 2021 1 次提交
    • M
      fix(cmake/whl): reduce wheel size · ae8b38f6
      由 Megvii Engine Team 提交于 
      as some reason: some version of trt/cudnn need
-Wl,--whole-archive attribute to fix cuda issue,
target A(CMAKE OBJECT,eg megbrain/megdnn): -Wl,--whole-archive depend on trt/cudnn
target B(SHARED LIBS,eg libmegengine.so) --> A with `PUBLIC` depends
target C(SHARED LIBS,eg _impertive.so) --> B with `PUBLIC` depends,
then ld will force link fatbin section into C, which will be undesired

what`s more, attribute PUBLIC/PRIVATE do not take effect
to OBJECT library(megbrain/megdnn)

what`s more, megengine/megengine_export have to PUBLIC for
mgb/imperative target, as SRC/include depends

so we pull cudalib depend from megbrain/megdnn to
megengine/megengine_export on linux os, to finall
target at windows os, for example lar on windows

GitOrigin-RevId: b278a69e1cdcc17cfd22da80247af3fe30f6d4e8
      ae8b38f6
  13. 21 1月, 2021 3 次提交
  15. 15 1月, 2021 1 次提交