关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>
OpenI
/
MegEngine
mirror of https://github.com/MegEngine/MegEngine.git
Not watched
23
13
Fork 8
Code
Releases 26 Wiki Activity
Issues 0
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.
3244 Commits
40 Branches
26 MiB
1054 Download
 
 
 
 
 
 
Branch: master
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'master'
${ noResults }
MegEngine/dnn/scripts/cutlass_generator
Megvii Engine Team 4e9b1c4eee feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 1 year ago
..
BUILD feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 1 year ago
README.md feat(dnn/cuda): generate cutlass kimpls using cmake and bazel 2 years ago
conv2d_operation.py feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 1 year ago
gemm_operation.py style(mgb/tools): add format for tools, dnn and ci 1 year ago
gen_list.py feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 1 year ago
generator.py feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 1 year ago
library.py feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 1 year ago
list.bzl feat(dnn): add rrconv wgrad, support int32 and uint8 region mask 1 year ago
manifest.py style(mgb/tools): add format for tools, dnn and ci 1 year 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）是旷视自主研发的开源深度学习框架，于2020年3月正式在 OpenI 启智社区开源，天元能够帮助开发者高效的完成深度学习算法的设计、训练、部署，有效提升AI研发工作效率。

https://megengine.org.cn/

C++ Cuda Python C Starlark other

Please read the following content carefully:

Dear OpenI User

Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.

For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》

Agree and continue
Disagree, exit