MegEngine

History

Megvii Engine Team dbd9483993 feat(dnn,src,imperative): add groupnorm op GitOrigin-RevId: `de3c3d10e5`		2 years ago
..
argmxx	chore(license): remove all license in file header	3 years ago

cond_take	refactor(imperative/ops): extends DnnOprCaller with template	2 years ago

cv	fix(build): upgrade bazel riscv toolchains	3 years ago

elemwise	fix(mgb): add error infomation for old version load new elemwise mode	2 years ago

elemwise_multi_type	fix(mgb): add error infomation for old version load new elemwise mode	2 years ago

local	chore(license): remove all license in file header	3 years ago

local_share	chore(license): remove all license in file header	3 years ago

megcore	chore(license): remove all license in file header	3 years ago

pooling	chore(license): remove all license in file header	3 years ago

simd_macro	chore(license): remove all license in file header	3 years ago

winograd	chore(license): remove all license in file header	3 years ago

adaptive_pooling.cpp	feat(gopt): support nchw44 global pooling with fuse_grain	2 years ago

add_update.cpp	chore(license): remove all license in file header	3 years ago

add_update_helper.h	chore(license): remove all license in file header	3 years ago

algo_base.cpp	chore(license): remove all license in file header	3 years ago

algo_base.h	chore(license): remove all license in file header	3 years ago

algo_chooser.h	refactor(dnn): remove algorithm cache queries	3 years ago

algorithm_cache.cpp	chore(license): remove all license in file header	3 years ago

argmxx_helper.h	chore(license): remove all license in file header	3 years ago

argsort.cpp	chore(license): remove all license in file header	3 years ago

asm_common_defs.h	chore(license): remove all license in file header	3 years ago

basic_types.cpp	chore(license): remove all license in file header	3 years ago

batch_conv_bias.cpp	chore(license): remove all license in file header	3 years ago

batch_normalization.cpp	chore(license): remove all license in file header	3 years ago

batched_matrix_mul.cpp	fix(mgb): make error infomation of invalid MatMul more readable	2 years ago

check_non_finite.cpp	refactor(imperative/ops): extends DnnOprCaller with template	2 years ago

checksum.cpp	chore(license): remove all license in file header	3 years ago

concat_split.cpp	chore(license): remove all license in file header	3 years ago

conv_bias.cpp	feat(arm): add AlgoFP32Winograd F43, and add filter size into name of winograd-related algorithms	2 years ago

conv_bias.h	chore(license): remove all license in file header	3 years ago

conv_pooling.cpp	chore(license): remove all license in file header	3 years ago

convolution.cpp	Merge pull request #484 from wangxiang9603:add-nchw44-deconv	2 years ago

convolution3d.cpp	fix(mgb): make error infomation of group conv input channel mismatch more readable	2 years ago

correlation.cpp	chore(license): remove all license in file header	3 years ago

cpuinfo_arch_vendor.cpp	chore(license): remove all license in file header	3 years ago

cpuinfo_arch_vendor.h	chore(license): remove all license in file header	3 years ago

cumsum.cpp	chore(license): remove all license in file header	3 years ago

cvt_color.cpp	chore(license): remove all license in file header	3 years ago

dct.cpp	chore(license): remove all license in file header	3 years ago

deformable_conv.cpp	chore(license): remove all license in file header	3 years ago

deformable_ps_roi_pooling.cpp	chore(license): remove all license in file header	3 years ago

diag.cpp	chore(license): remove all license in file header	3 years ago

dot.cpp	chore(license): remove all license in file header	3 years ago

dropout.cpp	chore(license): remove all license in file header	3 years ago

dtype.cpp	chore(license): remove all license in file header	3 years ago

elemwise_helper.cpp	chore(license): remove all license in file header	3 years ago

elemwise_helper.cuh	chore(license): remove all license in file header	3 years ago

eye.cpp	chore(license): remove all license in file header	3 years ago

fake_quant.cpp	chore(license): remove all license in file header	3 years ago

fill.cpp	chore(license): remove all license in file header	3 years ago

flag_warn.cpp	chore(license): remove all license in file header	3 years ago

flip.cpp	chore(license): remove all license in file header	3 years ago

gaussian_blur.cpp	chore(license): remove all license in file header	3 years ago

gaussian_blur_helper.h	chore(license): remove all license in file header	3 years ago

group_local.cpp	chore(license): remove all license in file header	3 years ago

group_norm.cpp	feat(dnn,src,imperative): add groupnorm op	2 years ago

handle.cpp	chore(license): remove all license in file header	3 years ago

handle_impl.h	feat(dnn,src,imperative): add groupnorm op	2 years ago

hash_ct.h	style(all): reformat c++ code	3 years ago

images2neibs.cpp	chore(license): remove all license in file header	3 years ago

indexing_multi_axis_vec.cpp	chore(license): remove all license in file header	3 years ago

indexing_multi_axis_vec_kdef.h	chore(license): remove all license in file header	3 years ago

indexing_one_hot.cpp	chore(license): remove all license in file header	3 years ago

lamb.cpp	refactor(imperative/ops): extends DnnOprCaller with template	2 years ago

layer_norm.cpp	refactor(ops/layer_norm): speed up the host speed of layer_norm	3 years ago

linspace.cpp	chore(license): remove all license in file header	3 years ago

lrn.cpp	chore(license): remove all license in file header	3 years ago

lsq.cpp	chore(license): remove all license in file header	3 years ago

lstm.cpp	chore(license): remove all license in file header	3 years ago

lstm_cell.cpp	chore(license): remove all license in file header	3 years ago

lstm_cell.h	chore(license): remove all license in file header	3 years ago

macro_helper.h	chore(license): remove all license in file header	3 years ago

macro_helper_epilogue.h	chore(license): remove all license in file header	3 years ago

mask_conv.cpp	chore(license): remove all license in file header	3 years ago

matrix_inverse.cpp	chore(license): remove all license in file header	3 years ago

matrix_mul.cpp	fix(mgb): make error infomation of invalid MatMul more readable	2 years ago

max_tensor_diff.cpp	chore(license): remove all license in file header	3 years ago

mesh_indexing.cpp	chore(license): remove all license in file header	3 years ago

metahelper.h	chore(license): remove all license in file header	3 years ago

named_tensor.cpp	chore(license): remove all license in file header	3 years ago

nchw_nchwxx_valid.cpp	chore(license): remove all license in file header	3 years ago

nchw_nchwxx_valid.h	chore(license): remove all license in file header	3 years ago

norm.cpp	feat(dnn/naive/norm,-dnn/cuda/norm,-dnn/test/norm): add norm dnn opr,	3 years ago

opr_delegate.cpp	chore(license): remove all license in file header	3 years ago

opr_delegate.h	chore(license): remove all license in file header	3 years ago

opr_trait.h	feat(dnn,src,imperative): add groupnorm op	2 years ago

padding.cpp	perf(imperative/src): improve pad host performance	3 years ago

param_pack.cpp	chore(license): remove all license in file header	3 years ago

pooling.cpp	feat(gopt): support nchw44 global pooling with fuse_grain	2 years ago

postprocess.h	chore(license): remove all license in file header	3 years ago

postprocess_helper.h	feat(rvv/fallback): make nchw44 happly on rvv	3 years ago

powc.cpp	chore(license): remove all license in file header	3 years ago

reduce.cpp	chore(license): remove all license in file header	3 years ago

reduce_helper.cpp	chore(license): remove all license in file header	3 years ago

reduce_helper.h	chore(license): remove all license in file header	3 years ago

reduce_helper_device.h	style(imperative/amp): reformat code	3 years ago

region_restricted_convolution.cpp	feat(dnn): add RegionRestrictedConv DGRAD support int32 and uint8	2 years ago

relayout.cpp	chore(license): remove all license in file header	3 years ago

relayout_format.cpp	fix(dnn): fix dnn run cd4 on cpu	3 years ago

relayout_helper.h	chore(license): remove all license in file header	3 years ago

remap.cpp	chore(license): remove all license in file header	3 years ago

resize.cpp	fix(mgb/dnn): suport fp16 for resize nhwc	3 years ago

resize.cuh	chore(license): remove all license in file header	3 years ago

rng.cpp	chore(license): remove all license in file header	3 years ago

rnn.cpp	chore(license): remove all license in file header	3 years ago

rnn.h	chore(license): remove all license in file header	3 years ago

rnn_cell.cpp	chore(license): remove all license in file header	3 years ago

rnn_cell.h	chore(license): remove all license in file header	3 years ago

roi_align.cpp	feat(imperative): support empty tensor in roi_align	3 years ago

roi_align_helper.h	chore(license): remove all license in file header	3 years ago

roi_copy.cpp	chore(license): remove all license in file header	3 years ago

roi_pooling.cpp	chore(license): remove all license in file header	3 years ago

roi_pooling_helper.h	chore(license): remove all license in file header	3 years ago

rotate.cpp	chore(license): remove all license in file header	3 years ago

rounding_converter.cuh	chore(license): remove all license in file header	3 years ago

separableConv.cpp	chore(license): remove all license in file header	3 years ago

separableFilter.cpp	chore(license): remove all license in file header	3 years ago

sliding_window_transpose.cpp	chore(license): remove all license in file header	3 years ago

small_vector.cpp	chore(license): remove all license in file header	3 years ago

softmax.cpp	chore(license): remove all license in file header	3 years ago

svd.cpp	chore(license): remove all license in file header	3 years ago

tensor_format.cpp	feat(dnn/opencl): optimize heuristic rule	3 years ago

tensor_iter.cpp	chore(license): remove all license in file header	3 years ago

tensor_remap.cpp	chore(license): remove all license in file header	3 years ago

tile_repeat.cpp	chore(license): remove all license in file header	3 years ago

tile_repeat_helper.cpp	chore(license): remove all license in file header	3 years ago

tile_repeat_helper.h	chore(license): remove all license in file header	3 years ago

topk.cpp	chore(license): remove all license in file header	3 years ago

tqt.cpp	chore(license): remove all license in file header	3 years ago

transpose.cpp	chore(license): remove all license in file header	3 years ago

type_cvt.cpp	chore(license): remove all license in file header	3 years ago

unroll_macro.h	feat(arm): add winograd F43 NCHW44 algo and winograd F43 44 algo	2 years ago

utils.cpp	chore(license): remove all license in file header	3 years ago

utils.cuh	chore(license): remove all license in file header	3 years ago

utils.h	chore(license): remove all license in file header	3 years ago

version.cpp	chore(license): remove all license in file header	3 years ago

version_symbol.h	chore(license): remove all license in file header	3 years ago

warp_affine.cpp	chore(license): remove all license in file header	3 years ago

warp_common.cpp	feat(dnn): warpperspective support multi src input	2 years ago

warp_common.h	feat(dnn): warpperspective support multi src input	2 years ago

warp_perspective.cpp	feat(mgb): warpperspective support multi src input	2 years ago

warp_perspective_helper.cpp	chore(license): remove all license in file header	3 years ago

warp_perspective_helper.h	chore(license): remove all license in file header	3 years ago

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

模型部署模型训练

C++ Cuda Python C SVG other

megengine@megvii.com huangxinda@megvii.com luzzyzhang@gmail.com wenjuan@megvii.com 1041563782@qq.com xxr@megvii.com 61608041+megvii-mge@users.noreply.github.com 1186454801@qq.com 35462954+XindaH@users.noreply.github.com acdoge.cao@gmail.com 75015964+HuaHua404@users.noreply.github.com 1271808136@qq.com 70926289+kagome1007@users.noreply.github.com chenjiahui@megvii.com liuke@megvii.com quqi.liu@shopee.com khj.application@aliyun.com zhanghaolong@megvii.com 974658390@qq.com 2283984853@qq.com lijiansong@ict.ac.cn loves0510@outlook.com p2oileen@whu.edu.cn 1157539822@qq.com mediosrity@gmail.com

Contributors (25+) All

Contributors (25+)
All