Megvii Engine Team
|
0cb60d646d
|
feat(imperative): add output_descs for apply_on_physical_tensor
GitOrigin-RevId: 5b036c2c5a
|
3 years ago |
Megvii Engine Team
|
7bd848ce04
|
fix(subgraph): fix hand-written backward for serval jit-elemwise ops
GitOrigin-RevId: ea3a40d96e
|
4 years ago |
Megvii Engine Team
|
7be7656c9f
|
fix(imperative): explicitly manage global structures
GitOrigin-RevId: 0f910c34b6
|
4 years ago |
Megvii Engine Team
|
b6ce02a152
|
fix(subgraph): fallback back to cg if jit unsupported
GitOrigin-RevId: 853a00a402
|
4 years ago |
Megvii Engine Team
|
3c61e0e02a
|
feat(ops): add JITFusion op
GitOrigin-RevId: 7dc35d4e80
|
4 years ago |
Megvii Engine Team
|
aa587446fc
|
feat(subgraph): support shape inference for CompiledOp
GitOrigin-RevId: a96b8f3446
|
4 years ago |
Megvii Engine Team
|
e6706be23a
|
refactor(imperative): remove infer_output_mem_desc
GitOrigin-RevId: bff62b33a0
|
4 years ago |
Megvii Engine Team
|
7f03ae9aed
|
fix(imperative): reduce tls usage
GitOrigin-RevId: a716b2ae98
|
4 years ago |
Megvii Engine Team
|
369c2ccc5a
|
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
|
4 years ago |
Megvii Engine Team
|
d02b0a08f7
|
feat(lite): fix typo
GitOrigin-RevId: 8c46aa3a30
|
4 years ago |
Megvii Engine Team
|
ca4374b6e2
|
perf(subgraph): remove unnecessary sync
GitOrigin-RevId: 1546286706
|
4 years ago |
Megvii Engine Team
|
989fdde255
|
refactor(subgraph): use graph queue to cache compiled op graphs
GitOrigin-RevId: cba8574c73
|
4 years ago |
Megvii Engine Team
|
2a063f8e87
|
fix(subgraph): fix scope mismatch of subgraph content
GitOrigin-RevId: 6e23456250
|
4 years ago |
Megvii Engine Team
|
53da5c79f4
|
feat(cg): add comp_seq_sync_device option
GitOrigin-RevId: c2199c59e9
|
4 years ago |
Megvii Engine Team
|
3fd3e000d1
|
feat(ops): add serval utility ops
GitOrigin-RevId: 623cb5ddfc
|
4 years ago |
Megvii Engine Team
|
18274e023c
|
fix(imperative): fix error message when applying custom function with non-tensor arguments
GitOrigin-RevId: 387d6fda4a
|
4 years ago |
Megvii Engine Team
|
5f558042b2
|
fix(imperative/ops): use tblgen to generate FastpathCopy
GitOrigin-RevId: b02157dacb
|
4 years ago |
Megvii Engine Team
|
522e556b83
|
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
|
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
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
5 years ago |
Megvii Engine Team
|
dc2507453b
|
feat(mge): add python custom op
GitOrigin-RevId: 35da0bb301
|
5 years ago |
Megvii Engine Team
|
9a79342d35
|
refactor(mge/virtualdep): remove virtualdep opdef
GitOrigin-RevId: bbe3ae3fa3
|
5 years ago |
Megvii Engine Team
|
88b402ef62
|
feat(mge/trace): tracing use id to set priority
GitOrigin-RevId: 6e1f1ece0e
|
5 years ago |
Megvii Engine Team
|
798f7b3ef6
|
feat(imperative): add virtual_deps op
GitOrigin-RevId: 5845520989
|
5 years ago |