Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
9a09f761bb
fix(whl): append buildid to wheel version
GitOrigin-RevId: 00dabfa147
4 years ago
Megvii Engine Team
af5ab4d976
fix(imperative/tensor): fix Py_DECREF on uninitialized pointer
GitOrigin-RevId: 889255537e
4 years ago
Megvii Engine Team
10d9d77e41
refactor(mge/distributed): remove get_free_ports func
GitOrigin-RevId: b85c4885be
4 years ago
Megvii Engine Team
0adf49b137
fix(mge/distributed): fix deadlock by mixing thread and fork
GitOrigin-RevId: c138cb9c28
4 years ago
Megvii Engine Team
ae8b38f634
fix(cmake/whl): reduce wheel size
as some reason: some version of trt/cudnn need
-Wl,--whole-archive attribute to fix cuda issue,
target A(CMAKE OBJECT,eg megbrain/megdnn): -Wl,--whole-archive depend on trt/cudnn
target B(SHARED LIBS,eg libmegengine.so) --> A with `PUBLIC` depends
target C(SHARED LIBS,eg _impertive.so) --> B with `PUBLIC` depends,
then ld will force link fatbin section into C, which will be undesired
what`s more, attribute PUBLIC/PRIVATE do not take effect
to OBJECT library(megbrain/megdnn)
what`s more, megengine/megengine_export have to PUBLIC for
mgb/imperative target, as SRC/include depends
so we pull cudalib depend from megbrain/megdnn to
megengine/megengine_export on linux os, to finall
target at windows os, for example lar on windows
GitOrigin-RevId: b278a69e1c
4 years ago
Megvii Engine Team
4aaae9950b
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
4 years ago
Megvii Engine Team
d04b4bc006
fix(interp): thread safety for drop and swapout
GitOrigin-RevId: 7684f160bf
4 years ago
Megvii Engine Team
3eb529d692
refactor(interpreter): recognize recomp on main thread rather than worker
GitOrigin-RevId: 4ba3942ce4
4 years ago
Megvii Engine Team
df976782fa
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
4 years ago
Megvii Engine Team
2eed7d83be
docs(mge): restore autodiff.Function docs
GitOrigin-RevId: cbd84df916
4 years ago
Megvii Engine Team
5100317660
feat(mge): make Function available at autodiff.Function
GitOrigin-RevId: 7c9ab65351
4 years ago
Megvii Engine Team
ca1d74ad3f
fix(mge/utils): fix types import error
GitOrigin-RevId: d2d4bd2725
4 years ago
Megvii Engine Team
abe3c165ba
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
4 years ago
Megvii Engine Team
c418d3cd95
fix(mge/data): add timeout event
GitOrigin-RevId: 43f2ba1456
4 years ago
Megvii Engine Team
c53abcdf1d
chore(mge): minor improvements related to grad
GitOrigin-RevId: 102467d79d
4 years ago
Megvii Engine Team
0a3ca25337
fix(mge): fix backward graph optimization
GitOrigin-RevId: 28fd00ac54
4 years ago
Megvii Engine Team
ea8eb4cf72
feat(mge/distributed): scalar support for distributed functions
GitOrigin-RevId: 53f3575baf
4 years ago
Megvii Engine Team
b83c77e105
test(mge/test_save_load): remove test model file
GitOrigin-RevId: 893b82b1c9
4 years ago
Megvii Engine Team
c522991385
test(mge/core): add host compute test, fix async_level and swap/drop test
GitOrigin-RevId: f9b5a83507
4 years ago
Megvii Engine Team
c8e04ce4df
fix(imperative/interpreter): use info->h_value for host compute
Rather than use host_only Tensor and modify Put's behavior.
GitOrigin-RevId: f890d66acb
4 years ago
Megvii Engine Team
a5a606792e
feat(imperative/interpreter): add more dispatch mode in apply_op
GitOrigin-RevId: 2663504470
4 years ago
Megvii Engine Team
248d8bf0dc
feat(imperative/ops): improve infer attrs validate function
GitOrigin-RevId: 6fab3b1402
4 years ago
Megvii Engine Team
8ed2077bcb
fix(imperative/interpreter): reset m_waitee on exception
GitOrigin-RevId: ecf5245d9c
4 years ago
Megvii Engine Team
8fe865d88a
feat(imperative/ops): add infer_output_attrs for Reshape
GitOrigin-RevId: 9150d7f84d
4 years ago
Megvii Engine Team
267d612723
fix(imperative/mini_graph): fix no_err value in shape infer
GitOrigin-RevId: 087d15db66
4 years ago
Megvii Engine Team
bc3561c1ee
fix(imperative/ops): fix inplace_add's output desc, remove value
GitOrigin-RevId: b294eead0a
4 years ago
Megvii Engine Team
6295f843a5
fix(mge/tensor): fix mem leak when getting value in trace
GitOrigin-RevId: 8bca2e7380
4 years ago
Megvii Engine Team
55042195d4
chore(winograd): add Convolutionv2 param
GitOrigin-RevId: 1a9e2ea340
4 years ago
Megvii Engine Team
7191c4bd9f
fix(mge/data): support timeout for serial stream dataloader
GitOrigin-RevId: 1ae5a8cfda
4 years ago
Megvii Engine Team
b7ed0cb850
fix(mge/quantization): replace `_reset` with "=" in observer
GitOrigin-RevId: ed6af9b98d
4 years ago
Megvii Engine Team
5d0c9bef4c
chore(mge/version): change dev version to 1.3
GitOrigin-RevId: b5e5327a46
4 years ago
Megvii Engine Team
0b4918b25e
test(mge/quantization): classmethod `from_float_module` of qat module
GitOrigin-RevId: 95c3d45f83
4 years ago
Megvii Engine Team
e9a7be46b1
feat(mgb): adapt to imperative runtime
GitOrigin-RevId: 3bccc17b62
4 years ago
Megvii Engine Team
08cc10324e
fix(imperative): fix persistent_cache
GitOrigin-RevId: 8f7bb5899f
4 years ago
Megvii Engine Team
998f71a8c9
perf(mge/imperative): do not busy wait in imperative's queue
GitOrigin-RevId: de8db51096
4 years ago
Megvii Engine Team
278b2baa8c
perf(mge): add memory optimization for backward graph
precompute ops in forward to reduce saved tensor size
GitOrigin-RevId: d67043ba82
4 years ago
Megvii Engine Team
5c7d48cdb9
fix(mge/functional): fix tensor split
GitOrigin-RevId: 0a112ab0bd
4 years ago
Megvii Engine Team
82847bc70a
docs(mge): add conv2d output shape computation formula
GitOrigin-RevId: 7ee476a80d
4 years ago
Megvii Engine Team
d3bfb0e983
fix(mge): fix trace exit code and reformat
GitOrigin-RevId: 145c06b7e7
4 years ago
Megvii Engine Team
23b9a98f5e
fix(mge): fix sublnear cuda and mem leak
GitOrigin-RevId: 82091ec9a6
4 years ago
Megvii Engine Team
c70a49ed2c
fix(mge): correct trace outputs when grad does copy
GitOrigin-RevId: 65c8956a7d
4 years ago
Megvii Engine Team
d4ada69d3b
refactor(mge): trace exception in compiled info
GitOrigin-RevId: 508f5463b9
4 years ago
Megvii Engine Team
c9c3429a6b
refactor(mge): fix sublinear
GitOrigin-RevId: 5bb0383781
4 years ago
Megvii Engine Team
de0742be25
refactor(mge): reopen passed assertions
GitOrigin-RevId: e0276e73e3
4 years ago
Megvii Engine Team
a90c937d22
feat(interpreter): add command buffer for inplace
GitOrigin-RevId: 020d1e88d4
4 years ago
Megvii Engine Team
09af925fb6
fix(mge): fix cpp trace function release
GitOrigin-RevId: 924f945c21
4 years ago
Megvii Engine Team
6919127fd4
fix(mge): fix event pool destructor order issue and do full sync at python exit
GitOrigin-RevId: b9fa1f45b1
4 years ago
Megvii Engine Team
60459bd79b
fix(mge/pytest): remove sync in pytest teardown & add full_sync
GitOrigin-RevId: ee278a8ac1
4 years ago
Megvii Engine Team
c49427d15a
feat(imperative): add inplace add_update option in optimizer
GitOrigin-RevId: b8feb49321
4 years ago