Megvii Engine Team
3b41840b68
fix(mgb): change caffepooling log level
GitOrigin-RevId: 290d243ef5
3 years ago
Megvii Engine Team
1999307015
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
3 years ago
Megvii Engine Team
32717b0ca4
fix(build): split some cpp, which consume two many mem when build
make build possible at 8G ddr env, when -j8
GitOrigin-RevId: d0c442b41d
3 years ago
Megvii Engine Team
a93741815b
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
3 years ago
Megvii Engine Team
a404cd7d06
fix(mgb/src): add tensorRT version check
GitOrigin-RevId: 7abfd30cab
3 years ago
Megvii Engine Team
c53cad2049
feat(cmake): format all cmake file
GitOrigin-RevId: 0a4ecab99b
3 years ago
Megvii Engine Team
a5803058b4
fix(dnn/x86): opt algo order
GitOrigin-RevId: 6dd14f9a96
3 years ago
Megvii Engine Team
c90e0b54be
perf(arm): optimize arm uint16 relayout with n=4
GitOrigin-RevId: 5779c6b9c1
3 years ago
Megvii Engine Team
f6d9909460
feat(dnn): add elemwise multi type support i16xf32 and u8xf32
GitOrigin-RevId: 2fe469bb4e
3 years ago
Megvii Engine Team
d9a46ea47b
fix(dnn): correct behaviour of floor div for int tensor
GitOrigin-RevId: 1444f69cce
3 years ago
Megvii Engine Team
0ad5eeaedd
feat(mgb/gopt): global layout transform support opencl
GitOrigin-RevId: 132605c7d9
3 years ago
Megvii Engine Team
2881934cb8
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
3 years ago
Megvii Engine Team
6bb5409976
feat(dnn/src): add images2neibs kernel of opencl and related test
GitOrigin-RevId: 82242b7437
3 years ago
Megvii Engine Team
6ce4a34403
feat(dnn): add fallback postprocess
GitOrigin-RevId: 4201a0f158
3 years ago
Megvii Engine Team
c96dbd29b8
fix(dnn/arm_common): support more monotonous case in arm typecvt for performance
GitOrigin-RevId: 9e28a64d93
3 years ago
Megvii Engine Team
ead611e11d
perf(dnn): slightly improve arm neon transcendental function performance
GitOrigin-RevId: 210d88f81e
3 years ago
Megvii Engine Team
0d16952470
fix(mgb/cuda): fix conv error when the input tensor is too large
GitOrigin-RevId: 1b1d693795
3 years ago
Megvii Engine Team
5e07e1e0f9
fix(dnn/falback): let cpu be able to execute int4 model
GitOrigin-RevId: 1a6b78f3b6
3 years ago
Megvii Engine Team
2696e4efaa
feat(dnn): add float16 for remap backward
GitOrigin-RevId: 0263030051
3 years ago
Megvii Engine Team
1f0cc891b0
feat(dnn): enable eye to support bool
GitOrigin-RevId: 76d874d5b7
3 years ago
Megvii Engine Team
11d75fecb5
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
3 years ago
Megvii Engine Team
2318ea3f15
fix(dnn): fix naive average pooling overflow bug for int8 type
GitOrigin-RevId: b60a7b6cf8
3 years ago
Megvii Engine Team
2d54ad185b
feat(lite): add global layout transform interface for load and run
GitOrigin-RevId: 65c2430ec2
3 years ago
Megvii Engine Team
ba2f0c2e48
fix(dnn/cuda): fix cudnn_conv algo of conv_bias opr for fp16 add z cases
GitOrigin-RevId: b29b009de0
3 years ago
Megvii Engine Team
30976c239f
fix(mgb/gopt): fix global layout transform
add a special opr_format modify function for concat operators to modify concat axis when input's layout has been changed
GitOrigin-RevId: 4094208057
3 years ago
Megvii Engine Team
ca7cec7a5d
fix(mgb/gopt): minor fixes for global layout transform
Merge Target::ARM and Target::X86 into Target::CPU to make global layout transform easier to use
GitOrigin-RevId: cc9363fa38
3 years ago
Megvii Engine Team
fe93013a6e
feat(mgb/gopt): global layout transform support nchw_nchwxx hybrid mode
GitOrigin-RevId: 6d5b55d7fc
3 years ago
Megvii Engine Team
b59e8ccf24
fix(mgb): fix cambricon bangc copybara
GitOrigin-RevId: 22b57364e8
3 years ago
Megvii Engine Team
3116e128c5
fix(ci/integration_test): fix benchmark torch version
GitOrigin-RevId: bd964ed505
3 years ago
Megvii Engine Team
c85631aa77
feat(dnn): use ref ptr interface for all backends
GitOrigin-RevId: f65feae5cc
3 years ago
Megvii Engine Team
d90cb7763c
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
3 years ago
Megvii Engine Team
89186edc5d
fix(dnn): correct reduce/argmxx/fakequant calculation with nan
GitOrigin-RevId: 7e78bdae91
3 years ago
Megvii Engine Team
68cdabd288
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
3 years ago
Megvii Engine Team
a1cba6cc27
fix(dnn): fix convbias crash on X86
GitOrigin-RevId: 3b33c83d03
3 years ago
Megvii Engine Team
9b4cd92ba3
fix(mgb/dnn): fix cudnnConvBiasActivation crash on nchw32 int8 with oc > 256
GitOrigin-RevId: 20c0b90575
3 years ago
Megvii Engine Team
23c1fda7e6
perf(arm_common): optimize sigmoid
GitOrigin-RevId: 7cb248a15b
3 years ago
Megvii Engine Team
c48d58daa8
feat(dnn/arm_common): add N1HW like elemwise broadcast mode
GitOrigin-RevId: 2895135801
3 years ago
Megvii Engine Team
26634db7a8
fix(dnn): support relayout for non-contigous layout
GitOrigin-RevId: 44a0adddba
3 years ago
Megvii Engine Team
056fd6bc59
feat(dnn/arm64): support stride_m in arm64 relayout
GitOrigin-RevId: c74193a23d
3 years ago
Megvii Engine Team
c50858ee13
fix(dnn): specialize pow to make it consistent
GitOrigin-RevId: cff3bbbadd
3 years ago
Megvii Engine Team
849f0ece9d
fix(dnn): drop batched matmul cublas algo when batch is 1
GitOrigin-RevId: 71126a27b0
3 years ago
Megvii Engine Team
b5bf56e0ee
style(dnn): add bypass of clang-format for dnn foreach_opr macro
GitOrigin-RevId: 2f4bfcff42
3 years ago
Megvii Engine Team
5af52746f7
fix(mgb): fix bug caused by conv filter size is too big
GitOrigin-RevId: 7fe3a9fcf1
3 years ago
liuke
b0ba6d3201
Merge pull request #207 from togetherwhenyouwant:feat-x86-matmul-6x16x2
GitOrigin-RevId: 148ae44ba0
3 years ago
Megvii Engine Team
10af44abba
fix(dnn/cuda): fix cudnn conv impl for nchw4_nchw hybrid layout
the conv_bias algo *_IMPLICIT_GEMM in cudnn less than 8.0.0 is disabled due to the incorrect result for int8x4->f32 configs
GitOrigin-RevId: 7cc52d0a85
3 years ago
Megvii Engine Team
5885b137fa
feat(dnn/arm): support layout like NHWC channel like broadcast on arm
GitOrigin-RevId: fb4300004c
3 years ago
Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
3 years ago
Megvii Engine Team
bfb30dcb81
chore(format): fix compile bugs after code format
GitOrigin-RevId: 11a4b06f6f
3 years ago
Megvii Engine Team
eeccf2bc0d
ci(check): add clang-format in check stage
GitOrigin-RevId: 25c24d7856
3 years ago
zjl
d2184af3b2
feat(dnn/src/x86/matmul): add matmul_6x16 for x86
3 years ago