Megvii Engine Team
|
8a73193c2d
|
feat(dtr): remove eviction threshold
GitOrigin-RevId: 35c2014bf3
|
3 years ago |
Megvii Engine Team
|
69d1fd0f32
|
refactor(opdef): split apply_on_physical_tensor into infer_output_mem_desc and execute
GitOrigin-RevId: 4d62b7cbbd
|
3 years ago |
Megvii Engine Team
|
43098fb8f1
|
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
|
4 years ago |
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
|
2f68aeb9b6
|
feat(imperative/jit): let trace support empty IO
GitOrigin-RevId: 97a55242bf
|
4 years ago |
Megvii Engine Team
|
522e556b83
|
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
|
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
|
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
|
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
|
846642a881
|
docs(misc): add some notes
GitOrigin-RevId: de243064c5
|
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
|
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 |
Megvii Engine Team
|
1525a02530
|
feat(mge/module): add python wrapper for unfold
GitOrigin-RevId: 562103186f
|
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
|
13b15fb08c
|
feat(megbrain): add correlation opr
GitOrigin-RevId: 6d44598891
|
4 years ago |
Megvii Engine Team
|
601a33a84d
|
refactor(mge/dtr): update dtr api
GitOrigin-RevId: dc366c65be
|
4 years ago |
Megvii Engine Team
|
c269a690e5
|
fix(mge/dtr): filter bad ops
GitOrigin-RevId: 380262dca0
|
4 years ago |
Megvii Engine Team
|
9b69a02fb2
|
fix(imperative/tensor): init m_offset when constructing a Tensor with DeviceTensorND
GitOrigin-RevId: b340e27c47
|
4 years ago |
Megvii Engine Team
|
346d242051
|
fix(param_pack): impl param pack concat in imperative_rt
GitOrigin-RevId: 91edd9c0bf
|
4 years ago |
Megvii Engine Team
|
9c17cfc45d
|
fix(imperative/ops): add check_dtype for Elemwise in infer_attrs
GitOrigin-RevId: c777855753
|
4 years ago |
Megvii Engine Team
|
7af1ec66db
|
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
|
4 years ago |
Megvii Engine Team
|
516769391c
|
fix(mge/imperative): fix proxy graph comp_node finalize
GitOrigin-RevId: 22ad0d0fc6
|
4 years ago |
Megvii Engine Team
|
13e6ea349d
|
feat(imperative/opr): rebase rng refactoring to dev & add python module
GitOrigin-RevId: ee5984c52d
|
4 years ago |
Megvii Engine Team
|
cded8ef1a6
|
feat(imperative): add rng opdef
GitOrigin-RevId: b62dcea7f5
|
4 years ago |
Megvii Engine Team
|
e18afa0b05
|
feat(mge/module): python wrapper for conv_transpose3d
GitOrigin-RevId: 61097b8713
|
4 years ago |
Megvii Engine Team
|
1a7112997c
|
feat(opr-mm): add backend argument for remote send/recv
GitOrigin-RevId: 841a0e45ab
|
4 years ago |
Megvii Engine Team
|
fe99cdc794
|
feat(interpreter): add dynamic sublinear
GitOrigin-RevId: 7de54fe743
|
4 years ago |
Megvii Engine Team
|
a3ab9f1d41
|
feat(blob_manager): make all memory allocation go through blob manager
GitOrigin-RevId: f79155e5c3
|
4 years ago |
Megvii Engine Team
|
58ebb26156
|
fix(imperative/tensor): fix ConstTensorCache
GitOrigin-RevId: 0767bcfa28
|
4 years ago |
Megvii Engine Team
|
4f0e6eae35
|
fix(mge/trace): re-open gopt level in trace
GitOrigin-RevId: 5ebc712690
|
4 years ago |
Megvii Engine Team
|
fa1ca0ea6c
|
fix(imperative/opr): fix apply_on_var_node for broadcast
GitOrigin-RevId: 686fff4f73
|
4 years ago |
Megvii Engine Team
|
221ec38a48
|
feat(imperative): reduce profiler overhead
GitOrigin-RevId: dded9d9391
|
4 years ago |
Megvii Engine Team
|
c8697a7005
|
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
|
4 years ago |