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 fc1ce273b7 fix(dnn/cuda): fix elemwise add cuda int8 bcast 4 years ago
..
local MegEngine: Initial commit of MegEngine. 5 years ago
megcore MegEngine: Initial commit of MegEngine. 5 years ago
add_update.cpp feat(mge): add bfloat16 support 5 years ago
argmxx.cpp MegEngine: Initial commit of MegEngine. 5 years ago
argsort.cpp MegEngine: Initial commit of MegEngine. 5 years ago
batch_conv_bias.cpp MegEngine: Initial commit of MegEngine. 5 years ago
batched_matrix_mul.cpp MegEngine: Initial commit of MegEngine. 5 years ago
benchmark.cpp fix(dnn/cuda): fix cublas matmul on sm60 5 years ago
benchmark.h MegEngine: Initial commit of MegEngine. 5 years ago
bn.cpp MegEngine: Initial commit of MegEngine. 5 years ago
chanwise_convolution.cpp feat(dnn/naive): support weight preprocess interface in dnn 5 years ago
chanwise_convolution3d.cpp MegEngine: Initial commit of MegEngine. 5 years ago
checksum.cpp MegEngine: Initial commit of MegEngine. 5 years ago
concat.cpp MegEngine: Initial commit of MegEngine. 5 years ago
cond_take.cpp MegEngine: Initial commit of MegEngine. 5 years ago
conv_bias.cpp fix(dnn/cuda): enable cuda algos for nchw quantized 5 years ago
conv_bias_int8.cpp feat(dnn/cuda): add conv2d-sass-kernel 5 years ago
conv_pooling.cpp MegEngine: Initial commit of MegEngine. 5 years ago
convolution.cpp fix(ci/megdnn_test/megbrain_test): split some 5 years ago
convolution3d.cpp fix(dnn/cuda): fix cublas matmul on sm60 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
dilated_convolution.cpp MegEngine: Initial commit of MegEngine. 5 years ago
dot.cpp MegEngine: Initial commit of MegEngine. 5 years ago
elemwise.cpp fix(dnn/cuda): fix elemwise add cuda int8 bcast 4 years ago
elemwise_multi_type.cpp MegEngine: Initial commit of MegEngine. 5 years ago
eye.cpp MegEngine: Initial commit of MegEngine. 5 years ago
fixture.cpp MegEngine: Initial commit of MegEngine. 5 years ago
fixture.h 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
group_conv.cpp fix(dnn/cuda): fix cublas matmul on sm60 5 years ago
group_conv3d.cpp fix(dnn/cuda): fix cublas matmul on sm60 5 years ago
group_local.cpp MegEngine: Initial commit of MegEngine. 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_one_hot.cpp MegEngine: Initial commit of MegEngine. 5 years ago
linspace.cpp MegEngine: Initial commit of MegEngine. 5 years ago
local.cpp MegEngine: Initial commit of MegEngine. 5 years ago
local_share.cpp MegEngine: Initial commit of MegEngine. 5 years ago
lrn.cpp 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(mge): add bfloat16 support 5 years ago
mesh_indexing.cpp MegEngine: Initial commit of MegEngine. 5 years ago
param_pack.cpp fix(dnn/test): fix dnn assert and format code 5 years ago
pooling.cpp feat(dnn/cuda): add conv2d-sass-kernel 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
relayout.cpp MegEngine: Initial commit of MegEngine. 5 years ago
relayout_format.cpp feat(dnn): add relayout_format for nchw to nchw4 and ic <=4 5 years ago
remap.cpp feat(dnn/cuda): add cuda remap 5 years ago
repeat.cpp 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 feat(mge): add bfloat16 support 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
rotate.cpp MegEngine: Initial commit of MegEngine. 5 years ago
sleep.cpp MegEngine: Initial commit of MegEngine. 5 years ago
split.cpp MegEngine: Initial commit of MegEngine. 5 years ago
svd.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.cpp MegEngine: Initial commit of MegEngine. 5 years ago
timer.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 feat(mge): add bfloat16 support 5 years ago
utils.cpp feat(dnn/cuda): add conv2d-sass-kernel 5 years ago
utils.h feat(dnn/cuda): add conv2d-sass-kernel 5 years ago
warp_affine.cpp MegEngine: Initial commit of MegEngine. 5 years ago
warp_perspective.cpp fix(dnn/cuda): fix INTMAX overflow in warp_perspective_cuda 4 years ago

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

C++ Cuda Python C SVG other