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
|
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
|
4 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
|
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
|
1d64792b41
|
refactor(profiler): detach profiler from interpreter
GitOrigin-RevId: f3954728d1
|
4 years ago |
Megvii Engine Team
|
90d3905799
|
feat(mge): add mge._exit
GitOrigin-RevId: e7f348a677
|
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 |
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
|
241b35a697
|
refactor(ops): remove BackwardGraph op
GitOrigin-RevId: eda20e5760
|
4 years ago |
Megvii Engine Team
|
8a918717c0
|
feat(mgb): add megengine inference
GitOrigin-RevId: 6ffec6b418
|
4 years ago |
Megvii Engine Team
|
b06b589960
|
feat(mgb): get static graph memory info
GitOrigin-RevId: f31745f8df
|
4 years ago |
Megvii Engine Team
|
0cf4ff702e
|
perf(dist): add fastpath for bcast params
GitOrigin-RevId: aa40b3cd72
|
4 years ago |
Megvii Engine Team
|
5e54821845
|
docs(syncbn): complete syncbn document
GitOrigin-RevId: 9b8d546f0b
|
4 years ago |
Megvii Engine Team
|
e417798f9e
|
fix(mge): correct pytype when calling apply from python
GitOrigin-RevId: 6abfa06ada
|
4 years ago |
Megvii Engine Team
|
c4048519c6
|
chore(mge/version): change dev version to 1.5
GitOrigin-RevId: 4058da1f98
|
4 years ago |
Megvii Engine Team
|
3352fe7003
|
refactor(dtr): import dtr as submodule
GitOrigin-RevId: abecd0f176
|
4 years ago |
Megvii Engine Team
|
a7581986bc
|
perf(dist): speed up bcast_val
GitOrigin-RevId: 21c4123b09
|
4 years ago |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
4 years ago |
Megvii Engine Team
|
d07ef83aa8
|
feat(mge/tools): add bash_completion for user to hint script
BREAKING CHANGE:
fix(mge/tools): change help script for other shell
BREAKING CHANGE:
fix(mge/tools): fix double -h error
BREAKING CHANGE:
fix(mge/tools): add support for _ parameter
BREAKING CHANGE:
GitOrigin-RevId: 537c3a2aae
|
4 years ago |
Megvii Engine Team
|
efc7f04fd3
|
refactor(mge/jit): using static global_enable for apply ctx insted of global variable
GitOrigin-RevId: dd82b53faf
|
4 years ago |
Megvii Engine Team
|
750bde9513
|
refactor(mge/jit): remove is_compiled flag in cpp tensor
GitOrigin-RevId: 15f90af735
|
4 years ago |
Megvii Engine Team
|
4f240ec2d3
|
refactor(mge/jit): make trace return any kind of output
GitOrigin-RevId: fd1265c661
|
4 years ago |
Megvii Engine Team
|
c6b552cf84
|
refactor(mge/functional): remove dependence to trace in functional implementations
GitOrigin-RevId: 0b18479fcc
|
4 years ago |