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 95ac055538 feat(dnn,mgb,imperative): add diag opr implement 4 years ago
..
adaptive_pooling style(all): reformat c++ code 4 years ago
add_update feat(dnn): use ref ptr interface for all backends 4 years ago
argmxx fix(dnn): correct reduce/argmxx/fakequant calculation with nan 4 years ago
argsort feat(dnn): use ref ptr interface for all backends 4 years ago
batch_conv_bias style(all): reformat c++ code 4 years ago
batch_normalization feat(dnn): use ref ptr interface for all backends 4 years ago
batched_matrix_mul feat(dnn): use ref ptr interface for all backends 4 years ago
check_non_finite feat(dnn/check_non_finite): addmul scale to check_non_finite opr 4 years ago
checksum feat(dnn): use ref ptr interface for all backends 4 years ago
concat feat(dnn): use ref ptr interface for all backends 4 years ago
cond_take feat(dnn): use ref ptr interface for all backends 4 years ago
conv_bias feat(dnn): use ref ptr interface for all backends 4 years ago
convolution feat(dnn): use ref ptr interface for all backends 4 years ago
convolution3d feat(dnn): use ref ptr interface for all backends 4 years ago
convpooling feat(dnn): use ref ptr interface for all backends 4 years ago
correlation style(all): reformat c++ code 4 years ago
cumsum feat(dnn): use ref ptr interface for all backends 4 years ago
cvt_color style(all): reformat c++ code 4 years ago
dct feat(dnn): use ref ptr interface for all backends 4 years ago
deformable_conv feat(dnn): use ref ptr interface for all backends 4 years ago
deformable_ps_roi_pooling style(all): reformat c++ code 4 years ago
diag feat(dnn,mgb,imperative): add diag opr implement 4 years ago
dot style(all): reformat c++ code 4 years ago
dropout feat(mgb/opr): add dropout kernel 4 years ago
elemwise feat(dnn): use ref ptr interface for all backends 4 years ago
elemwise_multi_type fix(build): split elemwise_multi_type cpp 4 years ago
eye feat(dnn): enable eye to support bool 4 years ago
fake_quant style(all): reformat c++ code 4 years ago
fill feat(dnn): use ref ptr interface for all backends 4 years ago
flip style(all): reformat c++ code 4 years ago
gaussian_blur style(all): reformat c++ code 4 years ago
group_local feat(dnn): use ref ptr interface for all backends 4 years ago
images2neibs feat(dnn/src): add images2neibs kernel of opencl and related test 4 years ago
indexing_multi_axis_vec feat(opr): indexing_multi_axis_vec support nd index 4 years ago
indexing_one_hot style(all): reformat c++ code 4 years ago
layer_norm feat(mgb/opr): add layernorm forward and backward kernel 4 years ago
linspace feat(dnn): use ref ptr interface for all backends 4 years ago
local feat(dnn): use ref ptr interface for all backends 4 years ago
local_share style(all): reformat c++ code 4 years ago
lrn style(all): reformat c++ code 4 years ago
lsq style(all): reformat c++ code 4 years ago
lstm fix(dnn): add naive lstm kernel 4 years ago
lstm_cell fix(dnn): add naive lstm kernel 4 years ago
mask_conv style(all): reformat c++ code 4 years ago
matrix_inverse feat(dnn): use ref ptr interface for all backends 4 years ago
matrix_mul feat(dnn): use ref ptr interface for all backends 4 years ago
max_tensor_diff style(all): reformat c++ code 4 years ago
mesh_indexing feat(dnn): use ref ptr interface for all backends 4 years ago
padding feat(dnn): use ref ptr interface for all backends 4 years ago
param_pack feat(dnn): use ref ptr interface for all backends 4 years ago
pooling fix(dnn): fix naive average pooling overflow bug for int8 type 4 years ago
powc style(all): reformat c++ code 4 years ago
reduce feat(dnn): use ref ptr interface for all backends 4 years ago
relayout style(all): reformat c++ code 4 years ago
relayout_format feat(dnn): use ref ptr interface for all backends 4 years ago
remap feat(dnn): add float16 for remap backward 4 years ago
repeat feat(dnn): use ref ptr interface for all backends 4 years ago
resize feat(dnn): use ref ptr interface for all backends 4 years ago
rng feat(dnn): use ref ptr interface for all backends 4 years ago
rnn fix(dnn): add naive lstm kernel 4 years ago
rnn_cell feat(arm): add arm rnn_cell/lstm_cell/lstm optimized kernel 4 years ago
roi_align style(all): reformat c++ code 4 years ago
roi_copy style(all): reformat c++ code 4 years ago
roi_pooling style(all): reformat c++ code 4 years ago
rotate feat(dnn): use ref ptr interface for all backends 4 years ago
separable_conv feat(dnn): use ref ptr interface for all backends 4 years ago
separable_filter feat(dnn): use ref ptr interface for all backends 4 years ago
sleep style(all): reformat c++ code 4 years ago
sliding_window_transpose style(all): reformat c++ code 4 years ago
softmax feat(mge): add softmax op use cudnn api 4 years ago
split feat(dnn): use ref ptr interface for all backends 4 years ago
svd feat(dnn): use ref ptr interface for all backends 4 years ago
tensor_remap style(all): reformat c++ code 4 years ago
tile feat(dnn): use ref ptr interface for all backends 4 years ago
topk feat(dnn): use ref ptr interface for all backends 4 years ago
tqt style(all): reformat c++ code 4 years ago
transpose feat(dnn): use ref ptr interface for all backends 4 years ago
type_cvt feat(dnn/arm): support layout like NHWC channel like broadcast on arm 4 years ago
warp_affine feat(dnn): use ref ptr interface for all backends 4 years ago
warp_perspective feat(dnn): use ref ptr interface for all backends 4 years ago
handle.cpp feat(dnn,mgb,imperative): add diag opr implement 4 years ago
handle.h style(all): reformat c++ code 4 years ago
lowbit_utils.cpp feat(dnn): use ref ptr interface for all backends 4 years ago
lowbit_utils.h style(all): reformat c++ code 4 years ago

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

C++ Cuda Python C SVG other