Megvii Engine Team
|
cf5e9488bb
|
fix(traced_module): fix module trace transformation
GitOrigin-RevId: ce11fe5e09
|
3 years ago |
Megvii Engine Team
|
97c90d9137
|
feat(traced_module): add _exclude_from_trace
GitOrigin-RevId: 615b769a02
|
3 years ago |
Megvii Engine Team
|
30e565e5b8
|
fix(traced_module): fix error message
GitOrigin-RevId: 3046225e30
|
3 years ago |
Megvii Engine Team
|
de8ffe0c12
|
refactor(imperative): unify interpreter option setting
GitOrigin-RevId: 53510445cc
|
3 years ago |
Megvii Engine Team
|
8b60bdfa10
|
fix(mge): replace _full_sync by sync
GitOrigin-RevId: e36ef45464
|
3 years ago |
Megvii Engine Team
|
20b42a8c3b
|
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
|
3 years ago |
Megvii Engine Team
|
2faa6ea5a9
|
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
|
3 years ago |
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
|
85ea882cb5
|
fix(mgb/ops): immutable tensor support empty storage
GitOrigin-RevId: 2851498fce
|
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
|
c3d63f14a5
|
feat(dispatch): add gdb debug utils
GitOrigin-RevId: 393b0c9c76
|
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 |
XindaH
|
bd3c4a0527
|
Merge pull request #431 from MegEngine/try-import
|
3 years ago |
huangxinda
|
b0c9705887
|
fix(ci): add MGE_WITH_CUSTOM_OP in github ci
|
3 years ago |
huangxinda
|
a77ae9a1c2
|
feat(ci): download ninja
|
3 years ago |
Megvii Engine Team
|
bac8cc48ee
|
fix(fastrun): fallback when redis config is empty
GitOrigin-RevId: b61630d007
|
3 years ago |
Megvii Engine Team
|
f4f20046c4
|
fix(mgb): fix tensorrt runtimeopr get output var shape bug
GitOrigin-RevId: b830706a89
|
3 years ago |
Megvii Engine Team
|
82be0aaced
|
test(dnn): fix compute capability requirement for NCHWX test
GitOrigin-RevId: d2f8022be1
|
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
|
3b41840b68
|
fix(mgb): change caffepooling log level
GitOrigin-RevId: 290d243ef5
|
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
|
32717b0ca4
|
fix(build): split some cpp, which consume two many mem when build
make build possible at 8G ddr env, when -j8
GitOrigin-RevId: d0c442b41d
|
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 |