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 be205727bc fix(mge): fix some warnings 5 years ago
..
argmxx MegEngine: Initial commit of MegEngine. 5 years ago
cond_take MegEngine: Initial commit of MegEngine. 5 years ago
cv feat(cmake/cross_build/host_build/windows): imp windows 5 years ago
elemwise fix(opr/basic_arith): make Elemwise really support empty IO 5 years ago
elemwise_multi_type MegEngine: Initial commit of MegEngine. 5 years ago
local refactor(dnn/cuda): invoke local opr in group local 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 feat(dnn/arm_common): add int8 nchw44 winograd f23_4x4 f23_8x8 compute float32/int16 output int8 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 fix(mge): fix some warnings 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/arm_common): add int8 nchw44 winograd f23_4x4 f23_8x8 compute float32/int16 output int8 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 ci(copybara): fix copybara of arm 5 years ago
handle_impl.h feat(mge/opr): add opr remap in opencl and naive 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 feat(dnn/common): add matmul impl for naive with matrix format mk4_dot 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 ci(copybara): fix copybara of arm 5 years ago
remap.cpp feat(mge/opr): add opr remap in opencl and naive 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 feat(mge): add bfloat16 support 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 feat(megbrain/dnn): add indexing remap int32 for naive and cuda 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/arm): add arm nchw44 fp32 pooling 5 years ago
utils.cpp fix(dnn): fix data type check for quantized convolution 5 years ago
utils.cuh MegEngine: Initial commit of MegEngine. 5 years ago
utils.h feat(mge): add bfloat16 support 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 ci(copybara): fix copybara of arm 5 years ago
warp_perspective.cpp feat(mge): add bfloat16 support 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 feat(dnn/arm_common): add int8 nchw44 winograd f23_4x4 f23_8x8 compute float32/int16 output int8 5 years ago

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

C++ Cuda Python C SVG other