Megvii Engine Team
|
d919aaebc7
|
test(imperative): reopen special interpolate test and sync when test rng
GitOrigin-RevId: e3d03b4d1d
|
3 years ago |
Megvii Engine Team
|
ca2deebc0f
|
fix(imperative/tensor): make @ operator has the same functionality as matmul functional
GitOrigin-RevId: bf6136cc1a
|
3 years ago |
Megvii Engine Team
|
e860a08386
|
refactor(mge/indexing): move indexing into c++
GitOrigin-RevId: 43fbdb22dd
|
3 years ago |
Megvii Engine Team
|
e6706be23a
|
refactor(imperative): remove infer_output_mem_desc
GitOrigin-RevId: bff62b33a0
|
3 years ago |
Megvii Engine Team
|
a5af35c18c
|
refactor(imperative): remove command buffer
GitOrigin-RevId: 83c8cb6d3b
|
3 years ago |
Megvii Engine Team
|
406115dba0
|
fix(imperative): syncbn fp16 support
GitOrigin-RevId: 6059d5b76b
|
3 years ago |
Megvii Engine Team
|
884865703d
|
test(trace): test subtensor on unknown shape
GitOrigin-RevId: 1b5cfa4e0a
|
3 years ago |
Megvii Engine Team
|
c34a75d0f4
|
fix(trace): assume result is not scalar when shape is valid
GitOrigin-RevId: beee2d0f28
|
3 years ago |
Megvii Engine Team
|
bebb2cf4c3
|
Merge pull request #428 from P2Oileen:fix-pad
GitOrigin-RevId: f33ea46ad6
|
3 years ago |
Megvii Engine Team
|
6157d9cfef
|
fix(traced_module): fix Module compatible issue and traced module getattr check
GitOrigin-RevId: 62eb3bfb10
|
3 years ago |
Megvii Engine Team
|
5e17b3e4c6
|
Merge pull request #426 from Qsingle:fix-pixel_suffle
GitOrigin-RevId: db9a0f7551
|
3 years ago |
Megvii Engine Team
|
2bebe80e93
|
fix(imperative): fix the default pickle protocol version of save
GitOrigin-RevId: fab2dc7369
|
3 years ago |
Megvii Engine Team
|
8e94af9d78
|
Merge pull request #400 from jieli-matrix:docstring-svd
GitOrigin-RevId: 3bcbea3440
|
3 years ago |
Megvii Engine Team
|
17f2dffb5b
|
fix(imperative/cpu/multithread): fix multithread at imperative
GitOrigin-RevId: 9120d5cb48
|
3 years ago |
Megvii Engine Team
|
3159eecadd
|
fix(init): fix fan_in and fan_out for group conv2d
GitOrigin-RevId: a6f41063f0
|
3 years ago |
Megvii Engine Team
|
275b63114d
|
fix(imperative): fix use collections error from python3.10
GitOrigin-RevId: 5dd019b336
|
3 years ago |
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
|
d3689c3f3c
|
feat(imperative/python): add transformation manager
GitOrigin-RevId: a3c1732ffd
|
3 years ago |
Megvii Engine Team
|
42759dc7f5
|
test(dtr): make dtr_resnet1202 isolated
GitOrigin-RevId: 4d863cc32e
|
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 |