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 b3e54eade1 feat(dnn/bn): use new cudnn BN kernel to support NHWC 2 years ago
..
cutlass_generator fix(dnn): fix_short_cutlass_name_gemm 2 years ago
gen_heuristic MegEngine: Initial commit of MegEngine. 4 years ago
Makefile refactor(dnn/cuda): refactor cutlass kernel generator for gemm and gemv 2 years ago
gen_cond_take_kern_impls.py feat(mge): add bfloat16 support 4 years ago
gen_cuda_batch_conv_bias_kern_impls.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_cuda_conv_bias_kern_impls.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_elemwise_each_mode.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_elemwise_kern_impls.py fix(mgb/rocm): remove begin-internal of rocm 3 years ago
gen_elemwise_multi_type_kern_impls.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_elemwise_multi_type_utils.py feat(mge/opr): add silu and gelu 2 years ago
gen_elemwise_special_kern_impls.py fix(mgb/rocm): remove begin-internal of rocm 3 years ago
gen_elemwise_utils.py feat(mge/opr): add silu and gelu 2 years ago
gen_flatbuffers_converter.py MegEngine: Initial commit of MegEngine. 4 years ago
gen_flatbuffers_schema.py refactor(mgb/dnn): refactor enum used in serializing 2 years ago
gen_param_defs.py refactor(mgb/dnn): refactor enum used in serializing 2 years ago
gen_tablegen.py refactor(mgb/dnn): refactor enum used in serializing 2 years ago
opr_param_defs.py feat(dnn/bn): use new cudnn BN kernel to support NHWC 2 years ago