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 43098fb8f1 feat(mge): add SlidingWindowTranspose opr 4 years ago
..
add_update.cpp chore(mge): update copyright years 4 years ago
check_has_inf.cpp feat(mgb/opr): add check_has_inf kernel and opr 3 years ago
conv_bias.cpp feat(dnn/cuda): add conv u4xs4 sass kernel 4 years ago
convolution.cpp chore(mge): update copyright years 4 years ago
cvt_color.cpp chore(mge): update copyright years 4 years ago
dct.cpp chore(mge): update copyright years 4 years ago
deformable_conv.cpp chore(mge): update copyright years 4 years ago
deformable_ps_roi_pooling.cpp chore(mge): update copyright years 4 years ago
elemwise_multi_type.cpp feat(dnn/cuda): add cuda elemwise int4 4 years ago
fixture.cpp chore(mge): update copyright years 4 years ago
fixture.h chore(mge): update copyright years 4 years ago
images2neibs.cpp feat(mge/module): add python wrapper for unfold 4 years ago
lsq.cpp feat(mgb/opr): add lsq opr 3 years ago
matrix_inverse.cpp chore(mge): update copyright years 4 years ago
matrix_mul.cpp chore(mge): update copyright years 4 years ago
mesh_indexing.cpp chore(mge): update copyright years 4 years ago
pooling.cpp feat(dnn/cuda): add cuda uint4 pooling 4 years ago
reduce.cpp chore(mge): update copyright years 4 years ago
relayout_format.cpp fix(dnn/cuda): add cuda nchw int8 conv impl with nchw4 to fix cu111 compatibility 4 years ago
resize.cpp chore(mge): update copyright years 4 years ago
rng.cpp feat(mge/random): add some random op and remove random/distrbution.py 3 years ago
rng.h chore(mge): update copyright years 4 years ago
sliding_window_transpose.cpp feat(mge): add SlidingWindowTranspose opr 3 years ago
svd.cpp chore(mge): update copyright years 4 years ago
type_cvt.cpp chore(mge): update copyright years 4 years ago
warp_afffine.cpp chore(mge): update copyright years 4 years ago
warp_perspective.cpp feat(dnn/cuda): add nhwc 4bit warp perspective 4 years ago

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

C++ Cuda Python C SVG other