Megvii Engine Team
|
f5b8fec4ca
|
fix(imperative): remove big tensor from host side
GitOrigin-RevId: 2047982d73
|
3 years ago |
Megvii Engine Team
|
68cde8734e
|
fix(mge/imperative): support broadcast with None
GitOrigin-RevId: dd330a2a1d
|
3 years ago |
Megvii Engine Team
|
0bdd0b1467
|
refactor(dispatch): switch to new dispatch system
GitOrigin-RevId: 32dd49a23a
|
3 years ago |
Megvii Engine Team
|
d3689c3f3c
|
feat(imperative/python): add transformation manager
GitOrigin-RevId: a3c1732ffd
|
3 years ago |
Megvii Engine Team
|
9ce1f0f5d1
|
refactor(dispatch): implement grad
GitOrigin-RevId: d8367f9587
|
3 years ago |
Megvii Engine Team
|
c609c031f1
|
refactor(dispatch): implement symbol
GitOrigin-RevId: c7bd86f5c1
|
3 years ago |
Megvii Engine Team
|
e32929dfd2
|
refactor(dispatch): implement scalar
GitOrigin-RevId: b244c2ca1a
|
3 years ago |
Megvii Engine Team
|
59084fa857
|
refactor(dispatch): implement lazy_eval
GitOrigin-RevId: 4e3f3a1c44
|
3 years ago |
Megvii Engine Team
|
d2b67c2a88
|
refactor(dispatch): implement trace
GitOrigin-RevId: f8d3005732
|
3 years ago |
Megvii Engine Team
|
39ac606b9c
|
refactor(dispatch): implement eval
GitOrigin-RevId: 32563e0a27
|
3 years ago |
Megvii Engine Team
|
42759dc7f5
|
test(dtr): make dtr_resnet1202 isolated
GitOrigin-RevId: 4d863cc32e
|
3 years ago |
Megvii Engine Team
|
7f03ae9aed
|
fix(imperative): reduce tls usage
GitOrigin-RevId: a716b2ae98
|
3 years ago |
Megvii Engine Team
|
d8d5edb31f
|
refactor(opdef): overwrite shared_from_this to avoid const_cast
GitOrigin-RevId: 5322dbbb3f
|
3 years ago |
Megvii Engine Team
|
4b0ecb5deb
|
fix(ops/recv): use std::vector to store shape to support scalar
GitOrigin-RevId: e1dac3c919
|
3 years ago |
Megvii Engine Team
|
2e42bc083d
|
feat(dispatch): implement new op dispatch system
GitOrigin-RevId: 355da6b814
|
3 years ago |
Megvii Engine Team
|
2be6ceda03
|
feat(imperative/utils): add serval utils
GitOrigin-RevId: f401663ae3
|
3 years ago |
Megvii Engine Team
|
e7c2ed1183
|
fix(imperative): add missing pragma once
GitOrigin-RevId: 7a9fff1171
|
3 years ago |
Megvii Engine Team
|
462364eb06
|
perf(interpreter): don't check host value if unnecessary
GitOrigin-RevId: 5306c71328
|
3 years ago |
Megvii Engine Team
|
d23d1352e7
|
fix(imperative/python): add the default warning for args descending
GitOrigin-RevId: cb5f065e6c
|
3 years ago |
Megvii Engine Team
|
5e80f0211e
|
fix(mge/network): fix Network not returning dump_info to align trace.dump
GitOrigin-RevId: 865adf75e2
|
3 years ago |
Megvii Engine Team
|
7ecdbf251a
|
fix(traced_module): fix ones/zeros functional compatiable
GitOrigin-RevId: 7ec2c4d3f5
|
3 years ago |
Megvii Engine Team
|
d24f198cfa
|
fix(imperative): clear custom op build cache
GitOrigin-RevId: cb145fe1ad
|
3 years ago |
Megvii Engine Team
|
bac8cc48ee
|
fix(fastrun): fallback when redis config is empty
GitOrigin-RevId: b61630d007
|
3 years ago |
Megvii Engine Team
|
619d78ed86
|
fix(imperative): check async error when getting value
GitOrigin-RevId: 52b8a29932
|
3 years ago |
Megvii Engine Team
|
2afa0af926
|
docs(imperative/jit): add docstring of dump
GitOrigin-RevId: 49eacc0e58
|
3 years ago |
Megvii Engine Team
|
04b1424169
|
fix(traced_module): fix TracedModule InternalGraph deepcopy exceed max recursion limit
GitOrigin-RevId: 2b52ad913d
|
3 years ago |
Megvii Engine Team
|
355782aecb
|
fix(traced_module): clear node after trace module
GitOrigin-RevId: f7f6024034
|
3 years ago |
Megvii Engine Team
|
fba544881b
|
fix(traced_module): add functional pad to __all__
GitOrigin-RevId: a4fdedfb7f
|
3 years ago |
Megvii Engine Team
|
bf0f3d31f5
|
fix(mge/traced_module): fix treedef repr
GitOrigin-RevId: 3df05e9c22
|
3 years ago |
Megvii Engine Team
|
a0b3a3c0b3
|
feat(imperative): add TracedModule checker
GitOrigin-RevId: 12de7b278e
|
3 years ago |
Megvii Engine Team
|
1999307015
|
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
|
3 years ago |
Megvii Engine Team
|
a93741815b
|
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
|
3 years ago |
Megvii Engine Team
|
bdca240f27
|
test(fastrun): add more tests for various persistent cache
GitOrigin-RevId: 45c45c8f19
|
3 years ago |
Megvii Engine Team
|
1657b8e881
|
fix(fastrun): fix persistent_cache in redis
GitOrigin-RevId: ada5862b05
|
3 years ago |
Megvii Engine Team
|
878ce91165
|
fix(mge/test): replace equal with allclose to fix rng test for ci
GitOrigin-RevId: 12758cf5d5
|
3 years ago |
Megvii Engine Team
|
c53cad2049
|
feat(cmake): format all cmake file
GitOrigin-RevId: 0a4ecab99b
|
3 years ago |
Megvii Engine Team
|
6011f51001
|
style(all): fix clang-format for MGB_DEFINE inside another macro
GitOrigin-RevId: 8c2b6a2aed
|
3 years ago |
Megvii Engine Team
|
111fa97514
|
Merge pull request #422 from Asthestarsfalll:zeroslike
GitOrigin-RevId: 4b464446b5
|
3 years ago |
Megvii Engine Team
|
48e0973875
|
docs(mge/functional): update functional.zeros docstring
GitOrigin-RevId: 4ce2752bfc
|
3 years ago |
Megvii Engine Team
|
a72e0cb568
|
feat(imperative,src): add jit builder for custom op
GitOrigin-RevId: 3bb0b46311
|
3 years ago |
Megvii Engine Team
|
3ff5ca5ffe
|
feat(mge/traced_module): support to modify the name of Node during graph surgery
GitOrigin-RevId: 9ecf6f2c5b
|
3 years ago |
Megvii Engine Team
|
3a21920955
|
fix(mge/traced_module): fix some bugs
GitOrigin-RevId: 88f98829ce
|
3 years ago |
Megvii Engine Team
|
714340e912
|
Merge pull request #421 from Asthestarsfalll:master
GitOrigin-RevId: 416b327415
|
3 years ago |
Megvii Engine Team
|
46d4bd8a59
|
feat(windows): let sdk do not care about more macro on win
GitOrigin-RevId: c522c2fd63
|
3 years ago |
Asthestarsfalll
|
520db818d0
|
docs(mge/functional): update functional.zeros_like docstring
|
3 years ago |
Asthestarsfalll
|
b7fe6a27ee
|
docs(mge/functional): update functional.ones_like docstring
|
3 years ago |
Megvii Engine Team
|
d9a46ea47b
|
fix(dnn): correct behaviour of floor div for int tensor
GitOrigin-RevId: 1444f69cce
|
3 years ago |
Megvii Engine Team
|
8918abb17b
|
Revert "test(fastrun): suppress persistent cache test"
This reverts commit 5d6e722deb .
GitOrigin-RevId: 74e184d5c6
|
3 years ago |
Megvii Engine Team
|
6b28a2f9ec
|
fix(copybara): re-open redis cache
GitOrigin-RevId: 055bf6aa4f
|
3 years ago |
Megvii Engine Team
|
60c14b683f
|
fix(copybara): remove redis client in imperative
GitOrigin-RevId: 86fe93f4c9
|
3 years ago |