Megvii Engine Team
|
f30c0e06a6
|
feat(mgb/opr): add lsq opr
GitOrigin-RevId: 45494a2b57
|
4 years ago |
Megvii Engine Team
|
eb66681f45
|
fix(mge/random): fix delete_rng_handel
GitOrigin-RevId: a836e13c23
|
3 years ago |
Megvii Engine Team
|
d6db4fea93
|
fix(mge/module): set no_cache=true when loading state dict
GitOrigin-RevId: 83281a3d47
|
4 years ago |
Megvii Engine Team
|
fea1bba23e
|
fix(mge/tools): fix module status error
GitOrigin-RevId: 38e3eeb34b
|
3 years ago |
Megvii Engine Team
|
6cd01d5a74
|
feat(imperative/functional): let elemwise support empty IO & add some tests
GitOrigin-RevId: a5dc3b997c
|
4 years ago |
Megvii Engine Team
|
dea5278172
|
feat(mgb/opr): let PowC & TypeCvt support empty IO
GitOrigin-RevId: f97b3005fd
|
4 years ago |
Megvii Engine Team
|
2f68aeb9b6
|
feat(imperative/jit): let trace support empty IO
GitOrigin-RevId: 97a55242bf
|
4 years ago |
Megvii Engine Team
|
809d5056cd
|
feat(mge/distributed): enable pt shm allreduce
GitOrigin-RevId: 1dd5a02a51
|
4 years ago |
Megvii Engine Team
|
0245594156
|
test(autograd): test jvp emulated by 2nd grad
GitOrigin-RevId: 47114fcd99
|
3 years ago |
Megvii Engine Team
|
8480302da8
|
fix(autograd): make higher order grad experimental
GitOrigin-RevId: 81e1eb0ebf
|
4 years ago |
Megvii Engine Team
|
72531f2b4a
|
test(autograd): add more tests for higher order grad
GitOrigin-RevId: 5fc308f87a
|
4 years ago |
Megvii Engine Team
|
522e556b83
|
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
|
4 years ago |
Megvii Engine Team
|
5198b7834c
|
fix(mge/functional): fix expand_dims for scalar
GitOrigin-RevId: 253ea608f7
|
4 years ago |
Megvii Engine Team
|
88898e63a5
|
fix(mgb): replace if_constexpr with runtime function to avoid potential
bug
GitOrigin-RevId: 27fe093d50
|
4 years ago |
Megvii Engine Team
|
25932352e9
|
refactor(mgb/dnn): rocm pooling rebase algochooser
GitOrigin-RevId: 95be929841
|
4 years ago |
Megvii Engine Team
|
1cfdbc565c
|
feat(dnn): add deterministic max pooling
GitOrigin-RevId: 9ab4c7a748
|
4 years ago |
Megvii Engine Team
|
20ab82d00c
|
fix(tee): fix tee crash
GitOrigin-RevId: 379f970c87
|
4 years ago |
Megvii Engine Team
|
933dd9a497
|
feat(mge/distributed): add cuda env check before forked thread
style(core/comp_node): reformat code
GitOrigin-RevId: 372452a8eb
|
4 years ago |
Megvii Engine Team
|
2a54196117
|
fix(tee): fix tee link
GitOrigin-RevId: db7b98524d
|
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
|
3597a6dbd7
|
feat(dnn/arm): nchw_nchw44 conv support 1x1s1
GitOrigin-RevId: 8c8f7d7c76
|
4 years ago |
Megvii Engine Team
|
c64b1c94cd
|
feat(imperative/functional): add roll in functional
GitOrigin-RevId: ff630f1fe7
|
4 years ago |
Megvii Engine Team
|
d915c5a3fd
|
refactor(mgb): make convolution3D handle noncontiguous tensors
GitOrigin-RevId: 3d3c31b021
|
4 years ago |
Megvii Engine Team
|
d04cd67faf
|
refactor(mgb): make conv-backward-filter handle noncontiguous tensors
GitOrigin-RevId: 44c586f912
|
4 years ago |
Megvii Engine Team
|
44376f702a
|
refactor(mgb): make conv-backward-data handle noncontiguous tensors
GitOrigin-RevId: 0a8f66f9d3
|
4 years ago |
Megvii Engine Team
|
7b2a76d1ee
|
refactor(mgb): make conv handle noncontiguous tensors
GitOrigin-RevId: 86282709b3
|
4 years ago |
Megvii Engine Team
|
ca2828ddcb
|
fix(dnn/x86): fix x86 int8 matmul ldc bug
GitOrigin-RevId: 2502f99000
|
4 years ago |
Megvii Engine Team
|
aa4e847692
|
fix(interpreter): release gil when interpreter blocking
GitOrigin-RevId: c48e9efa5b
|
4 years ago |
Megvii Engine Team
|
bd62a0a647
|
fix(interpreter): remove notice flag of produce_tensor
GitOrigin-RevId: ed65d0107f
|
4 years ago |
Megvii Engine Team
|
c78a7848a1
|
test(profiler): simple tests for profiler
GitOrigin-RevId: 92073ce357
|
4 years ago |
Megvii Engine Team
|
d6079db1f6
|
feat(profiler): add memory flow format for profiler
GitOrigin-RevId: f0eab26398
|
4 years ago |
Megvii Engine Team
|
4286bb9f51
|
feat(profiler): add launch script for profiler
GitOrigin-RevId: fdfb60b149
|
4 years ago |
Megvii Engine Team
|
9d47c3babd
|
feat(profiler): imperative profiler support tracing
GitOrigin-RevId: b247472feb
|
4 years ago |
Megvii Engine Team
|
cdcb46ba95
|
feat(profiler): add chrome timeline format for profiler
GitOrigin-RevId: e68071feca
|
4 years ago |
Megvii Engine Team
|
1d64792b41
|
refactor(profiler): detach profiler from interpreter
GitOrigin-RevId: f3954728d1
|
4 years ago |
Megvii Engine Team
|
f2027b8d17
|
refactor(interpreter): recompute with do_apply_op
GitOrigin-RevId: 150fb46a63
|
4 years ago |
Megvii Engine Team
|
5cb35c1bed
|
refactor(profiler): add state structs to replay recorded events
GitOrigin-RevId: 16a1f5d7ba
|
4 years ago |
Megvii Engine Team
|
84fc5c927d
|
refactor(profiler): remove dump logic for old profiler
GitOrigin-RevId: f9a3b6d7b9
|
4 years ago |
Megvii Engine Team
|
5b5a826166
|
refactor(profiler): use macro to simplify event recording/definition
GitOrigin-RevId: 4d9dcfd6c2
|
4 years ago |
Megvii Engine Team
|
1ce40b5bf7
|
refactor(interpreter): wrap accesses to channel/worker state
GitOrigin-RevId: 1d58f2c876
|
4 years ago |
Megvii Engine Team
|
90d3905799
|
feat(mge): add mge._exit
GitOrigin-RevId: e7f348a677
|
4 years ago |
Megvii Engine Team
|
846642a881
|
docs(misc): add some notes
GitOrigin-RevId: de243064c5
|
4 years ago |
Megvii Engine Team
|
fb8f1534ca
|
fix(cmake/debug): fix asan do not work at windows env
GitOrigin-RevId: 6638194fd7
|
4 years ago |
Megvii Engine Team
|
40085acbae
|
fix(mgb): remove unnecessary cudnn8 warning
GitOrigin-RevId: 04cf1bfca9
|
4 years ago |
Megvii Engine Team
|
54a4d70eb5
|
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
|
4 years ago |
Megvii Engine Team
|
721091faf0
|
fix(core): fix thread local is not supported in ios
GitOrigin-RevId: b7a6928f0b
|
4 years ago |
Megvii Engine Team
|
1d86528162
|
ci(config): change pip mirror to pypi cache proxy
GitOrigin-RevId: cfc00b7df1
|
4 years ago |
Megvii Engine Team
|
82b95a3aa2
|
fix(lar): fix build error
GitOrigin-RevId: 932f6ac4b3
|
4 years ago |
Megvii Engine Team
|
8cfe3c3937
|
fix(lar): update --verbose to --model-info
GitOrigin-RevId: 6d928ee46b
|
4 years ago |
Megvii Engine Team
|
2e5a473de1
|
feat(lar): support --verbose
GitOrigin-RevId: f690193b75
|
4 years ago |