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 c59be192cd feat(dnn/arm_common/elemwise): add arm_common support chw44 elemwise 5 years ago
..
argmxx MegEngine: Initial commit of MegEngine. 5 years ago
cond_take MegEngine: Initial commit of MegEngine. 5 years ago
cv fix(mgb): improve exception safety with smart pointers 5 years ago
elemwise feat(dnn/arm_common/elemwise): add arm_common support chw44 elemwise 5 years ago
elemwise_multi_type MegEngine: Initial commit of MegEngine. 5 years ago
local MegEngine: Initial commit of MegEngine. 5 years ago
local_share MegEngine: Initial commit of MegEngine. 5 years ago
megcore feat(mgb/core): add comp node for cambricon 5 years ago
pooling MegEngine: Initial commit of MegEngine. 5 years ago
simd_macro MegEngine: Initial commit of MegEngine. 5 years ago
winograd fix(mge/cmake): fix cmake for opencl and add some minor fixes 5 years ago
add_update.cpp MegEngine: Initial commit of MegEngine. 5 years ago
add_update_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
algo_chooser.h MegEngine: Initial commit of MegEngine. 5 years ago
argmxx_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
argsort.cpp MegEngine: Initial commit of MegEngine. 5 years ago
asm_common_defs.h MegEngine: Initial commit of MegEngine. 5 years ago
basic_types.cpp MegEngine: Initial commit of MegEngine. 5 years ago
batch_conv_bias.cpp MegEngine: Initial commit of MegEngine. 5 years ago
batch_normalization.cpp MegEngine: Initial commit of MegEngine. 5 years ago
batched_matrix_mul.cpp MegEngine: Initial commit of MegEngine. 5 years ago
checksum.cpp MegEngine: Initial commit of MegEngine. 5 years ago
concat_split.cpp MegEngine: Initial commit of MegEngine. 5 years ago
conv_bias.cpp feat(dnn): add nchw44 layout 5 years ago
conv_bias.h MegEngine: Initial commit of MegEngine. 5 years ago
conv_pooling.cpp MegEngine: Initial commit of MegEngine. 5 years ago
convolution.cpp feat(dnn): add nchw44 layout 5 years ago
convolution3d.cpp MegEngine: Initial commit of MegEngine. 5 years ago
cumsum.cpp MegEngine: Initial commit of MegEngine. 5 years ago
cvt_color.cpp MegEngine: Initial commit of MegEngine. 5 years ago
deformable_conv.cpp MegEngine: Initial commit of MegEngine. 5 years ago
deformable_ps_roi_pooling.cpp MegEngine: Initial commit of MegEngine. 5 years ago
dot.cpp MegEngine: Initial commit of MegEngine. 5 years ago
dtype.cpp MegEngine: Initial commit of MegEngine. 5 years ago
elemwise_helper.cpp MegEngine: Initial commit of MegEngine. 5 years ago
elemwise_helper.cuh MegEngine: Initial commit of MegEngine. 5 years ago
eye.cpp MegEngine: Initial commit of MegEngine. 5 years ago
flag_warn.cpp MegEngine: Initial commit of MegEngine. 5 years ago
flip.cpp MegEngine: Initial commit of MegEngine. 5 years ago
gaussian_blur.cpp MegEngine: Initial commit of MegEngine. 5 years ago
gaussian_blur_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
group_local.cpp MegEngine: Initial commit of MegEngine. 5 years ago
handle.cpp MegEngine: Initial commit of MegEngine. 5 years ago
handle_impl.h refactor(dnn/parampack): remove parampacksplit 5 years ago
hash_ct.h feat(dnn/aarch64): add arm64 nchw44 mk matmul 5 years ago
images2neibs.cpp MegEngine: Initial commit of MegEngine. 5 years ago
indexing_multi_axis_vec.cpp MegEngine: Initial commit of MegEngine. 5 years ago
indexing_multi_axis_vec_kdef.h MegEngine: Initial commit of MegEngine. 5 years ago
indexing_one_hot.cpp MegEngine: Initial commit of MegEngine. 5 years ago
linspace.cpp MegEngine: Initial commit of MegEngine. 5 years ago
lrn.cpp MegEngine: Initial commit of MegEngine. 5 years ago
macro_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
macro_helper_epilogue.h MegEngine: Initial commit of MegEngine. 5 years ago
mask_conv.cpp MegEngine: Initial commit of MegEngine. 5 years ago
matrix_inverse.cpp MegEngine: Initial commit of MegEngine. 5 years ago
matrix_mul.cpp MegEngine: Initial commit of MegEngine. 5 years ago
max_tensor_diff.cpp MegEngine: Initial commit of MegEngine. 5 years ago
mesh_indexing.cpp MegEngine: Initial commit of MegEngine. 5 years ago
metahelper.h MegEngine: Initial commit of MegEngine. 5 years ago
opr_delegate.cpp MegEngine: Initial commit of MegEngine. 5 years ago
opr_delegate.h MegEngine: Initial commit of MegEngine. 5 years ago
param_pack.cpp refactor(dnn/parampack): remove parampacksplit 5 years ago
pooling.cpp feat(dnn): add nchw44 layout 5 years ago
powc.cpp MegEngine: Initial commit of MegEngine. 5 years ago
reduce.cpp fix(mgb/opr/reduce): fix grad function of reduce mean and add testcase 5 years ago
reduce_helper.cpp MegEngine: Initial commit of MegEngine. 5 years ago
reduce_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
relayout.cpp MegEngine: Initial commit of MegEngine. 5 years ago
relayout_format.cpp MegEngine: Initial commit of MegEngine. 5 years ago
relayout_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
resize.cpp MegEngine: Initial commit of MegEngine. 5 years ago
rng.cpp MegEngine: Initial commit of MegEngine. 5 years ago
roi_align.cpp MegEngine: Initial commit of MegEngine. 5 years ago
roi_align_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
roi_copy.cpp MegEngine: Initial commit of MegEngine. 5 years ago
roi_pooling.cpp MegEngine: Initial commit of MegEngine. 5 years ago
roi_pooling_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
rotate.cpp MegEngine: Initial commit of MegEngine. 5 years ago
rounding_converter.cuh MegEngine: Initial commit of MegEngine. 5 years ago
separableConv.cpp MegEngine: Initial commit of MegEngine. 5 years ago
separableFilter.cpp MegEngine: Initial commit of MegEngine. 5 years ago
small_vector.cpp MegEngine: Initial commit of MegEngine. 5 years ago
svd.cpp MegEngine: Initial commit of MegEngine. 5 years ago
tensor_format.cpp MegEngine: Initial commit of MegEngine. 5 years ago
tensor_iter.cpp MegEngine: Initial commit of MegEngine. 5 years ago
tensor_remap.cpp MegEngine: Initial commit of MegEngine. 5 years ago
tile_repeat.cpp MegEngine: Initial commit of MegEngine. 5 years ago
tile_repeat_helper.cpp MegEngine: Initial commit of MegEngine. 5 years ago
tile_repeat_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
topk.cpp MegEngine: Initial commit of MegEngine. 5 years ago
transpose.cpp MegEngine: Initial commit of MegEngine. 5 years ago
type_cvt.cpp MegEngine: Initial commit of MegEngine. 5 years ago
unroll_macro.h feat(dnn/x86): add avx2 int8 stride1 chanwise multithread conv 5 years ago
utils.cpp MegEngine: Initial commit of MegEngine. 5 years ago
utils.cuh MegEngine: Initial commit of MegEngine. 5 years ago
utils.h feat(dnn): add nchw44 q8 direct conv stride 1 5 years ago
version.cpp MegEngine: Initial commit of MegEngine. 5 years ago
version_symbol.h MegEngine: Initial commit of MegEngine. 5 years ago
warp_affine.cpp MegEngine: Initial commit of MegEngine. 5 years ago
warp_common.cpp MegEngine: Initial commit of MegEngine. 5 years ago
warp_common.h MegEngine: Initial commit of MegEngine. 5 years ago
warp_perspective.cpp MegEngine: Initial commit of MegEngine. 5 years ago
warp_perspective_helper.cpp MegEngine: Initial commit of MegEngine. 5 years ago
warp_perspective_helper.h MegEngine: Initial commit of MegEngine. 5 years ago
winograd_filter_preprocess.cpp MegEngine: Initial commit of MegEngine. 5 years ago

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

C++ Cuda Python C SVG other