Megvii Engine Team
11d75fecb5
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
3 years ago
Megvii Engine Team
8a222d2c9c
fix(mge): replace functional paramter
GitOrigin-RevId: af3c68c588
3 years ago
Megvii Engine Team
f5cb21ed3a
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
d063d5774f
perf(functional): use fma to reduce elemwise but disable subgraph compilation
GitOrigin-RevId: c75a6e1a09
3 years ago
Megvii Engine Team
2a063f8e87
fix(subgraph): fix scope mismatch of subgraph content
GitOrigin-RevId: 6e23456250
3 years ago
Megvii Engine Team
3206af9db2
perf(functional/matmul): reimplement matmul with subgraph
GitOrigin-RevId: 456b2a51d3
3 years ago
Megvii Engine Team
070c811732
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
3 years ago
Megvii Engine Team
1040b77843
fix(mge/functional): fix F.topk(kth_only=True)
GitOrigin-RevId: ddecd1d14b
4 years ago
Megvii Engine Team
1bf18252c4
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
4 years ago
Megvii Engine Team
a5060a2bfe
feat(mgb/opr): add check_has_inf kernel and opr
GitOrigin-RevId: 0d042dbfce
4 years ago
Megvii Engine Team
984d85ca26
feat(mge/functional): argmin and argmax support negtive axis
GitOrigin-RevId: a1bd1102a6
4 years ago
Megvii Engine Team
65af9cfcdc
refactor(mge): use lower case for default string parameters in functional and module
GitOrigin-RevId: dbc1f27ff7
4 years ago
Megvii Engine Team
585aa561f3
fix(functional): replace usage of deprecated get_conv_execution_strategy()
GitOrigin-RevId: 61ef7a50ff
4 years ago
Megvii Engine Team
dcfb6a537e
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
4 years ago
Megvii Engine Team
98d6ba7e2f
refactor(mge/functional): move matinv from nn to math
GitOrigin-RevId: 75c48ce327
4 years ago
Megvii Engine Team
05692d058f
refactor(functional): move matmul dot svd to math pkg
GitOrigin-RevId: 15eb08bacb
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
87f4b46e3f
perf(mge/imperative): move convert_inputs from python to C++
GitOrigin-RevId: baef3d348c
4 years ago
Megvii Engine Team
c294b9d18b
refactor(mge/tensor): remove old implementation
remove core.tensor, raw_tensor,TensorWrapper
avoid create tensor with zero-stride numpy ndarray
GitOrigin-RevId: 4fe5c4c5ba
4 years ago
Megvii Engine Team
147cef52a0
refactor(mge/imperative): implement new tensor system
GitOrigin-RevId: 2dd4e460ac
4 years ago
Megvii Engine Team
69e3e32240
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
4 years ago
Megvii Engine Team
638ab52fdc
feat(mge/imperative): simulates scalar
GitOrigin-RevId: e81630e256
4 years ago
Megvii Engine Team
94dba16ff4
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
d502e79f85
feat(mge): make F.norm numpy compatible
GitOrigin-RevId: 4e5236171c
4 years ago
Megvii Engine Team
1e2117f66d
feat(mge): remove add_axis and remove_axis
GitOrigin-RevId: 59611d43f9
4 years ago
Megvii Engine Team
e034874ae7
feat(mge): rename clamp -> clip
GitOrigin-RevId: 2b594a611e
4 years ago
Megvii Engine Team
20cdd11d4d
docs(mge): fix docstring of cgtools
GitOrigin-RevId: ee8c93b9f7
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
Megvii Engine Team
f4860b9345
fix(mge/functional): fix trace topk
GitOrigin-RevId: c88ca8219b
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
4 years ago
Megvii Engine Team
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
4 years ago
Megvii Engine Team
10e942d952
refactor(mge): polish api
- refactor(mge): add support for optimizer.step().clear_grad() idiom
- refactor(mge): rename some methods of GradManager
- refactor(mge): remove tensor_nn and TensorDict
- refactor(mge): remove Buffer
- refactor(mge): remove requires_grad flag
- refactor(mge): add a default grad=None attribute to Tensor
- refactor(mge): deprecation for 1.0
GitOrigin-RevId: 3b723d9387
4 years ago
Megvii Engine Team
7e8f720953
refactor(mge/tensor): tensor reduce supports keepdims
GitOrigin-RevId: 8ed95e0fb8
4 years ago
Megvii Engine Team
6b380e8965
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
4 years ago