Megvii Engine Team
a398d4b5a5
chore(mge/functional): remove duplicated code
GitOrigin-RevId: f9efea46cb
4 years ago
Megvii Engine Team
92afcf3eb4
refactor(mge/imperative): move cgtools to megengine.utils and add load_and_inference in cgtools
GitOrigin-RevId: abfee3d4fa
4 years ago
Megvii Engine Team
1965276845
fix(mge): fix bool.sum()
GitOrigin-RevId: 62c482db40
4 years ago
Megvii Engine Team
fc74d5f2c3
feat(mge/dump): add more options for megbrain_graph.dump_graph
GitOrigin-RevId: 848e3c87c1
4 years ago
Megvii Engine Team
81b6a73382
fix(mge): fix bug of tensor.T
GitOrigin-RevId: 9fe9347b00
4 years ago
Megvii Engine Team
fc4c9292be
fix(mge/imperative): add missed __init__.py
GitOrigin-RevId: 0e863c8066
4 years ago
Megvii Engine Team
b111baf1b6
fix(mge/tensor): fix const target shape in reshape
GitOrigin-RevId: 7c04a9efba
4 years ago
Megvii Engine Team
9005cf74d2
fix(mge/imperative): fix tracer leak
GitOrigin-RevId: 1a8ac20b46
4 years ago
Megvii Engine Team
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
4 years ago
Megvii Engine Team
91efd67dbf
refactor(mge/jit): change dump options, add test
GitOrigin-RevId: fbc0d51c2b
4 years ago
Megvii Engine Team
099ffeac12
chore(mge): typo and reword
GitOrigin-RevId: 3848da4824
4 years ago
Megvii Engine Team
ac11c38a97
feat(mge/imperative): add graph load and cgtools for imperative
GitOrigin-RevId: ba251f452a
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
478f2c51dc
feat(mge/jit): add trace/dump options
GitOrigin-RevId: 0f43c14599
4 years ago
Megvii Engine Team
e9104ef157
fix(mge/parampack): fix copy stream, import cycle
GitOrigin-RevId: 673e11c5b6
4 years ago
Megvii Engine Team
e283663a02
fix(mge/imperative): update tests to new optimizer api
GitOrigin-RevId: 3d06e3db3c
4 years ago
Megvii Engine Team
9faa32fcb7
fix(mge/imperative): fix grad callback
GitOrigin-RevId: 6f843b0106
4 years ago
Megvii Engine Team
6070266766
refactor(mge/grad_manager): refactor gradmanager, add allreduce callback
GitOrigin-RevId: 086e2871e8
4 years ago
Megvii Engine Team
67859f04e1
fix(imperative): add __array__ and __array_wrap__ for tensorwrapper
GitOrigin-RevId: 87d4ab6c8e
4 years ago
Megvii Engine Team
5aa19f3d4a
test(mge/imperative): add more testcases for function
GitOrigin-RevId: e0675994f1
4 years ago
Megvii Engine Team
fc3eca845b
feat(mge/imperative/jit): add dump input shape and xornet example
GitOrigin-RevId: 5e2acd4052
4 years ago
Megvii Engine Team
1b2194cdce
feat(whl/internal): put imperative rt to megbrain
GitOrigin-RevId: f3b6e492d7
4 years ago
Megvii Engine Team
76dbaa2741
feat(mge/imperative): add name, make_h2d, dump_graph to graph runtime
GitOrigin-RevId: b8681a31a8
4 years ago
Megvii Engine Team
e6e29748b9
chore(mge/imperative): fix Graph.make_const
GitOrigin-RevId: 0f4c62aebf
4 years ago
Megvii Engine Team
ac3408bfaa
chore(mge): add VarNode.value
GitOrigin-RevId: 1dc0d0c711
4 years ago
Megvii Engine Team
4101d5bca0
refactor(mge/imperative): add BackwardGraph.interpret
GitOrigin-RevId: bb3a59380e
4 years ago
Megvii Engine Team
d4bad71190
feat(mge): add jit.trace
GitOrigin-RevId: ec647324c0
4 years ago
Megvii Engine Team
332959003f
chore(mge): add multipledispatch license
GitOrigin-RevId: 502c53324c
4 years ago
Megvii Engine Team
5474538f39
refactor(mge/imperative): fork multipledispatch
GitOrigin-RevId: a7c25a4302
4 years ago
Megvii Engine Team
ad9ac521ff
refactor(mge/imperative): remove abandoned code
GitOrigin-RevId: 0178bb5684
4 years ago
Megvii Engine Team
46f03b9f7c
perf(mge/imperative): optimize gpu memory with standalone grad_fn
GitOrigin-RevId: 44952b235e
4 years ago
Megvii Engine Team
7fadc16d3c
refactor(mge/functional): support tensor shape in interpolate and split
GitOrigin-RevId: 6430b64f01
4 years ago
Megvii Engine Team
7e8f720953
refactor(mge/tensor): tensor reduce supports keepdims
GitOrigin-RevId: 8ed95e0fb8
4 years ago
Megvii Engine Team
40d18c8920
fix(mge/imperative): fix tests when shape is tensor
GitOrigin-RevId: fd0095c1ec
4 years ago
Megvii Engine Team
92f7cceb11
fix(mge/functional): fix cond_take when index is numpy array
GitOrigin-RevId: 5fb93740f4
4 years ago
Megvii Engine Team
56cb5d6a16
fix(mge/functional): int operations(div, exp, pow)
GitOrigin-RevId: dc434fa7ec
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