Megvii Engine Team
b9e850a479
test(imperative): check env values after each pytest
GitOrigin-RevId: 826788113a
3 years ago
Megvii Engine Team
d984be591d
fix(imperative): restrict value converts to symbolvar
GitOrigin-RevId: 271267be69
3 years ago
Megvii Engine Team
94960ecf42
fix(imperative): restrict using convert_inputs in py_apply
GitOrigin-RevId: b021aac8a6
3 years ago
Megvii Engine Team
fd6f8e58b0
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
3 years ago
Megvii Engine Team
4d5faa3f0a
fix(imperative): using DnnOprCaller to avoid early destruction of dnn_opr
GitOrigin-RevId: 4a14f53738
3 years ago
Megvii Engine Team
273c0e8745
fix(autodiff): fix some bugs in relation to 2nd order grad
1. implement double backward for batchnorm
2. fix grad attach in nested grad manager
3. pad empty tensor for unsatisfied output_has_grad
4. support double backward for jit subgraph
5. support double backward for autodiff.Function
6. readd debug flag MGE_LOG_OP_DISPATCH
GitOrigin-RevId: cd31ddc620
3 years ago
Megvii Engine Team
bc9aa47ad8
feat(mge/indexing): support newaxis
GitOrigin-RevId: 8338c4b475
3 years ago
Megvii Engine Team
9779bc7f6d
fix(imperative): allow rng op infer shape fallible
GitOrigin-RevId: 687844500c
3 years ago
Megvii Engine Team
a4327c4d25
perf(imperative): add dim_expansion transform for conv/bn1d
GitOrigin-RevId: d14a69424d
3 years ago
Megvii Engine Team
2f3bc2db9d
perf(mge/utils): move astensor1d into C++
GitOrigin-RevId: e7c6659020
3 years ago
Megvii Engine Team
1709b3940b
perf(mge/functional): speed up Broadcast and Reshape
GitOrigin-RevId: a72f5460b6
3 years ago
Megvii Engine Team
3c3fc6f33c
refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++
GitOrigin-RevId: 01b5324392
3 years ago
Megvii Engine Team
0cb60d646d
feat(imperative): add output_descs for apply_on_physical_tensor
GitOrigin-RevId: 5b036c2c5a
3 years ago
Megvii Engine Team
2348a963f2
refactor(imperative): apply workspace limit hook to mini graph
GitOrigin-RevId: 27c51f3147
3 years ago
Megvii Engine Team
87f00232f2
fix(mge/gm): fix missing dtype checking while attach tensors
GitOrigin-RevId: f0aaea99b9
3 years ago
Megvii Engine Team
ab6d12caff
feat(mge): add conv padding mode
GitOrigin-RevId: 147ced856e
3 years ago
Megvii Engine Team
27346b0b65
test(opr): add scalar check for opr_test
GitOrigin-RevId: dcfd7ad5d6
3 years ago
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
42759dc7f5
test(dtr): make dtr_resnet1202 isolated
GitOrigin-RevId: 4d863cc32e
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