Megvii Engine Team
|
b078dda90b
|
feat(mge/random): add some random op and remove random/distrbution.py
GitOrigin-RevId: 4c05ebc266
|
4 years ago |
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
|
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
|
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
|
a5060a2bfe
|
feat(mgb/opr): add check_has_inf kernel and opr
GitOrigin-RevId: 0d042dbfce
|
4 years ago |
Megvii Engine Team
|
c64b1c94cd
|
feat(imperative/functional): add roll in functional
GitOrigin-RevId: ff630f1fe7
|
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
|
54a4d70eb5
|
feat(src/serialization): add support of serializing metadata
GitOrigin-RevId: b563c94451
|
4 years ago |
Megvii Engine Team
|
b7e596b4a1
|
perf(autograd): copy inputs before capture in backward_graph_grad_rule
GitOrigin-RevId: 8b9c067b2d
|
4 years ago |
Megvii Engine Team
|
2ac3c9dc8b
|
fix(trace): constants in backward graph treat as ImmutableTensor corectlly
GitOrigin-RevId: 5fd6a5e00c
|
4 years ago |
Megvii Engine Team
|
62bd6c823b
|
feat(cmake/debug): misc for build
* add asan build option
* fix cpuinfo build opt level
* fix host release build with out debug info
* opt "fix lite bazel/cmake symbols MR"
* other misc build opt
GitOrigin-RevId: 6ca286e195
|
4 years ago |
Megvii Engine Team
|
3e4e4c4604
|
feat(mgb/jit): add graph_opt_config and jit_config interfaces
GitOrigin-RevId: 170d9eeab2
|
4 years ago |
Megvii Engine Team
|
34262d904c
|
fix(imperative): fix the size of blob with offset
GitOrigin-RevId: bb1736dfe1
|
4 years ago |
Megvii Engine Team
|
787f187e7d
|
fix(imperative/src): fix dot backward error
GitOrigin-RevId: 02ba44a0e6
|
4 years ago |
tpoisonooo
|
7038a7f5d0
|
fix(quant): fix spell error
|
4 years ago |
Megvii Engine Team
|
355153e158
|
feat(mge/dtr): add DTR in computing graph
GitOrigin-RevId: 8941810319
|
4 years ago |
Megvii Engine Team
|
0ac642b5d5
|
fix(imperative): persistent cache write through on put
GitOrigin-RevId: f9408ae504
|
4 years ago |
Megvii Engine Team
|
29f7cdb84a
|
fix(mgb/opr): correct nvof out shape computation
GitOrigin-RevId: 16bf086e92
|
4 years ago |
Megvii Engine Team
|
71cc814eaf
|
feat(ci): add aarch64 linux ci
GitOrigin-RevId: 2c0d3a8cc2
|
4 years ago |
Megvii Engine Team
|
d8ee0d7b5c
|
fix(mge/distributed): fix the mutli dataloader test error
GitOrigin-RevId: 86c8925916
|
4 years ago |
Megvii Engine Team
|
e275dfeca1
|
feat(imperative/python): support pooling mode "average" for avg pool2d module
GitOrigin-RevId: 9fe442129f
|
4 years ago |
Megvii Engine Team
|
01354337a9
|
fix(mge/autodiff): fix incorrect handling of tuple dy
GitOrigin-RevId: beca8e3711
|
4 years ago |
Megvii Engine Team
|
86b69cacd0
|
fix(dnn): fixes for int4
GitOrigin-RevId: 845e164fd3
|
4 years ago |
Megvii Engine Team
|
d2673c5abf
|
fix(ci/windows): add windows cuda test
GitOrigin-RevId: 706be83032
|
4 years ago |
Megvii Engine Team
|
2d6827c168
|
fix(mgb/windows): temporary workround on cuda-windows python exit
code(127), as windows cuda driver unloading before atexit function
may remove this after upgrade cuda runtime
GitOrigin-RevId: cac37ca3dd
|
4 years ago |
Megvii Engine Team
|
241b35a697
|
refactor(ops): remove BackwardGraph op
GitOrigin-RevId: eda20e5760
|
4 years ago |