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 3d6fe33e2d Revert "fix(imperative): check async error when getting value" 3 years ago
..
common.cpp feat(cuda/comp_node): gets (maximum) GPU memory allocated/reserved 3 years ago
common.h style(all): reformat c++ code 3 years ago
grad.cpp style(all): reformat c++ code 3 years ago
grad.h style(all): reformat c++ code 3 years ago
grad_info.h style(all): reformat c++ code 3 years ago
grad_override.cpp style(all): reformat c++ code 3 years ago
graph_rt.cpp feat(mge): support dump with specific format 3 years ago
graph_rt.h style(all): reformat c++ code 3 years ago
helper.cpp style(all): reformat c++ code 3 years ago
helper.h style(all): reformat c++ code 3 years ago
imperative_rt.cpp style(all): reformat c++ code 3 years ago
imperative_rt.h chore(mge): update copyright years 4 years ago
intrusive_list.h style(all): reformat c++ code 3 years ago
module.cpp style(all): reformat c++ code 3 years ago
module_trace.cpp style(all): reformat c++ code 3 years ago
module_trace.h style(all): reformat c++ code 3 years ago
numpy_dtypes.cpp style(all): reformat c++ code 3 years ago
numpy_dtypes.h style(all): reformat c++ code 3 years ago
numpy_dtypes_bfloat16.cpp style(all): reformat c++ code 3 years ago
numpy_dtypes_intbx.cpp fix(imperative): fix error message for tensors with intbx data type 3 years ago
ops.cpp fix(mge/subgraph): fix mem leak 3 years ago
ops.h style(all): reformat c++ code 3 years ago
pyext17.h style(all): reformat c++ code 3 years ago
tensor.cpp Revert "fix(imperative): check async error when getting value" 3 years ago
tensor.h refactor(imperative): remove swap in dtr 3 years ago
trace.cpp style(all): reformat c++ code 3 years ago
trace.h style(all): reformat c++ code 3 years ago
trace_info.h style(all): reformat c++ code 3 years ago
utils.cpp refactor(imperative): remove enable_defrag interface as it is enabled by default 3 years ago
utils.h chore(mge): update copyright years 4 years ago

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

C++ Cuda Python C SVG other