MegDNN

MegDNN 是 MegEngine 的底层计算引擎，位于 dnn 目录。 它提供了与 dense tensor 相关的计算原语，比如 Convolution, Pooling, MatrixMul, Transpose 等。 它是跨平台的计算库，支持 x86 (with SSE4.2), arm (with NEON), CUDA (Kepler/Maxwell/Pascal)， OpenCL(mali/adreno/powervr/cuda/intel, etc), Hexagon 等。

MegDNN 文件结构

dnn
├── atlas-stub
├── cuda-stub
├── include
│   ├── megdnn
│   │   ├── config
│   │   ├── dtype # 定义数据类型
│   │   ├── internal  # 仅供 MegDNN 内部实现使用的头文件
│   │   │   ├── defs.h  # 定义了与 TensorND 的维数（ndim）相关的宏
│   │   │   ├── opr_header_epilogue.h  # 定义 MegDNN operator 需要的宏
│   │   │   ├── opr_header_prologue.h  # 将 opr_header_prologue.h 中的宏 undef 掉
│   │   │   ├── visibility_epilogue.h  # 定义了与 visibility 相关的宏
│   │   │   └── visibility_prologue.h  # 将 visibility_prologue.h 中定义的宏 undef 掉
│   │   ├── oprs  # 定义了 MegDNN 的所有 operator
│   │   ├── thin  # 相较于 std 的一些简化实现
│   │   ├── arch.h  # 定义了一些平台与编译器相关的宏
│   │   ├── basic_types.h  # 定义了一些 MegDNN 基础类型，如 ErrorHandler/TensorShape/TensorLayout/TensorND/Workspace 等
│   │   ├── common.h
│   │   ├── cuda.h  # 定义了一些 CUDA 特有的 API
│   │   ├── dtype.h
│   │   ├── handle.h
│   │   ├── opr_param_defs.h  # 定义了一些结构体，表示 operator 的 param （由 scripts/gen_param_defs.py 自动生成）
│   │   ├── opr_result_defs.h  # 定义了一些结构体，表示 operator 的 result （由 scripts/gen_param_defs.py 自动生成）
│   │   ├── oprs.h  # 包含了 oprs 文件夹下的所有头文件
│   │   ├── tensor_format.h
│   │   ├── tensor_iter.h # 定义了 TensorIter 类，方便枚举 TensorND 内的所有元素。
│   │   └── version.h  # 定义了 MegDNN 的 version
│   ├── hip_header.h
│   ├── megcore_*.h  # MegCore 是一个跨平台线程管理库，将 CUDA "device" 和 "stream" 的概念与 CPU NUMA（尚未实现）和线程的概念抽象出来。
│   └── megdnn.h # MegDNN 的主头文件，MegDNN 的用户应使用此头文件。
├── scripts
├── src
│   ├── aarch64
│   ├── arm_common
│   ├── armv7
│   ├── atlas
│   ├── cambricon
│   ├── common # 定义了各个平台的公共代码，比如各个 operator 的 deduce_layout, OperatorBase, Handle 的方法定义。
│   ├── cuda
│   ├── fallback
│   ├── naive
│   ├── rocm
│   ├── x86
│   └── CMakeLists.txt
├── test
└── CMakeLists.txt