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 0558b2123d feat(mge/opr): add interpolate nearest mode 3 years ago
..
megcore chore(mge): update copyright years 4 years ago
accuracy_shake_checker.cpp feat(mgb/dnn): add accuracy shake checker 4 years ago
accuracy_shake_checker.h refactor(mgb/dnn): fix group conv is_available 3 years ago
adaptive_pooling.h chore(mge): update copyright years 4 years ago
benchmark_basic_types.cpp chore(mge): update copyright years 4 years ago
benchmarker.h feat(dnn/cuda): add relayout format when width is an odd number 4 years ago
bn.h chore(mge): update copyright years 4 years ago
checker.cpp feat(mgb/opr): add check_has_inf kernel and opr 3 years ago
checker.h feat(mgb/opr): add check_has_inf kernel and opr 3 years ago
comparator.h chore(mge): update copyright years 4 years ago
comparator.inl chore(mge): update copyright years 4 years ago
cond_take.cpp chore(mge): update copyright years 4 years ago
cond_take.h chore(mge): update copyright years 4 years ago
conv_bias.cpp feat(imperative): add support for quantized conv transpose2d 3 years ago
conv_bias.h feat(imperative): add support for quantized conv transpose2d 3 years ago
conv_pooling.cpp chore(mge): update copyright years 4 years ago
conv_pooling.h chore(mge): update copyright years 4 years ago
convolution.cpp refactor(mgb): move algo's name from info to desc and delete some algo's unnecessary param() method 4 years ago
convolution.h feat(gopt): add deconv nchw4 int8 opt pass, add deconv nchw int8 4 years ago
convolution3d.cpp chore(mge): update copyright years 4 years ago
convolution3d.h chore(mge): update copyright years 4 years ago
correlation.h feat(megbrain): add correlation opr 4 years ago
cvt_color.h chore(mge): update copyright years 4 years ago
dct_ref.cpp chore(mge): update copyright years 4 years ago
dct_ref.h chore(mge): update copyright years 4 years ago
deduce_layout_proxy.h chore(mge): update copyright years 4 years ago
dtype.cpp chore(mge): update copyright years 4 years ago
elemwise.cpp feat(log): opt log 4 years ago
elemwise.h fix(dnn/opencl): fix elemwise negative stride support 4 years ago
elemwise_multi_type.cpp chore(mge): update copyright years 4 years ago
elemwise_multi_type.h chore(mge): update copyright years 4 years ago
exec_proxy.h feat(mgb/opr): add check_has_inf kernel and opr 3 years ago
extra_impl_helper.cpp chore(mge): update copyright years 4 years ago
extra_impl_helper.h chore(mge): update copyright years 4 years ago
fake_quant.h chore(mge): update copyright years 4 years ago
fast_run_cache.cpp chore(mge): update copyright years 4 years ago
fast_run_cache.h chore(mge): update copyright years 4 years ago
fill.h feat(dnn): add fill kernel 3 years ago
fix_gtest_on_platforms_without_exception.inl chore(mge): update copyright years 4 years ago
flip.h chore(mge): update copyright years 4 years ago
gaussian_blur.h chore(mge): update copyright years 4 years ago
get_dtype_from_static_type.h chore(mge): update copyright years 4 years ago
group_local.h chore(mge): update copyright years 4 years ago
images2neibs.h feat(mge/module): add python wrapper for unfold 4 years ago
index.cpp chore(mge): update copyright years 4 years ago
index.h chore(mge): update copyright years 4 years ago
indexing_multi_axis_vec.h chore(mge): update copyright years 4 years ago
indexing_one_hot.cpp chore(mge): update copyright years 4 years ago
indexing_one_hot.h chore(mge): update copyright years 4 years ago
inspect_type.h chore(mge): update copyright years 4 years ago
local.h chore(mge): update copyright years 4 years ago
lsq.h feat(mgb/opr): add lsq opr 3 years ago
mask_conv.h chore(mge): update copyright years 4 years ago
matrix_mul.cpp feat(dnn/cuda): add tensorcore matmul for fp16 data type 3 years ago
matrix_mul.h feat(dnn/cuda): add tensorcore matmul for fp16 data type 3 years ago
memory_manager.cpp chore(mge): update copyright years 4 years ago
memory_manager.h chore(mge): update copyright years 4 years ago
mesh_indexing.h chore(mge): update copyright years 4 years ago
named_tensor.cpp feat(dnn/common): add named tensor shape 3 years ago
null_dispatcher.h chore(mge): update copyright years 4 years ago
opr_algo_proxy.h feat(dnn): add deterministic max pooling 3 years ago
opr_proxy.h refactor(mgb): move algo's name from info to desc and delete some algo's unnecessary param() method 4 years ago
padding.h feat(mgb): add padding opr for megbrain 3 years ago
pooling.h chore(mge): update copyright years 4 years ago
powc.cpp chore(mge): update copyright years 4 years ago
powc.h feat(dnn): add indexing_one_hot and indexing_set_one_hot opr 4 years ago
random_state.cpp chore(mge): update copyright years 4 years ago
random_state.h chore(mge): update copyright years 4 years ago
relayout.cpp chore(mge): update copyright years 4 years ago
relayout.h chore(mge): update copyright years 4 years ago
remap.h chore(mge): update copyright years 4 years ago
resize.h feat(mge/opr): add interpolate nearest mode 3 years ago
rng.cpp feat(mgb/opr): add check_has_inf kernel and opr 3 years ago
rng.h feat(mgb/opr): add check_has_inf kernel and opr 3 years ago
roi_copy.h chore(mge): update copyright years 4 years ago
roi_pooling.h chore(mge): update copyright years 4 years ago
rotate.h chore(mge): update copyright years 4 years ago
separable.cpp chore(mge): update copyright years 4 years ago
separable_conv.h chore(mge): update copyright years 4 years ago
separable_filter.h chore(mge): update copyright years 4 years ago
sliding_window_transpose.h feat(mge): add SlidingWindowTranspose opr 3 years ago
small_vector.cpp feat(imperative): add support for quantized conv transpose2d 3 years ago
svd.cpp chore(mge): update copyright years 4 years ago
svd.h chore(mge): update copyright years 4 years ago
task_executor.cpp chore(mge): update copyright years 4 years ago
tensor.cpp chore(mge): update copyright years 4 years ago
tensor.h chore(mge): update copyright years 4 years ago
tensor.inl chore(mge): update copyright years 4 years ago
tensor_remap.cpp chore(mge): update copyright years 4 years ago
tensor_remap.h chore(mge): update copyright years 4 years ago
test_basic_types.cpp feat(imperative): add support for quantized conv transpose2d 3 years ago
tile_repeat.h chore(mge): update copyright years 4 years ago
timer.h chore(mge): update copyright years 4 years ago
topk.cpp fix(opencl): open opencl topk test when opencl beyond 2.0 3 years ago
topk.h chore(mge): update copyright years 4 years ago
tqt.h chore(mge): update copyright years 4 years ago
utils.cpp chore(mge): update copyright years 4 years ago
utils.h fix(core): fix nr_threads is zero 3 years ago
warp_affine.h chore(mge): update copyright years 4 years ago
warp_perspective.cpp chore(mge): update copyright years 4 years ago
warp_perspective.h fix(dnn/cuda): fix int4 epilogue stg bug 4 years ago
winograd_filter_preprocess.h chore(mge): update copyright years 4 years ago
workspace_wrapper.cpp chore(mge): update copyright years 4 years ago
workspace_wrapper.h chore(mge): update copyright years 4 years ago

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

C++ Cuda Python C SVG other