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 1997b1a289 feat(dnn/cuda): add correlation kernel 4 years ago
..
local chore(mge): update copyright years 4 years ago
megcore chore(mge): update copyright years 4 years ago
accuracy_shake.cpp feat(dnn): add conv_bwd_data and conv_bwd_filter accuracy shake check 4 years ago
adaptive_pooling.cpp chore(mge): update copyright years 4 years ago
add_update.cpp chore(mge): update copyright years 4 years ago
argmxx.cpp chore(mge): update copyright years 4 years ago
argsort.cpp chore(mge): update copyright years 4 years ago
batch_conv_bias.cpp fix(megdnn): fix megdnn benchmark testcase 4 years ago
batched_matrix_mul.cpp fix(megdnn): some fixes of execution policy 4 years ago
benchmark.cpp fix(megdnn): fix megdnn benchmark testcase 4 years ago
benchmark.h chore(mge): update copyright years 4 years ago
bn.cpp chore(mge): update copyright years 4 years ago
chanwise_convolution.cpp refactor(megdnn): add default algo for convolution forward 4 years ago
chanwise_convolution3d.cpp chore(mge): update copyright years 4 years ago
checksum.cpp chore(mge): update copyright years 4 years ago
concat.cpp chore(mge): update copyright years 4 years ago
cond_take.cpp chore(mge): update copyright years 4 years ago
conv_bias.cpp fix(dnn/cuda): add cuda nchw int8 conv impl with nchw4 to fix cu111 compatibility 4 years ago
conv_bias_int8.cpp feat(mgb): add recursive for fastrun and megdnn test 4 years ago
conv_pooling.cpp chore(mge): update copyright years 4 years ago
convolution.cpp feat(mgb/dnn): add accuracy shake checker 4 years ago
convolution3d.cpp chore(mge): update copyright years 4 years ago
correlation.cpp feat(dnn/cuda): add correlation kernel 4 years ago
cumsum.cpp chore(mge): update copyright years 4 years ago
cutlass_matmul.cpp refactor(mgb): move algo's name from info to desc and delete some algo's unnecessary param() method 4 years ago
cvt_color.cpp chore(mge): update copyright years 4 years ago
dct.cpp chore(mge): update copyright years 4 years ago
deformable_conv.cpp chore(mge): update copyright years 4 years ago
deformable_ps_roi_pooling.cpp chore(mge): update copyright years 4 years ago
dilated_convolution.cpp fix(megdnn): some fixes of execution policy 4 years ago
dot.cpp chore(mge): update copyright years 4 years ago
elemwise.cpp chore(mge): update copyright years 4 years ago
elemwise_multi_type.cpp chore(mge): update copyright years 4 years ago
eye.cpp chore(mge): update copyright years 4 years ago
fake_quant.cpp chore(mge): update copyright years 4 years ago
fixture.cpp chore(mge): update copyright years 4 years ago
fixture.h chore(mge): update copyright years 4 years ago
flip.cpp chore(mge): update copyright years 4 years ago
gaussian_blur.cpp chore(mge): update copyright years 4 years ago
group_conv.cpp refactor(megdnn): remove unnessary 1x1 algo 4 years ago
group_conv3d.cpp chore(mge): update copyright years 4 years ago
group_local.cpp chore(mge): update copyright years 4 years ago
images2neibs.cpp chore(mge): update copyright years 4 years ago
indexing_multi_axis_vec.cpp chore(mge): update copyright years 4 years ago
indexing_one_hot.cpp chore(mge): update copyright years 4 years ago
linspace.cpp chore(mge): update copyright years 4 years ago
local.cpp chore(mge): update copyright years 4 years ago
local_share.cpp feat(mgb): add recursive for fastrun and megdnn test 4 years ago
lrn.cpp chore(mge): update copyright years 4 years ago
mask_conv.cpp chore(mge): update copyright years 4 years ago
matrix_inverse.cpp chore(mge): update copyright years 4 years ago
matrix_mul.cpp fix(mgb/dnn): add cuda float naive matmul algo 4 years ago
mesh_indexing.cpp chore(mge): update copyright years 4 years ago
param_pack.cpp chore(mge): update copyright years 4 years ago
pooling.cpp chore(mge): update copyright years 4 years ago
powc.cpp chore(mge): update copyright years 4 years ago
reduce.cpp chore(mge): update copyright years 4 years ago
relayout.cpp chore(mge): update copyright years 4 years ago
relayout_format.cpp fix(dnn/cuda): add cuda nchw int8 conv impl with nchw4 to fix cu111 compatibility 4 years ago
remap.cpp chore(mge): update copyright years 4 years ago
repeat.cpp chore(mge): update copyright years 4 years ago
resize.cpp chore(mge): update copyright years 4 years ago
rng.cpp chore(mge): update copyright years 4 years ago
roi_align.cpp chore(mge): update copyright years 4 years ago
roi_copy.cpp chore(mge): update copyright years 4 years ago
roi_pooling.cpp chore(mge): update copyright years 4 years ago
rotate.cpp chore(mge): update copyright years 4 years ago
sleep.cpp chore(mge): update copyright years 4 years ago
split.cpp chore(mge): update copyright years 4 years ago
svd.cpp chore(mge): update copyright years 4 years ago
tensor_remap.cpp chore(mge): update copyright years 4 years ago
tile.cpp chore(mge): update copyright years 4 years ago
timer.h chore(mge): update copyright years 4 years ago
topk.cpp chore(mge): update copyright years 4 years ago
tqt.cpp chore(mge): update copyright years 4 years ago
transpose.cpp chore(mge): update copyright years 4 years ago
type_cvt.cpp chore(mge): update copyright years 4 years ago
utils.cpp chore(mge): update copyright years 4 years ago
utils.h chore(mge): update copyright years 4 years ago
warp_affine.cpp chore(mge): update copyright years 4 years ago
warp_perspective.cpp fix(mgb(ci)): fix tx1 ci testcase 4 years ago

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

C++ Cuda Python C SVG other