huangxinda
|
a0862865fe
|
feat(mge/third_party): update cutlass version
|
3 years ago |
Megvii Engine Team
|
972a8d5489
|
feat(ci): add model compatibility check in ci
GitOrigin-RevId: d17c21ab4f
|
3 years ago |
Megvii Engine Team
|
e694301721
|
fix(ops/jit): skip lookup include path when nvcc executable not found
GitOrigin-RevId: f5e7dce1c5
|
3 years ago |
Megvii Engine Team
|
02c1a0c3de
|
fix(imperative/format): warn once when parameter without format been attached
GitOrigin-RevId: 02174447f6
|
3 years ago |
Megvii Engine Team
|
c2deef1a97
|
feat(mge): aad atlas710 support
GitOrigin-RevId: 6458c5c23c
|
3 years ago |
Megvii Engine Team
|
df5ebd3da7
|
fix(imperative/ops): fix the vmemory problem in 1.9
GitOrigin-RevId: 728e983701
|
3 years ago |
Megvii Engine Team
|
4e66e0eb1f
|
feat(megdnn/softmax): add softmax operator in OpenCL
GitOrigin-RevId: e207d6ceb4
|
3 years ago |
Megvii Engine Team
|
b36b5bd8cb
|
refactor(mgb): check input when profiling
GitOrigin-RevId: 1d722dd741
|
3 years ago |
Megvii Engine Team
|
6c9b3a58e3
|
refactor(dnn): remove algorithm cache queries
GitOrigin-RevId: b7a1dc62d8
|
3 years ago |
Megvii Engine Team
|
8563f51404
|
fix(imperative): fix buildin reduce keepdim
GitOrigin-RevId: 38d90ab38a
|
3 years ago |
Megvii Engine Team
|
96d90be1c6
|
feat(dnn): fallback support int4 relayout
GitOrigin-RevId: 3625f58470
|
3 years ago |
Megvii Engine Team
|
eef0308b29
|
feat(serialization): add no_change_graph and version param whem dump model
GitOrigin-RevId: 65064452c9
|
3 years ago |
Megvii Engine Team
|
4ab5f970e9
|
fix(build): fix ci error
GitOrigin-RevId: 9cbf64dda2
|
3 years ago |
Megvii Engine Team
|
b9a6932341
|
feat(imperative): channel default model format to fbs v2
GitOrigin-RevId: 6066516c31
|
3 years ago |
Megvii Engine Team
|
283a2f77e7
|
feat(serialization): support test for new serialization format
GitOrigin-RevId: 9a8047ec0b
|
3 years ago |
Megvii Engine Team
|
50faabf614
|
feat(serialization): support the registry for new serialization format
GitOrigin-RevId: 8eacd5e77c
|
3 years ago |
Megvii Engine Team
|
a694fb3304
|
feat(serialization): implement the new serialization format
GitOrigin-RevId: 00f87f7ccd
|
3 years ago |
Megvii Engine Team
|
ca4a5da013
|
feat(serialization): add new serialization format define
GitOrigin-RevId: 177bfdd56b
|
3 years ago |
Megvii Engine Team
|
ba32360af4
|
feat(lite): add set opencl buffer kernel cache lite api
GitOrigin-RevId: 0605218238
|
3 years ago |
Megvii Engine Team
|
711b5bf502
|
fix(dnn/arm_common): fix some load beyond memory
GitOrigin-RevId: acd6363945
|
3 years ago |
Megvii Engine Team
|
3ebb8db01a
|
feat(third_party/cutlass): update to version 2.8
GitOrigin-RevId: 9de584b3b8
|
3 years ago |
Megvii Engine Team
|
da91e650a5
|
refactor(ops/layer_norm): speed up the host speed of layer_norm
GitOrigin-RevId: 6f359b5b29
|
3 years ago |
Megvii Engine Team
|
67cfce9f2f
|
fix(imperative/amp): add is_scalar check in elemwise and concat
GitOrigin-RevId: 61a612e92a
|
3 years ago |
Megvii Engine Team
|
d313f92610
|
fix(imperative/amp): fix format transformation for symbol trans
GitOrigin-RevId: 96cc237c67
|
3 years ago |
Megvii Engine Team
|
261a5bce23
|
feat(imperative/amp): add dimshuffle in set_format for nhwc
GitOrigin-RevId: 5ced9e1a31
|
3 years ago |
Megvii Engine Team
|
c9e56f4987
|
feat(imperative/amp): add dimshuffle before creating nhwc tensor
GitOrigin-RevId: 4461f9a0d3
|
3 years ago |
Megvii Engine Team
|
d57a071271
|
feat(imperative/amp): add fallback for op not supported for nhwc tensor
GitOrigin-RevId: 8411ce7bdc
|
3 years ago |
Megvii Engine Team
|
38a9aa9faf
|
feat(imperative/amp): add auto dimshuffle for elemwise and concat
GitOrigin-RevId: 6e3df4e064
|
3 years ago |
Megvii Engine Team
|
cd26376549
|
style(imperative/amp): reformat code
GitOrigin-RevId: 6e5a6e1eaf
|
3 years ago |
Megvii Engine Team
|
3892aa0b6e
|
fix(imperative/amp): fix bn params for nhwc amp
GitOrigin-RevId: 57a3b9d418
|
3 years ago |
Megvii Engine Team
|
6f0b582064
|
chore(imperative/amp): adapt dev
GitOrigin-RevId: 41eb0faadf
|
3 years ago |
Megvii Engine Team
|
ee984e8608
|
fix(imperative/amp): fix distributed backward callback for nhwc amp
GitOrigin-RevId: 4d725b0ea4
|
3 years ago |
Megvii Engine Team
|
15c6da6218
|
feat(imperative/amp): add nhwc support for adaptive pooling
GitOrigin-RevId: 7c5755308e
|
3 years ago |
Megvii Engine Team
|
c28a875fac
|
fix(imperative/amp): adapt new transformation
GitOrigin-RevId: 6edd577a70
|
3 years ago |
Megvii Engine Team
|
fd41302cc1
|
feat(imperative/amp): add set_format
GitOrigin-RevId: 91de6f49de
|
3 years ago |
Megvii Engine Team
|
fc633ce4ff
|
fix(imperative/amp): fix custom grad in Subgraph
GitOrigin-RevId: 1c728d6ab9
|
3 years ago |
Megvii Engine Team
|
673b295d75
|
feat(imperative/amp): remove conv_format and bn param_dim configs
GitOrigin-RevId: 848d34f63d
|
3 years ago |
Megvii Engine Team
|
7e9aa742e6
|
feat(imperative/amp): enable auto_convert_format by default
GitOrigin-RevId: 71ae311fed
|
3 years ago |
Megvii Engine Team
|
fc0f454685
|
fix(dnn/check_non_finite): adjust some details of CheckNonFinite
GitOrigin-RevId: 52ddd805b4
|
3 years ago |
Megvii Engine Team
|
3bd40887b6
|
feat(mgb/opr): add NHWC support for AdaptivePooling
GitOrigin-RevId: b23e37ac23
|
3 years ago |
Megvii Engine Team
|
e393d1cf65
|
feat(mge/amp): add convert_format module for NHWC training
GitOrigin-RevId: 1b41e1042c
|
3 years ago |
Megvii Engine Team
|
533fb5bf49
|
feat(imperative): support formatted tensor and add special op rules
GitOrigin-RevId: 77ff909f23
|
3 years ago |
Megvii Engine Team
|
4aa79c453b
|
perf(mge): override grad of matmul
GitOrigin-RevId: d9d97e70fe
|
3 years ago |
Megvii Engine Team
|
98b5ee78c1
|
feat(mge/dnn): add lamb optimizer
GitOrigin-RevId: 5a27157456
|
3 years ago |
Megvii Engine Team
|
a926878c01
|
feat(imperative): remove symbolvar of imperative
GitOrigin-RevId: 16da6d1491
|
3 years ago |
Megvii Engine Team
|
14813d13c0
|
fix(whl): fix whl broken: patchelf on big (> 4G) file will
make elf section broken, as a workaround, do strip firstly,
then do patchelf.
GitOrigin-RevId: c7fb7e25a6
|
3 years ago |
Megvii Engine Team
|
9e0583e13a
|
feat(dnn/arm_common): add arm_common chanwise dot 11x11
GitOrigin-RevId: 84e0815a59
|
3 years ago |
Megvii Engine Team
|
115bcbce2b
|
feat(lite): add fitting mode for load and run
GitOrigin-RevId: 8f21fda9d3
|
3 years ago |
Megvii Engine Team
|
02bfb8f8b9
|
feat(lite): add and fix some feature for load and run fitting mode
GitOrigin-RevId: bbddc9bb79
|
3 years ago |
Megvii Engine Team
|
c62ddba238
|
feat(dnn/opencl): optimize heuristic rule
GitOrigin-RevId: 971c93d926
|
3 years ago |