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 1b1ad56a82 fix(mgb/gopt): fix warp fusion opt pass 5 years ago
..
dnn chore(megbrain): fix megskull cnn/img opr compatiblity 5 years ago
internal feat(jit): add memfwd in jit executor opr 5 years ago
loop feat(dnn): add uint16 support 5 years ago
nvof feat(opr/nvof): add nvof operator 5 years ago
search_policy fix(mgb): fix fastrun log 5 years ago
standalone feat(mge/imperative): run oss test and restore cmake list build items 5 years ago
atlas_runtime_op.cpp fix(mgb/atlas): when batchsize more than model max batchsize 5 years ago
atlas_runtime_op.oprdecl fix(mgb): remove internal for cambricon and atlas 5 years ago
atlas_runtime_op.sereg.h fix(mgb): remove internal for cambricon and atlas 5 years ago
basic_arith.cpp fix(mgb/opr): fix Reduce static value inference 5 years ago
basic_arith.oprdecl MegEngine: Initial commit of MegEngine. 5 years ago
basic_arith.sereg.h MegEngine: Initial commit of MegEngine. 5 years ago
blas.cpp feat(mgb): add fastrun support for matmul/batchedmatrixmul 5 years ago
blas.oprdecl feat(mgb): add fastrun support for matmul/batchedmatrixmul 5 years ago
blas.sereg.h feat(mgb): add fastrun support for matmul/batchedmatrixmul 5 years ago
cond.cpp fix(mgb): fix wrong use of macro MGB_ENABLE_GRAD 5 years ago
cond.oprdecl MegEngine: Initial commit of MegEngine. 5 years ago
cond.sereg.h MegEngine: Initial commit of MegEngine. 5 years ago
imgproc.cpp fix(mgb/gopt): fix warp fusion opt pass 5 years ago
imgproc.oprdecl chore(megbrain): fix megskull cnn/img opr compatiblity 5 years ago
imgproc.sereg.h chore(megbrain): format sereg.h code 5 years ago
indexing.cpp fix(mgb): fix wrong use of macro MGB_ENABLE_GRAD 5 years ago
indexing.oprdecl MegEngine: Initial commit of MegEngine. 5 years ago
indexing.sereg.h MegEngine: Initial commit of MegEngine. 5 years ago
io.cpp feat(src/core): free weight preprocessed weight 5 years ago
io.oprdecl MegEngine: Initial commit of MegEngine. 5 years ago
io.sereg.h feat(opr): add a constant flavor for SharedDeviceTensor 5 years ago
mgb_cpp_opr.fbs feat(opr/standalone): import NMSKeep from MegSkull contrib 5 years ago
misc.cpp fix(mge/dnn): fix rng and topk oom in distributed training 5 years ago
misc.oprdecl feat(opr/nvof): add nvof operator 5 years ago
misc.sereg.h feat(opr/nvof): add nvof operator 5 years ago
muxing.cpp fix(mgb): fix wrong use of macro MGB_ENABLE_GRAD 5 years ago
nn_int.cpp MegEngine: Initial commit of MegEngine. 5 years ago
nn_int.oprdecl MegEngine: Initial commit of MegEngine. 5 years ago
nn_int.sereg.h MegEngine: Initial commit of MegEngine. 5 years ago
rand.cpp fix(mge/dnn): fix rng and topk oom in distributed training 5 years ago
rand.oprdecl MegEngine: Initial commit of MegEngine. 5 years ago
rand.sereg.h MegEngine: Initial commit of MegEngine. 5 years ago
tensor_gen.cpp fix(mgb): fix wrong use of macro MGB_ENABLE_GRAD 5 years ago
tensor_gen.oprdecl MegEngine: Initial commit of MegEngine. 5 years ago
tensor_gen.sereg.h MegEngine: Initial commit of MegEngine. 5 years ago
tensor_manip.cpp chore(winograd): remove winograd transform code 5 years ago
tensor_manip.oprdecl MegEngine: Initial commit of MegEngine. 5 years ago
tensor_manip.sereg.h chore(winograd): remove winograd transform code 5 years ago
utility.cpp feat(mgb/core): add shape hint for graph optimization 5 years ago
utility.oprdecl feat(mgb/core): add shape hint for graph optimization 5 years ago
utility.sereg.h feat(mgb/core): add shape hint for graph optimization 5 years ago

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

C++ Cuda Python C SVG other