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.
 
 
 
 
 
 
Megvii Engine Team f7b2bdae1a refactor(dnn): refactor algorithm type interface 5 years ago
..
add_update fix(mgb/rocm): remove begin-internal of rocm 5 years ago
argmxx fix(mgb/rocm): remove begin-internal of rocm 5 years ago
batch_normalization feat(megdnn/rocm): add bn for rocm backend 5 years ago
batched_matrix_mul refactor(dnn): refactor algorithm type interface 5 years ago
checksum fix(mgb/rocm): remove begin-internal of rocm 5 years ago
convolution refactor(dnn): refactor algorithm type interface 5 years ago
elemwise fix(dnn): generate mode files 5 years ago
eye fix(mgb/rocm): remove begin-internal of rocm 5 years ago
indexing_multi_axis_vec feat(mgb/rocm): add cmake for rocm and fix compile errors and bn 5 years ago
indexing_one_hot fix(mgb/rocm): remove begin-internal of rocm 5 years ago
linspace fix(mgb/rocm): remove begin-internal of rocm 5 years ago
matrix_mul fix(mgb/rocm): remove begin-internal of rocm 5 years ago
megcore fix(mgb/rocm): remove begin-internal of rocm 5 years ago
pooling fix(mgb/rocm): remove begin-internal of rocm 5 years ago
powc fix(mgb/rocm): remove begin-internal of rocm 5 years ago
reduce feat(mgb/rocm): add cmake for rocm and fix compile errors and bn 5 years ago
reduce_helper fix(mgb/rocm): remove begin-internal of rocm 5 years ago
relayout fix(mgb/rocm): remove begin-internal of rocm 5 years ago
rng fix(mgb/rocm): remove begin-internal of rocm 5 years ago
sleep fix(rocm): fix rocm megdnntest sleep and a cut code 5 years ago
type_cvt feat(mgb/rocm): add cmake for rocm and fix compile errors and bn 5 years ago
elemwise_helper.cpp feat(mgb/rocm): add cmake for rocm and fix compile errors and bn 5 years ago
elemwise_helper.h.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
error_info.h.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
handle.cpp feat(megdnn/rocm): add bn for rocm backend 5 years ago
handle.h fix(mgb/rocm): remove begin-internal of rocm 5 years ago
int_fastdiv.cpp fix(mgb/rocm): remove begin-internal of rocm 5 years ago
int_fastdiv.h.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
miopen_with_check.h fix(mgb/rocm): remove begin-internal of rocm 5 years ago
miopen_wrapper.cpp feat(megdnn/rocm): add average inclusive mode for pooling 5 years ago
miopen_wrapper.h fix(mgb/rocm): remove begin-internal of rocm 5 years ago
reduce_helper.h.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
reduce_helper.hipinl fix(mgb/rocm): remove begin-internal of rocm 5 years ago
rocblas_header.h fix(mgb/rocm): remove begin-internal of rocm 5 years ago
utils.cpp fix(mgb/rocm): remove begin-internal of rocm 5 years ago
utils.h fix(mgb/rocm): remove begin-internal of rocm 5 years ago
utils.h.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago

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

C++ Cuda Python C SVG other