Megvii Engine Team
b80fade3f9
fix(mge/core): avoid create RawTensor with zero-stride numpy ndarray
GitOrigin-RevId: a9b2940bdc
4 years ago
Megvii Engine Team
477820fe5d
refactor(mge): apply python-module changes
GitOrigin-RevId: 2c769864c3
4 years ago
Megvii Engine Team
59a9275c66
fix(mge): fix optimize_for_inference during trace.dump
GitOrigin-RevId: e10f7c323a
4 years ago
Megvii Engine Team
bd7f885a2b
fix(mge/pytest): remove __init__ in py-test
GitOrigin-RevId: e16eae9cbc
4 years ago
Megvii Engine Team
06041f8a7e
fix(mge/functional): fix trace warp_perspective
GitOrigin-RevId: 2071bb63a8
4 years ago
Megvii Engine Team
92afcf3eb4
refactor(mge/imperative): move cgtools to megengine.utils and add load_and_inference in cgtools
GitOrigin-RevId: abfee3d4fa
4 years ago
Megvii Engine Team
66f2dbd753
feat(mge/imperative): add module __repr__ for imperative
GitOrigin-RevId: ac13cc4659
4 years ago
Megvii Engine Team
53ec6b833e
fix(mge): fix graph binding refcnt
GitOrigin-RevId: c955eb260a
4 years ago
Megvii Engine Team
1965276845
fix(mge): fix bool.sum()
GitOrigin-RevId: 62c482db40
4 years ago
Megvii Engine Team
b970c6daf1
fix(functional): hide format interface and skip test_conv_bias
GitOrigin-RevId: 7de75bf084
4 years ago
Megvii Engine Team
fc74d5f2c3
feat(mge/dump): add more options for megbrain_graph.dump_graph
GitOrigin-RevId: 848e3c87c1
4 years ago
Megvii Engine Team
81b6a73382
fix(mge): fix bug of tensor.T
GitOrigin-RevId: 9fe9347b00
4 years ago
Megvii Engine Team
b111baf1b6
fix(mge/tensor): fix const target shape in reshape
GitOrigin-RevId: 7c04a9efba
4 years ago
Megvii Engine Team
715009b564
refactor(mge/api): remove external, dropout fix
GitOrigin-RevId: 5e6ff1a372
4 years ago
Megvii Engine Team
9005cf74d2
fix(mge/imperative): fix tracer leak
GitOrigin-RevId: 1a8ac20b46
4 years ago
Megvii Engine Team
70bc746c20
fix(mge): fix bug of identity()
GitOrigin-RevId: 4bfd3cafb5
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
4 years ago
Megvii Engine Team
91efd67dbf
refactor(mge/jit): change dump options, add test
GitOrigin-RevId: fbc0d51c2b
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
4 years ago
Megvii Engine Team
523ce65e0d
fix(mge/imperative): fix cgtools related tests
GitOrigin-RevId: 8f1eadb32e
4 years ago
Megvii Engine Team
dd39265e95
fix(mgb/dtype): enable TypeCvt for bool when trace(symbolic=True)
GitOrigin-RevId: 4e0fc63369
4 years ago
Megvii Engine Team
5c37a64e3f
feat(sdk/load_and_run): add dump with testcase for mge support imperative
GitOrigin-RevId: 17a0cf6293
4 years ago
Megvii Engine Team
ac11c38a97
feat(mge/imperative): add graph load and cgtools for imperative
GitOrigin-RevId: ba251f452a
4 years ago
Megvii Engine Team
10e942d952
refactor(mge): polish api
- refactor(mge): add support for optimizer.step().clear_grad() idiom
- refactor(mge): rename some methods of GradManager
- refactor(mge): remove tensor_nn and TensorDict
- refactor(mge): remove Buffer
- refactor(mge): remove requires_grad flag
- refactor(mge): add a default grad=None attribute to Tensor
- refactor(mge): deprecation for 1.0
GitOrigin-RevId: 3b723d9387
4 years ago
Megvii Engine Team
9389a805d8
fix(mge/oprmm): fix grad for collective comm
GitOrigin-RevId: e55dc47a6e
4 years ago
Megvii Engine Team
478f2c51dc
feat(mge/jit): add trace/dump options
GitOrigin-RevId: 0f43c14599
4 years ago
Megvii Engine Team
44d0b5daf5
feat(imperative): enable to() to copy to device
GitOrigin-RevId: f9caf17d24
4 years ago
Megvii Engine Team
69a7c55f55
test(mge/function): fix test for new optimizer api
GitOrigin-RevId: 8ae7720fe6
4 years ago
Megvii Engine Team
e283663a02
fix(mge/imperative): update tests to new optimizer api
GitOrigin-RevId: 3d06e3db3c
4 years ago
Megvii Engine Team
67859f04e1
fix(imperative): add __array__ and __array_wrap__ for tensorwrapper
GitOrigin-RevId: 87d4ab6c8e
4 years ago
Megvii Engine Team
5aa19f3d4a
test(mge/imperative): add more testcases for function
GitOrigin-RevId: e0675994f1
4 years ago
Megvii Engine Team
939bf3b28d
fix(mge/imperative): impl hashable for nms and cond_take
GitOrigin-RevId: 56918db014
4 years ago
Megvii Engine Team
8dc23e0fdf
fix(mge/functional): fix indexing_one_hot and remote_recv
GitOrigin-RevId: 00bdfb502b
4 years ago
Megvii Engine Team
2beb65b19d
fix(mge/module): tensor shape will not work when constructing numpy array
GitOrigin-RevId: 5a0d705970
4 years ago
Megvii Engine Team
403a1e7b24
feat(mge/imperative): add graph profiler
GitOrigin-RevId: c73563f337
4 years ago
Megvii Engine Team
a3b2232ba7
feat(mge): add trace.dump
GitOrigin-RevId: ea4c9d33c8
4 years ago
Megvii Engine Team
d4bad71190
feat(mge): add jit.trace
GitOrigin-RevId: ec647324c0
4 years ago
Megvii Engine Team
5474538f39
refactor(mge/imperative): fork multipledispatch
GitOrigin-RevId: a7c25a4302
4 years ago
Megvii Engine Team
03320a0548
fix(functional): change to user-friendly nms api and fix concat
GitOrigin-RevId: 76b4190270
4 years ago
Megvii Engine Team
46f03b9f7c
perf(mge/imperative): optimize gpu memory with standalone grad_fn
GitOrigin-RevId: 44952b235e
4 years ago
Megvii Engine Team
a226f02e7a
fix(mge/imperative): fix syncbn in symbolic mode
GitOrigin-RevId: a9794318a7
4 years ago
Megvii Engine Team
b9df2db1c5
build(third_party): update megray commit id
GitOrigin-RevId: cc753f941b
4 years ago
Megvii Engine Team
0067dcf068
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
4 years ago
Megvii Engine Team
7fadc16d3c
refactor(mge/functional): support tensor shape in interpolate and split
GitOrigin-RevId: 6430b64f01
4 years ago
Megvii Engine Team
e1fba6ece7
test(mge/distributed): add get_device_count_by_fork to fix distributed test skip
GitOrigin-RevId: 9ffd8a6149
4 years ago
Megvii Engine Team
60076f4715
test(mge/parampack): add parampack related opr test
GitOrigin-RevId: 550c6b6443
4 years ago
Megvii Engine Team
7e8f720953
refactor(mge/tensor): tensor reduce supports keepdims
GitOrigin-RevId: 8ed95e0fb8
4 years ago
Megvii Engine Team
6e882c1a86
feat(whl/imperative): compat for build python whl imperative and legacy runtime
GitOrigin-RevId: 7f6629ae1f
4 years ago
Megvii Engine Team
40d18c8920
fix(mge/imperative): fix tests when shape is tensor
GitOrigin-RevId: fd0095c1ec
4 years ago
Megvii Engine Team
92f7cceb11
fix(mge/functional): fix cond_take when index is numpy array
GitOrigin-RevId: 5fb93740f4
4 years ago