MegEngine

History

Megvii Engine Team 42c3f441ce feat(dnn/cambricon): add MaskedFill opr on cambricon GitOrigin-RevId: `e9e4f72415`		9 months ago
..
argmxx	chore(license): remove all license in file header	2 years ago

cond_take	refactor(imperative/ops): extends DnnOprCaller with template	1 year ago

cv	feat(megbrain): support huawei OpenHarmony NDK	7 months ago

elemwise	fix(rocm): fix rocm elemwise uint16 compile	7 months ago

elemwise_multi_type	feat(opencl/bazel): support bazel build with ocl only	11 months ago

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

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

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

multi_head_attn	feat(opr): add multiattention naive proxy backend	11 months ago

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

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

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

adaptive_pooling.cpp	feat(gopt): support nchw44 global pooling with fuse_grain	1 year ago

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

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

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

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

algo_chooser.h	refactor(dnn): remove algorithm cache queries	1 year ago

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

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

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

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

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

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

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

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

check_non_finite.cpp	refactor(imperative/ops): extends DnnOprCaller with template	1 year ago

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

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

conv_bias.cpp	feat(dnn/fallback): add gi fp16 nchw88 winograd f43 algo	1 year ago

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

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

convolution.cpp	feat(oprs): support zero shape	1 year ago

convolution3d.cpp	feat(oprs): support zero shape	1 year ago

correlation.cpp	feat(oprs): support zero shape	1 year ago

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

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

cross.cpp	feat(dnn,megbrain,imperative): add cross opr	9 months ago

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

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

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

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

deformable_ps_roi_pooling.cpp	feat(oprs): support zero shape	1 year ago

diag.cpp	feat(oprs): support zero shape	1 year ago

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

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

dtype.cpp	fix(dnn/dtype): fix race condition in ParameterizedDType registry	1 year ago

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

elemwise_helper.cuh	feat(dnn/elemwise): support elemwise uint16 compute	8 months ago

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

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

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

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

flip.cpp	feat(opr): add flip, rotate, resize op to imperative	8 months ago

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

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

general_norm.cpp	fix(imperative): fix the parameter acquisition problem of generalnorm	1 year ago

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

group_norm.cpp	feat(dnn,src,imperative): add instancenorm	1 year ago

handle.cpp	feat(opencl/bazel): support bazel build with ocl only	11 months ago

handle_impl.h	feat(dnn,megbrain,imperative): add multinomial opr	6 months ago

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

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

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

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

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

lamb.cpp	refactor(imperative/ops): extends DnnOprCaller with template	1 year ago

layer_norm.cpp	refactor(ops/layer_norm): speed up the host speed of layer_norm	1 year ago

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

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

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

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

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

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

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

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

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

masked_fill.cpp	feat(opr): add masked_fill op	1 year ago

matrix_inverse.cpp	feat(oprs): support zero shape	1 year ago

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

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

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

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

multi_head_attn.cpp	feat(opr): add multiattention cuda proxy backend	11 months ago

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

nchw_nchwxx_valid.cpp	feat(dnn): add fp16 hybrid direct conv algo	1 year ago

nchw_nchwxx_valid.h	feat(opencl/bazel): support bazel build with ocl only	11 months ago

non_zero.cpp	feat(mge): add kernel for where operator	8 months ago

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

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

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

opr_trait.h	feat(dnn/cambricon): add MaskedFill opr on cambricon	5 months ago

padding.cpp	perf(imperative/src): improve pad host performance	1 year ago

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

pooling.cpp	feat(oprs): support zero shape	1 year ago

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

postprocess_helper.h	feat(rvv/fallback): make nchw44 happly on rvv	1 year ago

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

reduce.cpp	feat(oprs): support zero shape	1 year ago

reduce_helper.cpp	feat(opr): optimize general_norm cuda forward and backward implementation	1 year ago

reduce_helper.h	feat(opr): optimize general_norm cuda forward and backward implementation	1 year ago

reduce_helper_device.h	style(imperative/amp): reformat code	1 year ago

region_restricted_convolution.cpp	feat(oprs): support zero shape	1 year ago

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

relayout_format.cpp	fix(opencl): support OpenCL NHWC_NHWCD4I channel	1 year ago

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

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

resize.cpp	feat(opr): add interpolate trilinear	10 months ago

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

rng.cpp	feat(dnn,megbrain,imperative): add multinomial opr	6 months ago

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

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

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

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

roi_align.cpp	feat(imperative): support empty tensor in roi_align	1 year ago

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

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

roi_pooling.cpp	feat(oprs): support zero shape	1 year ago

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

rotate.cpp	feat(opr): add flip, rotate, resize op to imperative	8 months ago

rounding_converter.cuh	fix(dnn): correctly using MEGDNN_DISABLE_FLOAT16 directives	1 year ago

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

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

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

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

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

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

tensor_format.cpp	feat(dnn/opencl): optimize heuristic rule	1 year ago

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

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

tile_repeat.cpp	fix(fallback/tile): fix fallback tile opr workspace size	6 months ago

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

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

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

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

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

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

unroll_macro.h	feat(dnn): add fp16 hybrid direct conv algo	1 year ago

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

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

utils.h	feat(oprs): support zero shape	1 year ago

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

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

warp_affine.cpp	feat(oprs): support zero shape	1 year ago

warp_common.cpp	feat(dnn): warpperspective support multi src input	1 year ago

warp_common.h	fix(midout): add more midout tag	1 year ago

warp_perspective.cpp	feat(oprs): support zero shape	1 year ago

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

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

where.cpp	feat(mge): add kernel for where operator	8 months ago

天元（MegEngine）是旷视自主研发的开源深度学习框架，于2020年3月正式在 OpenI 启智社区开源，天元能够帮助开发者高效的完成深度学习算法的设计、训练、部署，有效提升AI研发工作效率。

https://megengine.org.cn/

ai开发工具 openi deep-learning

C++ Cuda Python C Starlark 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

1271808136@qq.com 35462954+XindaH@users.noreply.github.com 70926289+kagome1007@users.noreply.github.com zhanghaolong@megvii.com acdoge.cao@gmail.com 75015964+HuaHua404@users.noreply.github.com khj.application@aliyun.com 17733781361@163.com chenjiahui@megvii.com liuke@megvii.com quqi.liu@shopee.com 974658390@qq.com 2283984853@qq.com lijiansong@ict.ac.cn loves0510@outlook.com p2oileen@whu.edu.cn 1157539822@qq.com

How to access data resources in code

Contributors (25+) All

Contributors (25+)
All