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 4b141f8de4 fix(mgb): add usable-depend-on-shape attr 4 years ago
..
adaptive_pooling chore(mge): update copyright years 5 years ago
add_update feat(log): opt log 4 years ago
argmxx chore(mge): update copyright years 5 years ago
argsort chore(mge): update copyright years 5 years ago
batch_conv_bias refactor(mgb/dnn): add negative attribute for algo 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
concat chore(mge): update copyright years 5 years ago
cond_take chore(mge): update copyright years 5 years ago
conv_bias refactor(mgb): move algo's name from info to desc and delete some algo's unnecessary param() method 4 years ago
convolution fix(mgb): add usable-depend-on-shape attr 4 years ago
convolution3d refactor(mgb): move algo's name from info to desc and delete some algo's unnecessary param() method 4 years ago
convolution_helper chore(mge): update copyright years 5 years ago
convpooling chore(mge): update copyright years 5 years ago
cub MegEngine: Initial commit of MegEngine. 5 years ago
cumsum feat(log): opt log 4 years ago
cv chore(mge): update copyright years 5 years ago
cvt_color chore(mge): update copyright years 5 years ago
dct chore(mge): update copyright years 5 years ago
deformable_conv refactor(mgb/dnn): add negative attribute for algo 4 years ago
deformable_ps_roi_pooling chore(mge): update copyright years 5 years ago
dot chore(mge): update copyright years 5 years ago
elemwise chore(mge): update copyright years 5 years ago
elemwise_multi_type chore(mge): update copyright years 5 years ago
eye chore(mge): update copyright years 5 years ago
fake_quant chore(mge): update copyright years 5 years ago
flip chore(mge): update copyright years 5 years ago
gaussian_blur chore(mge): update copyright years 5 years ago
group_local chore(mge): update copyright years 5 years ago
images2neibs 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
local chore(mge): update copyright years 5 years ago
local_share fix(mgb): add usable-depend-on-shape attr 4 years ago
lrn chore(mge): update copyright years 5 years ago
mask_conv chore(mge): update copyright years 5 years ago
matrix_inverse chore(mge): update copyright years 5 years ago
matrix_mul fix(mgb): add usable-depend-on-shape attr 4 years ago
max_tensor_diff chore(mge): update copyright years 5 years ago
megcore feat(log): opt log 4 years ago
mesh_indexing chore(mge): update copyright years 5 years ago
param_pack chore(mge): update copyright years 5 years ago
pooling chore(mge): update copyright years 5 years ago
powc chore(mge): update copyright years 5 years ago
reduce chore(mge): update copyright years 5 years ago
reduce_helper chore(mge): update copyright years 5 years ago
relayout feat(log): opt log 4 years ago
relayout_format fix(dnn/cuda): add cuda nchw int8 conv impl with nchw4 to fix cu111 compatibility 4 years ago
remap feat(log): opt log 4 years ago
repeat chore(mge): update copyright years 5 years ago
resize feat(log): opt log 4 years ago
rng chore(mge): update copyright years 5 years ago
roi_align chore(mge): update copyright years 5 years ago
roi_copy chore(mge): update copyright years 5 years ago
roi_pooling chore(mge): update copyright years 5 years ago
rotate chore(mge): update copyright years 5 years ago
separable_conv chore(mge): update copyright years 5 years ago
separable_filter chore(mge): update copyright years 5 years ago
sleep chore(mge): update copyright years 5 years ago
split chore(mge): update copyright years 5 years ago
svd chore(mge): update copyright years 5 years ago
tensor_remap chore(mge): update copyright years 5 years ago
tile chore(mge): update copyright years 5 years ago
topk chore(mge): update copyright years 5 years ago
tqt fix(mgb/dnn): fix backward computation of tqt 4 years ago
transpose chore(mge): update copyright years 5 years ago
type_cvt chore(mge): update copyright years 5 years ago
warp_affine feat(log): opt log 4 years ago
warp_perspective feat(log): opt log 4 years ago
cuda_shfl_compat.cuh chore(mge): update copyright years 5 years ago
cudnn_with_check.h chore(mge): update copyright years 5 years ago
cudnn_wrapper.cpp feat(log): opt log 4 years ago
cudnn_wrapper.h build(cuda): support cu111 build 4 years ago
elemwise_helper.cpp chore(mge): update copyright years 5 years ago
elemwise_helper.cuh chore(mge): update copyright years 5 years ago
error_info.cuh chore(mge): update copyright years 5 years ago
fp16_help.cuh chore(mge): update copyright years 5 years ago
handle.cpp fix(dnn): fix cudnn crash when finalize called after cudnn dtor 4 years ago
handle.h fix(opencl): enable image on mali(cl2.1) 5 years ago
handle_create.cpp chore(mge): update copyright years 5 years ago
int_fastdiv.cpp chore(mge): update copyright years 5 years ago
int_fastdiv.cuh chore(mge): update copyright years 5 years ago
query_blocksize.cpp chore(mge): update copyright years 5 years ago
query_blocksize.cuh chore(mge): update copyright years 5 years ago
query_blocksize_impl.cu chore(mge): update copyright years 5 years ago
reduce_helper.cuh chore(mge): update copyright years 5 years ago
reduce_helper.cuinl chore(mge): update copyright years 5 years ago
utils.cpp chore(mge): update copyright years 5 years ago
utils.cuh fix(dnn/cuda): add cuda nchw int8 conv impl with nchw4 to fix cu111 compatibility 4 years ago
utils.h chore(mge): update copyright years 5 years ago

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

C++ Cuda Python C SVG other