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.
2205 Commits
28 Branches
25 MB
3465 Download
 
 
 
 
 
 
Tree: 1999307015
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '1999307015'
${ noResults }
MegEngine/dnn/src/cuda
Megvii Engine Team 1999307015 feat(mgb/opr): add dropout kernel 3 years ago
..
adaptive_pooling style(all): reformat c++ code 3 years ago
add_update style(all): reformat c++ code 3 years ago
argmxx feat(dnn): use ref ptr interface for all backends 3 years ago
argsort style(all): reformat c++ code 3 years ago
batch_conv_bias style(all): reformat c++ code 3 years ago
batch_normalization feat(dnn): use ref ptr interface for all backends 3 years ago
batched_matrix_mul feat(dnn): use ref ptr interface for all backends 3 years ago
check_non_finite feat(dnn/check_non_finite): addmul scale to check_non_finite opr 3 years ago
checksum feat(dnn): use ref ptr interface for all backends 3 years ago
concat style(all): reformat c++ code 3 years ago
cond_take style(all): reformat c++ code 3 years ago
conv_bias fix(mgb/cuda): fix conv error when the input tensor is too large 3 years ago
convolution feat(dnn): use ref ptr interface for all backends 3 years ago
convolution3d feat(dnn): use ref ptr interface for all backends 3 years ago
convolution_helper style(all): reformat c++ code 3 years ago
convpooling feat(dnn): use ref ptr interface for all backends 3 years ago
correlation style(all): reformat c++ code 3 years ago
cub ci(check): add clang-format in check stage 3 years ago
cumsum feat(dnn): use ref ptr interface for all backends 3 years ago
cutlass style(all): reformat c++ code 3 years ago
cv style(all): reformat c++ code 3 years ago
cvt_color style(all): reformat c++ code 3 years ago
dct feat(dnn): use ref ptr interface for all backends 3 years ago
deformable_conv style(all): reformat c++ code 3 years ago
deformable_ps_roi_pooling style(all): reformat c++ code 3 years ago
dot style(all): reformat c++ code 3 years ago
dropout feat(mgb/opr): add dropout kernel 3 years ago
elemwise style(all): reformat c++ code 3 years ago
elemwise_multi_type feat(dnn): use ref ptr interface for all backends 3 years ago
eye feat(dnn): enable eye to support bool 3 years ago
fake_quant fix(dnn): correct reduce/argmxx/fakequant calculation with nan 3 years ago
fill style(all): reformat c++ code 3 years ago
flip style(all): reformat c++ code 3 years ago
gaussian_blur style(all): reformat c++ code 3 years ago
group_local feat(dnn): use ref ptr interface for all backends 3 years ago
images2neibs style(all): reformat c++ code 3 years ago
indexing_multi_axis_vec feat(opr): indexing_multi_axis_vec support nd index 3 years ago
indexing_one_hot style(all): reformat c++ code 3 years ago
layer_norm feat(mgb/opr): add layernorm forward and backward kernel 3 years ago
linspace style(all): reformat c++ code 3 years ago
local style(all): reformat c++ code 3 years ago
local_share feat(dnn): use ref ptr interface for all backends 3 years ago
lrn feat(dnn): use ref ptr interface for all backends 3 years ago
lsq style(all): reformat c++ code 3 years ago
mask_conv style(all): reformat c++ code 3 years ago
matrix_inverse feat(dnn): use ref ptr interface for all backends 3 years ago
matrix_mul feat(dnn): use ref ptr interface for all backends 3 years ago
max_tensor_diff style(all): reformat c++ code 3 years ago
megcore style(all): reformat c++ code 3 years ago
mesh_indexing style(all): reformat c++ code 3 years ago
padding feat(dnn): use ref ptr interface for all backends 3 years ago
param_pack feat(dnn): use ref ptr interface for all backends 3 years ago
pooling feat(dnn): use ref ptr interface for all backends 3 years ago
powc style(all): reformat c++ code 3 years ago
reduce feat(dnn): use ref ptr interface for all backends 3 years ago
reduce_helper chore(mge): update copyright years 4 years ago
relayout feat(dnn): use ref ptr interface for all backends 3 years ago
relayout_format feat(dnn): use ref ptr interface for all backends 3 years ago
remap feat(dnn): add float16 for remap backward 3 years ago
repeat feat(dnn): use ref ptr interface for all backends 3 years ago
resize style(all): reformat c++ code 3 years ago
rng style(all): reformat c++ code 3 years ago
roi_align feat(dnn): use ref ptr interface for all backends 3 years ago
roi_copy feat(dnn): use ref ptr interface for all backends 3 years ago
roi_pooling style(all): reformat c++ code 3 years ago
rotate style(all): reformat c++ code 3 years ago
separable_conv style(all): reformat c++ code 3 years ago
separable_filter style(all): reformat c++ code 3 years ago
sleep style(all): reformat c++ code 3 years ago
sliding_window_transpose style(all): reformat c++ code 3 years ago
split style(all): reformat c++ code 3 years ago
svd feat(dnn): use ref ptr interface for all backends 3 years ago
tensor_remap style(all): reformat c++ code 3 years ago
tile feat(dnn): use ref ptr interface for all backends 3 years ago
topk style(all): reformat c++ code 3 years ago
tqt style(all): reformat c++ code 3 years ago
transpose style(all): reformat c++ code 3 years ago
type_cvt feat(dnn): use ref ptr interface for all backends 3 years ago
warp_affine style(all): reformat c++ code 3 years ago
warp_perspective feat(dnn): use ref ptr interface for all backends 3 years ago
atomic_add.cuh style(all): reformat c++ code 3 years ago
cuda_shfl_compat.cuh style(all): reformat c++ code 3 years ago
cudnn_with_check.h chore(mge): update copyright years 4 years ago
cudnn_wrapper.cpp fix(mgb/dnn): fix cudnnConvBiasActivation crash on nchw32 int8 with oc > 256 3 years ago
cudnn_wrapper.h style(all): reformat c++ code 3 years ago
elemwise_helper.cpp feat(dnn): use ref ptr interface for all backends 3 years ago
elemwise_helper.cuh style(all): reformat c++ code 3 years ago
elemwise_helper_q4.cuh style(all): reformat c++ code 3 years ago
error_info.cuh style(all): reformat c++ code 3 years ago
fp16_help.cuh style(all): reformat c++ code 3 years ago
handle.cpp fix(mgb/src): add tensorRT version check 3 years ago
handle.h style(all): reformat c++ code 3 years ago
handle_create.cpp feat(mgb/opr): add dropout kernel 3 years ago
int_fastdiv.cpp style(all): reformat c++ code 3 years ago
int_fastdiv.cuh style(all): reformat c++ code 3 years ago
integer_subbyte_utils.cuh style(all): reformat c++ code 3 years ago
memory_utils.cuh style(all): reformat c++ code 3 years ago
query_blocksize.cpp style(all): reformat c++ code 3 years ago
query_blocksize.cuh style(all): reformat c++ code 3 years ago
query_blocksize_impl.cu style(all): reformat c++ code 3 years ago
reduce_helper.cuh style(all): reformat c++ code 3 years ago
reduce_helper.cuinl chore(mge): update copyright years 4 years ago
transpose_utils.cuh style(all): reformat c++ code 3 years ago
utils.cpp style(all): reformat c++ code 3 years ago
utils.cuh fix(mgb): fix bug caused by conv filter size is too big 3 years ago
utils.h style(all): reformat c++ code 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