Megvii Engine Team
bfb30dcb81
chore(format): fix compile bugs after code format
GitOrigin-RevId: 11a4b06f6f
3 years ago
Megvii Engine Team
2108696588
feat(mge/imperative): add local response normalization
GitOrigin-RevId: 939a4d26dd
3 years ago
Megvii Engine Team
ca4c93dee7
fix(mge): fix layer norm amp bug
GitOrigin-RevId: dba691fcbf
3 years ago
Megvii Engine Team
c674bf0e6e
feat(debug): add gdb-scripts
GitOrigin-RevId: 43f9674f50
3 years ago
Megvii Engine Team
177dec94c5
feat(mgb/opr): add bgr2gray mode for cvtcolor opr
GitOrigin-RevId: d50415b236
3 years ago
Megvii Engine Team
000517c641
fix(grad): stop using exception in grad_override
GitOrigin-RevId: 00ae38d48b
3 years ago
Megvii Engine Team
d02b0a08f7
feat(lite): fix typo
GitOrigin-RevId: 8c46aa3a30
3 years ago
Megvii Engine Team
cebda6ff8d
feat(mge/imperative): add ctc loss
GitOrigin-RevId: e29854a98e
3 years ago
Megvii Engine Team
f5cb21ed3a
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
3 years ago
Megvii Engine Team
b6142bee9a
feat(imperative): support tensor with uint16 date type
GitOrigin-RevId: 57ba0633c7
3 years ago
Megvii Engine Team
4e95c13617
feat(sgd): sgd supports nesterov momentum
GitOrigin-RevId: 13eda179da
3 years ago
Megvii Engine Team
dc28a43409
fix(mgb/bn): fix empty tensor input problem and other minor problems
GitOrigin-RevId: 1708b76cb8
3 years ago
Megvii Engine Team
270f1aa28f
feat(mgb/serialization): add Accessor for OprLoader to fix BN output compatibility
GitOrigin-RevId: 3b95da02c8
3 years ago
Megvii Engine Team
c0ccd0ea7e
feat(mge/bn): add NHWC support for bn
GitOrigin-RevId: 0a5bb6f72d
3 years ago
chenjiahui
6b863cc53a
feat(imperative): add pixel_shuffle opr
Closes #211
GitOrigin-RevId: 1846714791
3 years ago
Megvii Engine Team
b982be567a
feat(mge/imperative): add permutation support for the tensor
GitOrigin-RevId: 7ed0447bfe
3 years ago
Megvii Engine Team
3977b7aa0b
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
3 years ago
Megvii Engine Team
884a07ffd2
fix(test/random): set a random seed for random unit test
GitOrigin-RevId: ad4b01eac7
3 years ago
Megvii Engine Team
3f0bb47a43
ci(imperative): add test requires
GitOrigin-RevId: d7a945caf3
3 years ago
Megvii Engine Team
11a383bc79
feat(mgb): add tensorboard tool c++ layer interface and python parsing
GitOrigin-RevId: ae98bc887f
3 years ago
Megvii Engine Team
8084e4e2ee
feat(mgb): add tensorboard tool python layer interface
GitOrigin-RevId: 065bc4d153
3 years ago
Megvii Engine Team
97b1b7774d
feat(mgb): add megbrain layer norm opr with subgraph
GitOrigin-RevId: 9b7fa821f8
3 years ago
Megvii Engine Team
65d554ed8e
refactor(dtr): improve dtr in computing graph
GitOrigin-RevId: d599a08e52
3 years ago
Megvii Engine Team
f642b05e17
test(mge): update traced_module unit test
GitOrigin-RevId: 3948d50d79
3 years ago
Megvii Engine Team
fb20cb36a0
docs(mge/traced_module): update traced_module api doc
GitOrigin-RevId: 19a95d26c7
3 years ago
Megvii Engine Team
c7a8d945c7
fix(mge/traced_module): let graph record total_id
GitOrigin-RevId: f99178f3ac
3 years ago
Megvii Engine Team
4de62ad6cd
docs(mge/device): fix docstring of get_cuda_compute_capability
GitOrigin-RevId: 4ec04e8b9b
3 years ago
Megvii Engine Team
fb49a2834f
refactor(mgb/dnn): refactor enum used in serializing
GitOrigin-RevId: e57af4a59c
3 years ago
Megvii Engine Team
6082c353e7
feat(dnn/rocm): support bool in type_cvt and elemwise
GitOrigin-RevId: ad5ec7bc1c
4 years ago
Megvii Engine Team
93f70a9539
feat(src): add MGB_CUSTOM_OP macro
GitOrigin-RevId: beddcf9745
3 years ago
Megvii Engine Team
b3e958d0bc
fix(src): fix the warnings and copy.bara.sky in custom op
GitOrigin-RevId: 4ade455897
3 years ago
Megvii Engine Team
cdb692d2fa
refactor(imperative): add TODO tag for some functions
GitOrigin-RevId: e295a1fa55
3 years ago
Megvii Engine Team
90dd07161c
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
3 years ago
Megvii Engine Team
cbf024bf20
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
3 years ago
Megvii Engine Team
43fb2c4afb
feat(opr): let roll support empty IO
GitOrigin-RevId: b9a59b623a
3 years ago
Megvii Engine Team
b2827cb1b1
feat(opr): let Dot, MatrixMul and BatchedMatrixMul support empty input
GitOrigin-RevId: 10a3c5b106
3 years ago
Megvii Engine Team
50f738772b
feat(opr): let Copy and Identity support empty IO
GitOrigin-RevId: 4e49d8eae8
3 years ago
Megvii Engine Team
74cbc10d82
feat(opr): let batchnorm support empty IO
GitOrigin-RevId: 219411c80c
3 years ago
Megvii Engine Team
8ae61752a0
feat(opr): let nms support empty IO
GitOrigin-RevId: 4c51b1aedb
3 years ago
Megvii Engine Team
1a1748daf1
feat(opr): let Argsort support empty IO
GitOrigin-RevId: 05fcac6e47
3 years ago
Megvii Engine Team
7234efe1e7
feat(opr): let random op support empty output
GitOrigin-RevId: a2174975aa
3 years ago
Megvii Engine Team
3bc94738ba
fix(profiler): do not assume clock duration and time_point type
GitOrigin-RevId: 2015bc30bd
3 years ago
Megvii Engine Team
b7e6bd7bba
feat(profiler): custom event support device
GitOrigin-RevId: 4709c44b39
3 years ago
Megvii Engine Team
defbc20efb
fix(profiler): fix profiler in dtr
GitOrigin-RevId: 79b5a6b526
3 years ago
Megvii Engine Team
d4c71f92dc
perf(profiler): reduce profiler host overhead
GitOrigin-RevId: 92cea560f5
3 years ago
Megvii Engine Team
8fddd8089d
fix(profiler): respect record_device option
GitOrigin-RevId: 7c9a8cfba7
4 years ago
Megvii Engine Team
59d5976622
fix(profiler): collect records when thread exit
GitOrigin-RevId: fb796dd61c
4 years ago
Megvii Engine Team
a841d8e76f
refactor(profiler): refactor event processing
GitOrigin-RevId: 26f0c5a6e1
4 years ago
Megvii Engine Team
dff7719e87
feat(mge/distributed): add preload host data with op
fix(mge/distributed): change api name with preload
fix(mge/distributed): fix recursive model in preload tensor
fix(mge/distributed): fix recursive when cache contain None
GitOrigin-RevId: 80e2a6dd70
3 years ago
Megvii Engine Team
ca4374b6e2
perf(subgraph): remove unnecessary sync
GitOrigin-RevId: 1546286706
3 years ago