Megvii Engine Team
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
3 years ago |
Megvii Engine Team
|
3ddc32d3e3
|
feat(android/whl): support android whl
GitOrigin-RevId: 05df16b494
|
3 years ago |
Megvii Engine Team
|
3252016e05
|
Merge pull request #401 from LosReturn:patch-1
GitOrigin-RevId: 440af8bd3d
|
3 years ago |
Megvii Engine Team
|
86ee4638bf
|
Merge pull request #402 from AA1HSHH:docstring-reshape
GitOrigin-RevId: 1ec572eb7c
|
3 years ago |
Megvii Engine Team
|
7ea104d788
|
Revert "fix(mge): replace _full_sync by sync"
This reverts commit e36ef45464
GitOrigin-RevId: 2d913c8ac9
|
3 years ago |
Megvii Engine Team
|
cbbca5fb10
|
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
|
3 years ago |
Megvii Engine Team
|
1d2510b6d7
|
fix(module): fix module dumped in old version without _short_name attr
GitOrigin-RevId: a1c815f613
|
3 years ago |
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
|
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
|
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
|
1657b8e881
|
fix(fastrun): fix persistent_cache in redis
GitOrigin-RevId: ada5862b05
|
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 |
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
|
6b28a2f9ec
|
fix(copybara): re-open redis cache
GitOrigin-RevId: 055bf6aa4f
|
3 years ago |
Megvii Engine Team
|
7fa70df9f2
|
style(imperative): remove absolute imports in megengine
GitOrigin-RevId: 7004929069
|
3 years ago |
Megvii Engine Team
|
cf1db2616e
|
fix(fastrun): replace py_redis with cpp_redis to avoid deadlock
GitOrigin-RevId: 9af7fa5c97
|
3 years ago |
Megvii Engine Team
|
b75d10094d
|
fix(mge/elwise): removed back to fp32 mode
GitOrigin-RevId: 6d1720f4af
|
3 years ago |
Megvii Engine Team
|
f75261d76f
|
fix(mge/utils): fix module_status error
GitOrigin-RevId: 9e004d98a1
|
3 years ago |
Megvii Engine Team
|
9fd2e66350
|
feat(mge/elwise): removed back to fp32 mode
GitOrigin-RevId: a665a279a6
|
3 years ago |
Megvii Engine Team
|
26146e5ab1
|
feat(mge/uniform): add the judgment of low and high
GitOrigin-RevId: 21788d1992
|
3 years ago |
kxz@thumt102-1
|
8f48da7ffe
|
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
|
3 years ago |
Megvii Engine Team
|
6102b96ca1
|
Merge pull request #407 from ShrimpLau:docstring_all_reduce_xxx
GitOrigin-RevId: 3d15b14fd1
|
3 years ago |
Megvii Engine Team
|
2881934cb8
|
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
|
3 years ago |
quqi.liu
|
0f5c26d746
|
docs(mge/distributed): change the allreduce related functions document
|
3 years ago |