Megvii Engine Team
715009b564
refactor(mge/api): remove external, dropout fix
GitOrigin-RevId: 5e6ff1a372
4 years ago
Megvii Engine Team
9005cf74d2
fix(mge/imperative): fix tracer leak
GitOrigin-RevId: 1a8ac20b46
4 years ago
Megvii Engine Team
70bc746c20
fix(mge): fix bug of identity()
GitOrigin-RevId: 4bfd3cafb5
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
4 years ago
Megvii Engine Team
91efd67dbf
refactor(mge/jit): change dump options, add test
GitOrigin-RevId: fbc0d51c2b
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
4 years ago
Megvii Engine Team
523ce65e0d
fix(mge/imperative): fix cgtools related tests
GitOrigin-RevId: 8f1eadb32e
4 years ago
Megvii Engine Team
dd39265e95
fix(mgb/dtype): enable TypeCvt for bool when trace(symbolic=True)
GitOrigin-RevId: 4e0fc63369
4 years ago
Megvii Engine Team
5c37a64e3f
feat(sdk/load_and_run): add dump with testcase for mge support imperative
GitOrigin-RevId: 17a0cf6293
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
9389a805d8
fix(mge/oprmm): fix grad for collective comm
GitOrigin-RevId: e55dc47a6e
4 years ago
Megvii Engine Team
478f2c51dc
feat(mge/jit): add trace/dump options
GitOrigin-RevId: 0f43c14599
4 years ago
Megvii Engine Team
44d0b5daf5
feat(imperative): enable to() to copy to device
GitOrigin-RevId: f9caf17d24
4 years ago
Megvii Engine Team
69a7c55f55
test(mge/function): fix test for new optimizer api
GitOrigin-RevId: 8ae7720fe6
4 years ago
Megvii Engine Team
66b6daf777
test(mge/optimizer): fix test for new optimizer api
GitOrigin-RevId: 482ee62652
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
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
939bf3b28d
fix(mge/imperative): impl hashable for nms and cond_take
GitOrigin-RevId: 56918db014
4 years ago
Megvii Engine Team
8dc23e0fdf
fix(mge/functional): fix indexing_one_hot and remote_recv
GitOrigin-RevId: 00bdfb502b
4 years ago
Megvii Engine Team
2beb65b19d
fix(mge/module): tensor shape will not work when constructing numpy array
GitOrigin-RevId: 5a0d705970
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
087ceb52ef
feat(mge/imperative): add more optimizer trace tests
GitOrigin-RevId: 4127de1d22
4 years ago
Megvii Engine Team
403a1e7b24
feat(mge/imperative): add graph profiler
GitOrigin-RevId: c73563f337
4 years ago
Megvii Engine Team
d06f248d90
fix(whl/mgb/imperative): fix symbols conflict runtime crash
GitOrigin-RevId: d60c4052c1
4 years ago
Megvii Engine Team
1b2194cdce
feat(whl/internal): put imperative rt to megbrain
GitOrigin-RevId: f3b6e492d7
4 years ago
Megvii Engine Team
a3b2232ba7
feat(mge): add trace.dump
GitOrigin-RevId: ea4c9d33c8
4 years ago
Megvii Engine Team
0e82b959a1
feat(mge/imperative): add sublinear options
GitOrigin-RevId: f0e917f716
4 years ago
Megvii Engine Team
afddefb677
feat(mge/imperative): add more trace test
GitOrigin-RevId: b02e420a8a
4 years ago
Megvii Engine Team
d4bad71190
feat(mge): add jit.trace
GitOrigin-RevId: ec647324c0
4 years ago
Megvii Engine Team
5474538f39
refactor(mge/imperative): fork multipledispatch
GitOrigin-RevId: a7c25a4302
4 years ago
Megvii Engine Team
03320a0548
fix(functional): change to user-friendly nms api and fix concat
GitOrigin-RevId: 76b4190270
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
a226f02e7a
fix(mge/imperative): fix syncbn in symbolic mode
GitOrigin-RevId: a9794318a7
4 years ago
Megvii Engine Team
b9df2db1c5
build(third_party): update megray commit id
GitOrigin-RevId: cc753f941b
4 years ago
Megvii Engine Team
0067dcf068
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
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
e1fba6ece7
test(mge/distributed): add get_device_count_by_fork to fix distributed test skip
GitOrigin-RevId: 9ffd8a6149
4 years ago
Megvii Engine Team
60076f4715
test(mge/parampack): add parampack related opr test
GitOrigin-RevId: 550c6b6443
4 years ago
Megvii Engine Team
7e8f720953
refactor(mge/tensor): tensor reduce supports keepdims
GitOrigin-RevId: 8ed95e0fb8
4 years ago
Megvii Engine Team
6e882c1a86
feat(whl/imperative): compat for build python whl imperative and legacy runtime
GitOrigin-RevId: 7f6629ae1f
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
f4927db2fb
feat(mge/functional): support where func
GitOrigin-RevId: 9df6421ebe
4 years ago
Megvii Engine Team
227ec04bd3
fix(mge/test): fix pytest failed issue
GitOrigin-RevId: 96df470406
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