Megvii Engine Team
|
df3474ca1d
|
perf(functional): rewrite serval elemwise ops with jit subgraph
GitOrigin-RevId: 26247e21d9
|
3 years ago |
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
|
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
|
3159eecadd
|
fix(init): fix fan_in and fan_out for group conv2d
GitOrigin-RevId: a6f41063f0
|
3 years ago |
Megvii Engine Team
|
95ac055538
|
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
|
3 years ago |
Megvii Engine Team
|
cf5e9488bb
|
fix(traced_module): fix module trace transformation
GitOrigin-RevId: ce11fe5e09
|
3 years ago |
Megvii Engine Team
|
de8ffe0c12
|
refactor(imperative): unify interpreter option setting
GitOrigin-RevId: 53510445cc
|
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
|
d24f198cfa
|
fix(imperative): clear custom op build cache
GitOrigin-RevId: cb145fe1ad
|
3 years ago |
Megvii Engine Team
|
619d78ed86
|
fix(imperative): check async error when getting value
GitOrigin-RevId: 52b8a29932
|
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
|
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
|
878ce91165
|
fix(mge/test): replace equal with allclose to fix rng test for ci
GitOrigin-RevId: 12758cf5d5
|
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 |
Qsingle
|
14499e83d9
|
fix the type error of pixel shuffle
add the test of dtype float
|
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
|
cf1db2616e
|
fix(fastrun): replace py_redis with cpp_redis to avoid deadlock
GitOrigin-RevId: 9af7fa5c97
|
3 years ago |
Megvii Engine Team
|
f75261d76f
|
fix(mge/utils): fix module_status error
GitOrigin-RevId: 9e004d98a1
|
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
|
f67aa6466e
|
test(fastrun): suppress persistent cache test
GitOrigin-RevId: 5d6e722deb
|
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 |
Megvii Engine Team
|
1f0cc891b0
|
feat(dnn): enable eye to support bool
GitOrigin-RevId: 76d874d5b7
|
3 years ago |
Megvii Engine Team
|
11d75fecb5
|
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
|
3 years ago |
Megvii Engine Team
|
7a023c059a
|
feat(mge/traced_module): add optimization api
GitOrigin-RevId: eaa7402640
|
3 years ago |
Megvii Engine Team
|
6c692b26a7
|
feat(mge/traced_module): add some fuse passes
GitOrigin-RevId: 065f9df32e
|
3 years ago |
Megvii Engine Team
|
0185c1a9b9
|
feat(mge/traced_module): add argspec for top TracedModule
GitOrigin-RevId: 8e31a00c7e
|
3 years ago |
Megvii Engine Team
|
e6c271ae46
|
fix(mge/traced_module): fix some bugs for graph surgery
GitOrigin-RevId: 6328a84cbc
|
3 years ago |
Megvii Engine Team
|
7b19bc76fb
|
feat(traced_module): support traced module backward compatible serialization
GitOrigin-RevId: aaa9e51c74
|
3 years ago |
Megvii Engine Team
|
89186edc5d
|
fix(dnn): correct reduce/argmxx/fakequant calculation with nan
GitOrigin-RevId: 7e78bdae91
|
3 years ago |
Megvii Engine Team
|
68cdabd288
|
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
|
3 years ago |
Megvii Engine Team
|
5c94db16f2
|
feat(mge/functional): add groups support for conv_transpose2d & 3d
GitOrigin-RevId: b75b792fb4
|
3 years ago |
Megvii Engine Team
|
7591718d20
|
feat(mge): add functional test
GitOrigin-RevId: aa0be62686
|
3 years ago |
Megvii Engine Team
|
846cc13ca8
|
fix(mge/functional): fix default dtype of F.full
GitOrigin-RevId: 5cd34af9bf
|
3 years ago |
Megvii Engine Team
|
0ddabb06cc
|
fix(imperative/astype): fix astype when target dtype is int4
GitOrigin-RevId: 1c5195176d
|
3 years ago |
Megvii Engine Team
|
d909950f64
|
refactor(imperative): remove swap in dtr
GitOrigin-RevId: 5c9e42f74a
|
3 years ago |