OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
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.
2699 Commits
28 Branches
25 MB
3465 Download
 
 
 
 
 
 
Tree: c49d3070ba
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'c49d3070ba'
${ noResults }
MegEngine/dnn/test/cuda
Megvii Engine Team c49d3070ba refactor(imperative/ops): extends DnnOprCaller with template 3 years ago
..
local chore(license): remove all license in file header 3 years ago
megcore chore(license): remove all license in file header 3 years ago
accuracy_shake.cpp chore(license): remove all license in file header 3 years ago
adaptive_pooling.cpp chore(license): remove all license in file header 3 years ago
add_update.cpp chore(license): remove all license in file header 3 years ago
argmxx.cpp chore(license): remove all license in file header 3 years ago
argsort.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
batched_matrix_mul.cpp chore(license): remove all license in file header 3 years ago
benchmark.cpp chore(license): remove all license in file header 3 years ago
benchmark.h chore(license): remove all license in file header 3 years ago
bn.cpp chore(license): remove all license in file header 3 years ago
chanwise_convolution.cpp chore(license): remove all license in file header 3 years ago
chanwise_convolution3d.cpp chore(license): remove all license in file header 3 years ago
check_non_finite.cpp chore(license): remove all license in file header 3 years ago
checksum.cpp chore(license): remove all license in file header 3 years ago
concat.cpp chore(license): remove all license in file header 3 years ago
cond_take.cpp chore(license): remove all license in file header 3 years ago
conv_bias.cpp chore(license): remove all license in file header 3 years ago
conv_bias_int8.cpp 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
conv_test_utils.cpp chore(license): remove all license in file header 3 years ago
conv_test_utils.h chore(license): remove all license in file header 3 years ago
conv_v8.cpp refactor(dnn): make cudnn_frontend work 3 years ago
convolution.cpp chore(license): remove all license in file header 3 years ago
convolution3d.cpp chore(license): remove all license in file header 3 years ago
correlation.cpp chore(license): remove all license in file header 3 years ago
cumsum.cpp chore(license): remove all license in file header 3 years ago
cutlass_matmul.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
dilated_convolution.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
elemwise.cpp chore(license): remove all license in file header 3 years ago
elemwise_multi_type.cpp feat(mgb/dnn): add modes that the output type is bool in elemwise 2 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
fixture.cpp chore(license): remove all license in file header 3 years ago
fixture.h 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
group_conv.cpp chore(license): remove all license in file header 3 years ago
group_conv3d.cpp 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
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_one_hot.cpp chore(license): remove all license in file header 3 years ago
lamb.cpp feat(mge/dnn): add lamb optimizer 3 years ago
layer_norm.cpp chore(license): remove all license in file header 3 years ago
linspace.cpp chore(license): remove all license in file header 3 years ago
local.cpp chore(license): remove all license in file header 3 years ago
local_share.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
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 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
norm.cpp feat(dnn/naive/norm,-dnn/cuda/norm,-dnn/test/norm): add norm dnn opr, 3 years ago
padding.cpp chore(license): remove all license in file header 3 years ago
param_pack.cpp refactor(imperative/ops): extends DnnOprCaller with template 2 years ago
pooling.cpp chore(license): remove all license in file header 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
relayout.cpp chore(license): remove all license in file header 3 years ago
relayout_format.cpp chore(license): remove all license in file header 3 years ago
remap.cpp chore(license): remove all license in file header 3 years ago
repeat.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
rng.cpp chore(license): remove all license in file header 3 years ago
roi_align.cpp 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
rotate.cpp chore(license): remove all license in file header 3 years ago
sleep.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
softmax.cpp chore(license): remove all license in file header 3 years ago
split.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_remap.cpp chore(license): remove all license in file header 3 years ago
tile.cpp chore(license): remove all license in file header 3 years ago
timer.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
utils.cpp chore(license): remove all license in file header 3 years ago
utils.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_perspective.cpp chore(license): remove all license in file header 3 years ago
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

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

C++ Cuda Python C SVG other