OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
2853 Commits
28 Branches
25 MB
3464 Download
 
 
 
 
 
 
Tree: dbd9483993
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'dbd9483993'
${ noResults }
MegEngine/dnn/scripts/cutlass_generator
Megvii Engine Team 4e9b1c4eee feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 2 years ago
..
BUILD feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 2 years ago
README.md feat(dnn/cuda): generate cutlass kimpls using cmake and bazel 3 years ago
conv2d_operation.py feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 2 years ago
gemm_operation.py style(mgb/tools): add format for tools, dnn and ci 2 years ago
gen_list.py feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 2 years ago
generator.py feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 2 years ago
library.py feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 2 years ago
list.bzl feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 2 years ago
manifest.py style(mgb/tools): add format for tools, dnn and ci 2 years ago

README.md

Generate device kernel registration code for CUTLASS kernels

Usage

python3 generator.py [--operations {gemm, gemv, conv2d, deconv}] [--type {simt, tensorop8816, tensorop8832}]
                     output
  • operations: operation kind, including gemm|gemv|conv2d|deconv
  • type: opcode class, simt|tensorop8816|tensorop8832
  • output: the output directory for CUTLASS kernels

Generate file list for bazel

We generate list.bzl because the genrule method of bazel requires that the output file list be specified in the analysis phase.

Please call gen_list.py when new operations are added.

python3 gen_list.py
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境，不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序，请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉，欢迎访问 MegStudio 平台

C++ Cuda Python C SVG other