Megvii Engine Team
|
1f0cc891b0
|
feat(dnn): enable eye to support bool
GitOrigin-RevId: 76d874d5b7
|
4 years ago |
Megvii Engine Team
|
c85631aa77
|
feat(dnn): use ref ptr interface for all backends
GitOrigin-RevId: f65feae5cc
|
4 years ago |
Megvii Engine Team
|
68cdabd288
|
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
|
4 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
4 years ago |
Megvii Engine Team
|
bfb30dcb81
|
chore(format): fix compile bugs after code format
GitOrigin-RevId: 11a4b06f6f
|
4 years ago |
Megvii Engine Team
|
b3e54eade1
|
feat(dnn/bn): use new cudnn BN kernel to support NHWC
GitOrigin-RevId: 9d80f2009d
|
4 years ago |
Megvii Engine Team
|
d69b59035d
|
feat(dnn): add an get_all_algorithms_safe interface
GitOrigin-RevId: e3734e4531
|
4 years ago |
Megvii Engine Team
|
103d7f33ba
|
refactor(dnn/rocm): update hip license header
GitOrigin-RevId: 79d684755d
|
5 years ago |
Megvii Engine Team
|
5aa52d3863
|
feat(dnn/rocm): add adaptive pooling opr
GitOrigin-RevId: e844b3e770
|
4 years ago |
Megvii Engine Team
|
83cf4ee64e
|
refactor(dnn/rocm): remove some useless includes
GitOrigin-RevId: 3d2c315a36
|
4 years ago |
Megvii Engine Team
|
323a4642e6
|
feat(dnn/rocm): add topk opr
GitOrigin-RevId: 5ecb079854
|
4 years ago |
Megvii Engine Team
|
f4784f4af1
|
feat(dnn/rocm): add argsort opr
GitOrigin-RevId: b4c3eb4707
|
4 years ago |
Megvii Engine Team
|
6082c353e7
|
feat(dnn/rocm): support bool in type_cvt and elemwise
GitOrigin-RevId: ad5ec7bc1c
|
5 years ago |
Megvii Engine Team
|
694aa1bd92
|
feat(dnn): add heuristic cache
GitOrigin-RevId: 35e942b5e3
|
4 years ago |
Megvii Engine Team
|
3abe0b2462
|
fix(mgb): fix rocm pooling
GitOrigin-RevId: 44876d398e
|
4 years ago |
Megvii Engine Team
|
1af350c6d2
|
feat(dnn): add fill kernel
GitOrigin-RevId: d2cee3a7a0
|
5 years ago |
Megvii Engine Team
|
f76a2cc2c6
|
feat(mge/opr): add silu and gelu
GitOrigin-RevId: 75aa42947e
|
5 years ago |
Megvii Engine Team
|
25932352e9
|
refactor(mgb/dnn): rocm pooling rebase algochooser
GitOrigin-RevId: 95be929841
|
5 years ago |
Megvii Engine Team
|
a5060a2bfe
|
feat(mgb/opr): add check_has_inf kernel and opr
GitOrigin-RevId: 0d042dbfce
|
5 years ago |
Megvii Engine Team
|
4b141f8de4
|
fix(mgb): add usable-depend-on-shape attr
GitOrigin-RevId: 3a14fa6b6f
|
5 years ago |
Megvii Engine Team
|
69a146c8c3
|
build(rocm): support rocm-3.9
GitOrigin-RevId: 85f8911736
|
5 years ago |
Megvii Engine Team
|
a31b7c6ee3
|
build(rocm): partially support hcc compilation
GitOrigin-RevId: ca9f1f8e8e
|
5 years ago |
Megvii Engine Team
|
621ae0a1e8
|
fix(dnn): replace kernel launch syntax with macro for hcc
GitOrigin-RevId: f9e69d4825
|
5 years ago |
Megvii Engine Team
|
78fff72a95
|
feat(dnn): add param_pack for rocm
GitOrigin-RevId: 2180504c71
|
5 years ago |
Megvii Engine Team
|
c338e876ec
|
refactor(mgb/dnn): add negative attribute for algo
GitOrigin-RevId: 88b1ce94a5
|
5 years ago |
Megvii Engine Team
|
ec1a99acc2
|
refactor(mgb/dnn): replace reproducible with attribute
GitOrigin-RevId: d49015714c
|
5 years ago |
Megvii Engine Team
|
55974e8cf9
|
feat(log): opt log
* opt log at release mode
* add MGE_OVERRIDE_LOG_LEVEL for runtime debug
//! env to config LogLevel
//! DEBUG = 0, INFO = 1, WARN = 2, ERROR = 3, NO_LOG = 4
//! for example , export MGE_OVERRIDE_LOG_LEVEL=0, means set LogLevel to DEBUG
GitOrigin-RevId: 16cd674c56
|
5 years ago |
Megvii Engine Team
|
c33a717314
|
feat(dnn): repalce is_reproducible with algo attribute in opencl, cpu, rocm and cuda
GitOrigin-RevId: 86dead0a11
|
5 years ago |
Megvii Engine Team
|
85fa988348
|
refactor(dnn): add get_algorithm_from_desc interface
GitOrigin-RevId: 6d211ca167
|
5 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
5 years ago |
Megvii Engine Team
|
409a877267
|
feat(dnn): add algo interface for rocm&fallback matmul and batched matrix mul
GitOrigin-RevId: dea03a0f7a
|
5 years ago |
Megvii Engine Team
|
a1877ee0fa
|
refactor(dnn): refactor algo interface, use algoinfo instead of global algorithm
GitOrigin-RevId: 479718ac75
|
5 years ago |
Megvii Engine Team
|
f7b2bdae1a
|
refactor(dnn): refactor algorithm type interface
GitOrigin-RevId: 843d885f82
|
5 years ago |
Megvii Engine Team
|
89303cd829
|
feat(megdnn/rocm): add bn for rocm backend
GitOrigin-RevId: 8bd49599b2
|
5 years ago |
Megvii Engine Team
|
aea829c9fa
|
feat(megdnn/rocm): add average inclusive mode for pooling
GitOrigin-RevId: 08162bd1fb
|
5 years ago |
Megvii Engine Team
|
cc952b2b92
|
fix(rocm): fix rocm megdnntest sleep and a cut code
GitOrigin-RevId: 26de5ca98b
|
5 years ago |
Megvii Engine Team
|
7f857bd471
|
feat(mgb/rocm): add cmake for rocm and fix compile errors and bn
GitOrigin-RevId: c73ed4adc3
|
5 years ago |
Megvii Engine Team
|
199eefbd4c
|
fix(dnn): generate mode files
GitOrigin-RevId: 9b1e840f00
|
5 years ago |
Megvii Engine Team
|
9510136223
|
fix(mgb/rocm): remove begin-internal of rocm
GitOrigin-RevId: 1523833fcb
|
5 years ago |