Megvii Engine Team
cd02d7c85e
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
4 years ago
Megvii Engine Team
e82fa4ec23
fix(gopt): using new_inp for build_chain in DelayBroadcast pass
GitOrigin-RevId: efc6377197
4 years ago
Megvii Engine Team
a09fc5f784
fix(mgb/serialization): disable inplace arith graph opt in graph load
GitOrigin-RevId: d63baf8356
4 years ago
Megvii Engine Team
fc8d13cda3
fix(build): fix linkage on ubuntu1604
GitOrigin-RevId: ee2a8360b0
4 years ago
Megvii Engine Team
43b4d4a430
feat(imperative/test): add more correctness tests
GitOrigin-RevId: 2ef6027bb1
4 years ago
Megvii Engine Team
91a3580f75
refactor(mge/cgtools): remove load_and_inference and use GraphInference
GitOrigin-RevId: 0e688ebd59
4 years ago
Megvii Engine Team
4485e780ae
fix(mge/functional): fix linspace device and open other trace tests
GitOrigin-RevId: 4667c4adec
4 years ago
Megvii Engine Team
334eda8717
refactor(mge): test trace inside opr_test
GitOrigin-RevId: 2cf1135c1c
4 years ago
Megvii Engine Team
938152af63
fix(mge/functional): convert input type to float32 for more elemwise op
GitOrigin-RevId: cf3bf8cb80
4 years ago
Megvii Engine Team
194660462c
feat(mge/funcitonal): add cvt_color opr python interface
GitOrigin-RevId: 29e069fb23
4 years ago
Megvii Engine Team
412d1f0cdc
fix(mge/tensor): implement abtract method to fix lint errors
GitOrigin-RevId: d53f2eac6a
4 years ago
Megvii Engine Team
dbb3dd681f
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
4 years ago
Megvii Engine Team
ff05667b48
test(dist): refactor distributed test with fixtures
GitOrigin-RevId: e69acb72da
4 years ago
Megvii Engine Team
9fb8444d24
fix(imperative): catch python exception in c++
GitOrigin-RevId: 16a2abfdad
4 years ago
Megvii Engine Team
e3a3e0cd3c
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
4 years ago
Megvii Engine Team
8e50a6daa7
feat(mge/utils): add GraphInference in cgtools
GitOrigin-RevId: 72f2201169
4 years ago
Megvii Engine Team
adc49de803
feat(mge/imperative): expose c++ tensor reference count
GitOrigin-RevId: 1940881adc
4 years ago
Megvii Engine Team
fe1680b378
fix(imperative/ops): improve infer_output_attrs for broadcast
GitOrigin-RevId: 6b7ed55769
4 years ago
Megvii Engine Team
27a9326c66
fix(mge/module): fix frozen batch norm
GitOrigin-RevId: 143d468a37
4 years ago
Megvii Engine Team
c3ba02801a
chore(mge): correct license headers
GitOrigin-RevId: 2dd4438f90
4 years ago
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