• 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
cudnn.cmake 2.6 KB