wangwei
/
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.
1561 Commits
28 Branches
25 MB
604 Download
 
 
 
 
 
 
Tree: 1cfdbc565c
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '1cfdbc565c'
${ noResults }
MegEngine/dnn/src/rocm
Megvii Engine Team a5060a2bfe feat(mgb/opr): add check_has_inf kernel and opr 4 years ago
..
add_update feat(log): opt log 4 years ago
argmxx feat(log): opt log 4 years ago
batch_normalization feat(log): opt log 4 years ago
batched_matrix_mul refactor(mgb/dnn): add negative attribute for algo 4 years ago
checksum chore(mge): update copyright years 5 years ago
convolution fix(dnn): replace kernel launch syntax with macro for hcc 4 years ago
elemwise chore(mge): update copyright years 5 years ago
eye chore(mge): update copyright years 5 years ago
indexing_multi_axis_vec feat(log): opt log 4 years ago
indexing_one_hot feat(log): opt log 4 years ago
linspace chore(mge): update copyright years 5 years ago
matrix_mul fix(mgb): add usable-depend-on-shape attr 4 years ago
megcore chore(mge): update copyright years 5 years ago
param_pack feat(dnn): add param_pack for rocm 4 years ago
pooling chore(mge): update copyright years 5 years ago
powc chore(mge): update copyright years 5 years ago
reduce feat(mgb/opr): add check_has_inf kernel and opr 4 years ago
reduce_helper fix(mgb/rocm): remove begin-internal of rocm 5 years ago
relayout chore(mge): update copyright years 5 years ago
rng chore(mge): update copyright years 5 years ago
sleep chore(mge): update copyright years 5 years ago
type_cvt chore(mge): update copyright years 5 years ago
elemwise_helper.cpp chore(mge): update copyright years 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(dnn): add param_pack for rocm 4 years ago
handle.h build(rocm): support rocm-3.9 4 years ago
int_fastdiv.cpp chore(mge): update copyright years 5 years ago
int_fastdiv.h.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
miopen_with_check.h build(rocm): partially support hcc compilation 4 years ago
miopen_wrapper.cpp feat(log): opt log 4 years ago
miopen_wrapper.h chore(mge): update copyright years 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 chore(mge): update copyright years 5 years ago
utils.cpp chore(mge): update copyright years 5 years ago
utils.h chore(mge): update copyright years 5 years ago
utils.h.hip fix(mgb/rocm): remove begin-internal of rocm 5 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