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 e0fffa5286 fix(mge/tensor): fix mem leak when getting value in trace 4 years ago
..
common.cpp perf(mge/imperative): move convert_inputs from python to C++ 4 years ago
common.h fix(imperative): fix hardcode of default device 4 years ago
grad.cpp Revert "perf(mge): add memory optimization for backward graph" 4 years ago
grad.h feat(mge): restore remote send/recv 4 years ago
grad_info.h fix(mge): fix segfault with Function returning unused grads 4 years ago
grad_override.cpp feat(imperative): add inplace add_update option in optimizer 4 years ago
graph_rt.cpp refactor(mge/virtualdep): remove virtualdep opdef 4 years ago
graph_rt.h fix(imperative): set exception which worker thread throws for rendezvous 4 years ago
helper.cpp perf(mge/imperative): move convert_inputs from python to C++ 4 years ago
helper.h fix(imperative): fix persistent_cache 4 years ago
imperative_rt.cpp refactor(mge): migrate to new core implementation 4 years ago
imperative_rt.h refactor(mge): cpp license 4 years ago
intrusive_list.h refactor(mge/imperative): implement new tensor system 4 years ago
module.cpp refactor(mge): migrate to new core implementation 4 years ago
numpy_dtypes.cpp chore(mge): clean up before merge to dev 4 years ago
numpy_dtypes.h perf(mge/imperative): move convert_inputs from python to C++ 4 years ago
numpy_dtypes_bfloat16.cpp refactor(mge): cpp license 4 years ago
numpy_dtypes_intbx.cpp refactor(mge): cpp license 4 years ago
ops.cpp feat(mge): add python custom op 4 years ago
ops.h refactor(mge): cpp license 4 years ago
pyext17.h refactor(mge/imperative): implement new tensor system 4 years ago
tensor.cpp fix(mge/tensor): fix mem leak when getting value in trace 4 years ago
tensor.h Revert "perf(mge): add memory optimization for backward graph" 4 years ago
trace.cpp fix(mge): fix trace exit code and reformat 4 years ago
trace.h refactor(mge): trace exception in compiled info 4 years ago
trace_info.h fix(mge): fix trace exit code and reformat 4 years ago
utils.cpp feat(imperative): port persistent cache 4 years ago
utils.h refactor(mge): cpp license 4 years ago

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

C++ Cuda Python C SVG other