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.
2271 Commits
28 Branches
25 MB
3280 Download
 
 
 
 
 
 
Tree: 1add4517ad
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '1add4517ad'
${ noResults }
MegEngine/dnn/scripts
Megvii Engine Team 95ac055538 feat(dnn,mgb,imperative): add diag opr implement 3 years ago
..
cutlass_generator fix(dnn): fix_short_cutlass_name_gemm 3 years ago
gen_heuristic MegEngine: Initial commit of MegEngine. 5 years ago
Makefile refactor(dnn/cuda): refactor cutlass kernel generator for gemm and gemv 3 years ago
gen_cond_take_kern_impls.py feat(mge): add bfloat16 support 5 years ago
gen_cuda_batch_conv_bias_kern_impls.py MegEngine: Initial commit of MegEngine. 5 years ago
gen_cuda_conv_bias_kern_impls.py MegEngine: Initial commit of MegEngine. 5 years ago
gen_elemwise_each_mode.py MegEngine: Initial commit of MegEngine. 5 years ago
gen_elemwise_kern_impls.py fix(mgb/rocm): remove begin-internal of rocm 4 years ago
gen_elemwise_multi_type_kern_impls.py MegEngine: Initial commit of MegEngine. 5 years ago
gen_elemwise_multi_type_utils.py feat(mge/opr): add silu and gelu 3 years ago
gen_elemwise_special_kern_impls.py fix(mgb/rocm): remove begin-internal of rocm 4 years ago
gen_elemwise_utils.py feat(mge/opr): add silu and gelu 3 years ago
gen_flatbuffers_converter.py fix(mgb/serialize): fix flatbuffer compatibility issues 3 years ago
gen_flatbuffers_schema.py fix(mgb/serialize): fix flatbuffer compatibility issues 3 years ago
gen_param_defs.py refactor(mgb/dnn): refactor enum used in serializing 3 years ago
gen_tablegen.py fix(mgb/serialize): fix flatbuffer compatibility issues 3 years ago
opr_param_defs.py feat(dnn,mgb,imperative): add diag opr implement 3 years ago
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