Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
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
41b9db85e2
|
3 years ago | |
|---|---|---|
| .. | ||
| adaptive_pooling | 3 years ago | |
| add_update | 3 years ago | |
| argmxx | 3 years ago | |
| argsort | 3 years ago | |
| batch_conv_bias | 3 years ago | |
| batch_normalization | 3 years ago | |
| batched_matrix_mul | 3 years ago | |
| check_non_finite | 3 years ago | |
| checksum | 3 years ago | |
| concat | 3 years ago | |
| cond_take | 3 years ago | |
| conv_bias | 3 years ago | |
| convolution | 3 years ago | |
| convolution3d | 3 years ago | |
| convolution_helper | 3 years ago | |
| convpooling | 3 years ago | |
| correlation | 3 years ago | |
| cub | 4 years ago | |
| cumsum | 3 years ago | |
| cutlass | 3 years ago | |
| cv | 3 years ago | |
| cvt_color | 3 years ago | |
| dct | 3 years ago | |
| deformable_conv | 3 years ago | |
| deformable_ps_roi_pooling | 3 years ago | |
| diag | 3 years ago | |
| dot | 3 years ago | |
| dropout | 3 years ago | |
| elemwise | 3 years ago | |
| elemwise_multi_type | 3 years ago | |
| eye | 3 years ago | |
| fake_quant | 3 years ago | |
| fill | 3 years ago | |
| flip | 3 years ago | |
| gaussian_blur | 3 years ago | |
| group_local | 3 years ago | |
| images2neibs | 3 years ago | |
| indexing_multi_axis_vec | 3 years ago | |
| indexing_one_hot | 3 years ago | |
| kernel_common | 3 years ago | |
| lamb | 3 years ago | |
| layer_norm | 3 years ago | |
| linspace | 3 years ago | |
| local | 3 years ago | |
| local_share | 3 years ago | |
| lrn | 3 years ago | |
| lsq | 3 years ago | |
| mask_conv | 3 years ago | |
| matrix_inverse | 3 years ago | |
| matrix_mul | 3 years ago | |
| max_tensor_diff | 3 years ago | |
| megcore | 3 years ago | |
| mesh_indexing | 3 years ago | |
| norm | 3 years ago | |
| padding | 3 years ago | |
| param_pack | 3 years ago | |
| pooling | 3 years ago | |
| powc | 3 years ago | |
| ptx/uint4_int4 | 3 years ago | |
| reduce | 3 years ago | |
| reduce_helper | 3 years ago | |
| relayout | 3 years ago | |
| relayout_format | 3 years ago | |
| remap | 3 years ago | |
| repeat | 3 years ago | |
| resize | 3 years ago | |
| rng | 3 years ago | |
| roi_align | 3 years ago | |
| roi_copy | 3 years ago | |
| roi_pooling | 3 years ago | |
| rotate | 3 years ago | |
| separable_conv | 3 years ago | |
| separable_filter | 3 years ago | |
| sleep | 3 years ago | |
| sliding_window_transpose | 3 years ago | |
| softmax | 3 years ago | |
| split | 3 years ago | |
| svd | 3 years ago | |
| tensor_remap | 3 years ago | |
| tile | 3 years ago | |
| topk | 3 years ago | |
| tqt | 3 years ago | |
| transpose | 3 years ago | |
| type_cvt | 3 years ago | |
| warp_affine | 3 years ago | |
| warp_perspective | 3 years ago | |
| atomic_add.cuh | 3 years ago | |
| cuda_shfl_compat.cuh | 3 years ago | |
| cudnn_with_check.h | 3 years ago | |
| cudnn_wrapper.cpp | 3 years ago | |
| cudnn_wrapper.h | 3 years ago | |
| cudnn_wrapper_v8.cpp | 3 years ago | |
| cudnn_wrapper_v8.h | 3 years ago | |
| elemwise_helper.cpp | 3 years ago | |
| elemwise_helper.cuh | 3 years ago | |
| elemwise_helper_q4.cuh | 3 years ago | |
| error_info.cuh | 3 years ago | |
| fp16_help.cuh | 3 years ago | |
| handle.cpp | 3 years ago | |
| handle.h | 3 years ago | |
| handle_create.cpp | 3 years ago | |
| int_fastdiv.cpp | 3 years ago | |
| int_fastdiv.cuh | 3 years ago | |
| integer_subbyte_utils.cuh | 3 years ago | |
| memory_utils.cuh | 3 years ago | |
| ptx_loader.cpp | 3 years ago | |
| ptx_loader.h | 3 years ago | |
| query_blocksize.cpp | 3 years ago | |
| query_blocksize.cuh | 3 years ago | |
| query_blocksize_impl.cu | 3 years ago | |
| reduce_helper.cuh | 3 years ago | |
| reduce_helper.cuinl | 3 years ago | |
| transpose_utils.cuh | 3 years ago | |
| utils.cpp | 3 years ago | |
| utils.cuh | 3 years ago | |
| utils.h | 3 years ago | |
MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境，不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序，请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉，欢迎访问 MegStudio 平台
C++ Cuda Python C SVG other
Contributors (25+)
megengine@megvii.com
huangxinda@megvii.com
luzzyzhang@gmail.com
wenjuan@megvii.com
1041563782@qq.com
xxr@megvii.com
61608041+megvii-mge@users.noreply.github.com
1186454801@qq.com
35462954+XindaH@users.noreply.github.com
acdoge.cao@gmail.com
75015964+HuaHua404@users.noreply.github.com
1271808136@qq.com
70926289+kagome1007@users.noreply.github.com
chenjiahui@megvii.com
liuke@megvii.com
quqi.liu@shopee.com
khj.application@aliyun.com
zhanghaolong@megvii.com
974658390@qq.com
2283984853@qq.com
lijiansong@ict.ac.cn
loves0510@outlook.com
p2oileen@whu.edu.cn
1157539822@qq.com
mediosrity@gmail.com