Megvii Engine Team
fc4c9292be
fix(mge/imperative): add missed __init__.py
GitOrigin-RevId: 0e863c8066
5 years ago
Megvii Engine Team
895711868c
fix(tools): add license for gen_header_for_bin_reduce.py in order to open source
GitOrigin-RevId: f310055a3c
5 years ago
Megvii Engine Team
aac8de554e
fix(mge/parampacksplit): fix parampacksplit refcnt error
GitOrigin-RevId: c964465596
5 years ago
Megvii Engine Team
a7b9ece4a7
feat(mgb/comp_node): add set_prealloc_config
GitOrigin-RevId: e725e7efdd
5 years ago
Megvii Engine Team
066da0bff1
fix(mgb/comp_node): add lock_guard to sd.mtx & update warning message & add test
GitOrigin-RevId: ae0bf4b479
5 years ago
Megvii Engine Team
b111baf1b6
fix(mge/tensor): fix const target shape in reshape
GitOrigin-RevId: 7c04a9efba
5 years ago
Megvii Engine Team
4ab259f558
refactor(mge): add some deprecation and migration guides
GitOrigin-RevId: 4793adf02b
5 years ago
Megvii Engine Team
715009b564
refactor(mge/api): remove external, dropout fix
GitOrigin-RevId: 5e6ff1a372
5 years ago
Megvii Engine Team
9005cf74d2
fix(mge/imperative): fix tracer leak
GitOrigin-RevId: 1a8ac20b46
5 years ago
Megvii Engine Team
70bc746c20
fix(mge): fix bug of identity()
GitOrigin-RevId: 4bfd3cafb5
5 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
5 years ago
Megvii Engine Team
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
5 years ago
Megvii Engine Team
e6715910c5
docs(mge/imperative): add docstring for GradManager
GitOrigin-RevId: 4c326206b8
5 years ago
Megvii Engine Team
ce55fbf62b
ci(imperative): run imperative python test
5 years ago
Megvii Engine Team
60778d1071
chore(deps): add cutlass
5 years ago
Megvii Engine Team
b254bb15a4
chore(deps): update megray
5 years ago
Megvii Engine Team
95ed871717
chore(deps): add mlir dependency
5 years ago
Megvii Engine Team
1babd7aa34
chore(deps): update intel-mkl-dnn
5 years ago
Megvii Engine Team
0307598a80
fix(dnn): keep consistent limit between deduce and compute
GitOrigin-RevId: 8de5f17ced
5 years ago
Megvii Engine Team
354b478c1f
fix(opr): remove constant value infer for const SharedDeviceTensor
GitOrigin-RevId: 8fa47b35ad
5 years ago
Megvii Engine Team
78d7d400d1
feat(opr): add a constant flavor for SharedDeviceTensor
Also, add a CONSTANT value inference tag to outputs of
MultipleDeviceTensorHolder.
GitOrigin-RevId: 82a805ed5f
5 years ago
Megvii Engine Team
14e1a20599
fix(test): disable mocked tests on Windows
GitOrigin-RevId: 898cfdb09d
5 years ago
Megvii Engine Team
76c79d79a1
fix(mgb/opr): fix ConvBias not passing on prep_filter
GitOrigin-RevId: 0dc9f9d133
5 years ago
Megvii Engine Team
498f7b2ea7
fix(opr): fix compile on GCC6
GitOrigin-RevId: b59756c923
5 years ago
Megvii Engine Team
b9f3077d26
feat(opr): remove old weight preprocess adapter in fastrun
GitOrigin-RevId: f467c0db80
5 years ago
Megvii Engine Team
75eebb7c42
feat(opr): use weight preprocess feature of MegDNN
GitOrigin-RevId: 779041f8a8
5 years ago
Megvii Engine Team
66509a54b3
test(mgb): enable gmock
GitOrigin-RevId: 259b0e4fb3
5 years ago
Megvii Engine Team
fe3ee3cd54
refactor(opr): refactor OprArityTrait
GitOrigin-RevId: fa065cde4e
5 years ago
Megvii Engine Team
91efd67dbf
refactor(mge/jit): change dump options, add test
GitOrigin-RevId: fbc0d51c2b
5 years ago
Megvii Engine Team
099ffeac12
chore(mge): typo and reword
GitOrigin-RevId: 3848da4824
5 years ago
Megvii Engine Team
539aac4482
fix(dnn/cuda): fix sass loader
GitOrigin-RevId: f7dec65a2a
5 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
5 years ago
Megvii Engine Team
023d60aee4
fix(mgb/atlas): use aclmdlGetOutputSizeByIndex to create DataBuffer
GitOrigin-RevId: df7f1b225d
5 years ago
Megvii Engine Team
116eee5231
build(third_party): update megray
GitOrigin-RevId: da5e05f82b
5 years ago
Megvii Engine Team
e507228e74
feat(mge/examples): add distributed training examples using launcher
GitOrigin-RevId: 5db26f58eb
5 years ago
Megvii Engine Team
23437864f9
fix(mgb/jit): mlir doesn't support broadcast
GitOrigin-RevId: 08bfc4c34a
5 years ago
Megvii Engine Team
f87bba6806
feat(mgb/jit): add scalar support for mlir
GitOrigin-RevId: 27b1649c04
5 years ago
Megvii Engine Team
11b121a7d2
fix(mgb/jit): link libdevice.bc when generate nvvm ir
GitOrigin-RevId: 49289d65c4
5 years ago
Megvii Engine Team
aa7f28b835
fix(mgb/jit): fix gpu kernel args order
GitOrigin-RevId: a7f0e56747
5 years ago
Megvii Engine Team
9767ca8f19
feat(mgb/jit): refactor code and add more elemwise mode
GitOrigin-RevId: eb6bcadf54
5 years ago
Megvii Engine Team
f9109bed8b
feat(mge): add mlir cmake
GitOrigin-RevId: e86db34b18
5 years ago
Megvii Engine Team
a51d5b4c31
feat(mgb/jit): add mlir backend for cpu and cuda
GitOrigin-RevId: 814fed047e
5 years ago
Megvii Engine Team
523ce65e0d
fix(mge/imperative): fix cgtools related tests
GitOrigin-RevId: 8f1eadb32e
5 years ago
Megvii Engine Team
dd39265e95
fix(mgb/dtype): enable TypeCvt for bool when trace(symbolic=True)
GitOrigin-RevId: 4e0fc63369
5 years ago
Megvii Engine Team
5c37a64e3f
feat(sdk/load_and_run): add dump with testcase for mge support imperative
GitOrigin-RevId: 17a0cf6293
5 years ago
Megvii Engine Team
ac11c38a97
feat(mge/imperative): add graph load and cgtools for imperative
GitOrigin-RevId: ba251f452a
5 years ago
Megvii Engine Team
76f367962f
fix(mge/trace): fix op order in symbolic
GitOrigin-RevId: fbf081a199
5 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
5 years ago
Megvii Engine Team
9389a805d8
fix(mge/oprmm): fix grad for collective comm
GitOrigin-RevId: e55dc47a6e
5 years ago
Megvii Engine Team
6f581906a1
refactor(mge/profiler): refactor profiler
GitOrigin-RevId: 279aa779a6
5 years ago