Megvii Engine Team
fd6f8e58b0
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
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
2f3bc2db9d
perf(mge/utils): move astensor1d into C++
GitOrigin-RevId: e7c6659020
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
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
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
de8ffe0c12
refactor(imperative): unify interpreter option setting
GitOrigin-RevId: 53510445cc
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
a72e0cb568
feat(imperative,src): add jit builder for custom op
GitOrigin-RevId: 3bb0b46311
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
68cdabd288
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
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
Megvii Engine Team
b6142bee9a
feat(imperative): support tensor with uint16 date type
GitOrigin-RevId: 57ba0633c7
3 years ago
Megvii Engine Team
dc28a43409
fix(mgb/bn): fix empty tensor input problem and other minor problems
GitOrigin-RevId: 1708b76cb8
3 years ago
Megvii Engine Team
32ad926559
feat(mge): improve presentation of async errors
GitOrigin-RevId: af5426c37d
3 years ago
Megvii Engine Team
e6a8b0256e
fix(mge): ignore errors caused by earlier async errors
GitOrigin-RevId: ce2028d38a
3 years ago
Megvii Engine Team
ee790cb62c
fix(imperative/python): fix warp_perspective doc and arange dtype
GitOrigin-RevId: dc4e021ec0
3 years ago
Megvii Engine Team
5431929ec7
feat(functional): let advance indexing support empty tensor and add more tests
GitOrigin-RevId: 49e1492934
3 years ago
Megvii Engine Team
432fdb7e6a
feat(mgb/opr): let SetSubtensor support empty IO
GitOrigin-RevId: 13909e3b11
3 years ago
Megvii Engine Team
e954b8f998
feat(mgb/opr): let Subtensor support empty IO
GitOrigin-RevId: a768498104
3 years ago
Megvii Engine Team
e5217db9f9
feat(traced_module): move traced_module out of the experimental folder
GitOrigin-RevId: 36c76b5277
3 years ago
Megvii Engine Team
9279104b11
feat(mge): add opdef serialization and apply_module_trace
GitOrigin-RevId: 5b45bded1d
3 years ago
Megvii Engine Team
a7a3bf2d6c
test(subgraph): simple test for subgraph
GitOrigin-RevId: 3d6ecd5db7
3 years ago
Megvii Engine Team
2950dd8d69
test(imperative): test autodiff.Function with non tensor arguments
GitOrigin-RevId: 6114f48d21
3 years ago
Megvii Engine Team
207a346351
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
4 years ago
Megvii Engine Team
070c811732
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
4 years ago
Megvii Engine Team
7225b0f09f
fix(mge/utils): use static infer manager to get value of network.varnode
GitOrigin-RevId: ecc47edab8
4 years ago
Megvii Engine Team
8480302da8
fix(autograd): make higher order grad experimental
GitOrigin-RevId: 81e1eb0ebf
4 years ago
Megvii Engine Team
72531f2b4a
test(autograd): add more tests for higher order grad
GitOrigin-RevId: 5fc308f87a
4 years ago
Megvii Engine Team
522e556b83
feat(autodiff): support higher order grad
GitOrigin-RevId: 86390d2179
4 years ago
Megvii Engine Team
787f187e7d
fix(imperative/src): fix dot backward error
GitOrigin-RevId: 02ba44a0e6
4 years ago
Megvii Engine Team
e417798f9e
fix(mge): correct pytype when calling apply from python
GitOrigin-RevId: 6abfa06ada
4 years ago
Megvii Engine Team
8e5bf94857
fix(mge/utils): fix bug of VarNode inplace operations
GitOrigin-RevId: fa9eec7079
4 years ago
Megvii Engine Team
226f24fa8e
test(mge/distributed): add gather scatter reduce broadcast grad test
GitOrigin-RevId: 8245e11f1d
4 years ago
Megvii Engine Team
7af1ec66db
fix(tensor): del valid tensors when compnode finalizing
GitOrigin-RevId: bace1f2b51
4 years ago
Megvii Engine Team
282dfc6232
refactor(imperative): alloc enum type class on heap
GitOrigin-RevId: d2b2acea22
4 years ago
Megvii Engine Team
65af9cfcdc
refactor(mge): use lower case for default string parameters in functional and module
GitOrigin-RevId: dbc1f27ff7
4 years ago
Megvii Engine Team
9dbe71dd1a
feat(mge/utils): add array method for varnode
GitOrigin-RevId: 6e4d05b475
4 years ago
Megvii Engine Team
cdbb4a20ff
fix(mge/tensor): fix tensor's serialization behavior
GitOrigin-RevId: 4d74a4b46e
4 years ago
Megvii Engine Team
e6b06914f7
feat(mge/tensor): support non-Tensor value in `_reset` and remove depreciated tests
GitOrigin-RevId: faf6c78aa8
4 years ago
Megvii Engine Team
3f2e15f534
fix(mge/logger): fix set_mgb_log_level
GitOrigin-RevId: 6c94c51483
4 years ago
Megvii Engine Team
dcfb6a537e
refactor(mge/functional): move functional api
GitOrigin-RevId: 9cd3e09996
4 years ago