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 65d554ed8e refactor(dtr): improve dtr in computing graph 3 years ago
..
common.cpp feat(dnn/rocm): support bool in type_cvt and elemwise 3 years ago
common.h chore(mge): update copyright years 4 years ago
grad.cpp refactor(opmeth): add OpMethCache struct 3 years ago
grad.h fix(autograd): make higher order grad experimental 3 years ago
grad_info.h feat(autodiff): support higher order grad 3 years ago
grad_override.cpp fix(imperative/grad): fix hardcode dtype in subtensor_grad_rule 3 years ago
graph_rt.cpp refactor(dtr): improve dtr in computing graph 3 years ago
graph_rt.h chore(mge): update copyright years 4 years ago
helper.cpp fix(dnn): fixes for int4 4 years ago
helper.h chore(mge): update copyright years 4 years ago
imperative_rt.cpp refactor(subgraph): add generic encoded_graph 3 years ago
imperative_rt.h chore(mge): update copyright years 4 years ago
intrusive_list.h chore(mge): update copyright years 4 years ago
module.cpp chore(mge): update copyright years 4 years ago
module_trace.cpp feat(mge): add opdef serialization and apply_module_trace 3 years ago
module_trace.h feat(mge): add opdef serialization and apply_module_trace 3 years ago
numpy_dtypes.cpp fix(imperative/dtype): fix is_dtype_equal 4 years ago
numpy_dtypes.h chore(mge): update copyright years 4 years ago
numpy_dtypes_bfloat16.cpp chore(mge): update copyright years 4 years ago
numpy_dtypes_intbx.cpp chore(mge): update copyright years 4 years ago
ops.cpp feat(src): add MGB_CUSTOM_OP macro 3 years ago
ops.h feat(imperative): add adaptor between custom op and imperative runtime 3 years ago
pyext17.h feat(mge): improve presentation of async errors 3 years ago
tensor.cpp perf(profiler): reduce profiler host overhead 3 years ago
tensor.h feat(mge): add opdef serialization and apply_module_trace 3 years ago
trace.cpp refactor(ops): remove BackwardGraph op 4 years ago
trace.h chore(mge): update copyright years 4 years ago
trace_info.h refactor(mge/jit): make trace return any kind of output 4 years ago
utils.cpp fix(imperative): persistent cache write through on put 4 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