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 f5833a5294 fix(dnn/cuda): fix cublas matmul on sm60 5 years ago
..
local MegEngine: Initial commit of MegEngine. 5 years ago
megcore MegEngine: Initial commit of MegEngine. 5 years ago
add_update.cpp MegEngine: Initial commit of MegEngine. 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 MegEngine: Initial commit of MegEngine. 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): fix cublas matmul on sm60 5 years ago
conv_bias_int8.cpp MegEngine: Initial commit of MegEngine. 5 years ago
conv_pooling.cpp MegEngine: Initial commit of MegEngine. 5 years ago
convolution.cpp fix(dnn/cuda): fix cublas matmul on sm60 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 MegEngine: Initial commit of MegEngine. 5 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 fix(dnn/cuda): fix cublas matmul on sm60 5 years ago
mesh_indexing.cpp MegEngine: Initial commit of MegEngine. 5 years ago
param_pack.cpp MegEngine: Initial commit of MegEngine. 5 years ago
pooling.cpp MegEngine: Initial commit of MegEngine. 5 years ago
powc.cpp MegEngine: Initial commit of MegEngine. 5 years ago
reduce.cpp 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
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 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
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 MegEngine: Initial commit of MegEngine. 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 MegEngine: Initial commit of MegEngine. 5 years ago
utils.cpp MegEngine: Initial commit of MegEngine. 5 years ago
utils.h MegEngine: Initial commit of MegEngine. 5 years ago
warp_affine.cpp MegEngine: Initial commit of MegEngine. 5 years ago
warp_perspective.cpp MegEngine: Initial commit of MegEngine. 5 years ago

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

C++ Cuda Python C SVG other