Megvii Engine Team
ad87f78a14
chore(imperative): refine tblgen for generating op name
GitOrigin-RevId: f47ceae726
4 years ago
Megvii Engine Team
0a191f8d48
fix(misc): some fixes to make faster_rcnn run in no-symbolic-trace
GitOrigin-RevId: 0fc40ad9c8
5 years ago
Megvii Engine Team
f95bb7b7c9
perf(mge/profiler): reduce profiler overhead
GitOrigin-RevId: 4e1fff9dc4
5 years ago
Megvii Engine Team
4f3875eb4f
docs(mge): add missing docstring and fix sphinx build warnings
GitOrigin-RevId: 4ce73cfd80
5 years ago
Megvii Engine Team
98d6ba7e2f
refactor(mge/functional): move matinv from nn to math
GitOrigin-RevId: 75c48ce327
5 years ago
Megvii Engine Team
17323dbd73
feat(dist): collect return values in dist.launcher
GitOrigin-RevId: 519e768ce9
4 years ago
Megvii Engine Team
5b697c714c
chore(imperative): remove unused reference to megbrain_graph
GitOrigin-RevId: 2c31f13bb8
5 years ago
Megvii Engine Team
f0170b9c21
fix(whl): opt whl version
example:
1:export PUBLIC_VERSION_POSTFIX=rc1 for create a rc1 version
2:export PUBLIC_VERSION_POSTFIX=dev20210225 for create daily version
3:export PUBLIC_VERSION_POSTFIX=rc1.dev20210225 and export LOCAL_VERSION=mc40 for create special version
GitOrigin-RevId: 8caa7646d0
4 years ago
Megvii Engine Team
47db29aaa2
fix(mge/module): add kwargs param for all modules
GitOrigin-RevId: 7245e669a7
5 years ago
Megvii Engine Team
6fb19b664b
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
5 years ago
Megvii Engine Team
09de5a0725
feat(mgb/serialization): be able to serialize operator names
GitOrigin-RevId: d295abb5da
5 years ago
Megvii Engine Team
bb8f292816
refactor(mge/func): change `conv_execution_strategy` to `execution_strategy`
GitOrigin-RevId: 7aef9935ee
5 years ago
Megvii Engine Team
096d340c18
fix(whl): fix whl version, fix create manylinux2014 image
GitOrigin-RevId: 3faa59d945
5 years ago
Megvii Engine Team
f211124865
feat(mge/trace): support dict return value processing in trace
GitOrigin-RevId: 5b1c08848b
5 years ago
Megvii Engine Team
cbff4d7c1a
fix(imperative/ops): fix infer_output_attrs_fallible for reshape
GitOrigin-RevId: a93567d79a
5 years ago
Megvii Engine Team
d1be31279a
fix(mge/quantization): fix tqt load and convert issue and observer calculate params issue
GitOrigin-RevId: f8511f72ad
5 years ago
Megvii Engine Team
30d6b4f65d
fix(mge): fix scalar parameter change to 1-dim parameter after backward and optimize
GitOrigin-RevId: 1794369a71
5 years ago
Megvii Engine Team
cf27dd642c
fix(cuda): use cudnn8.0.4 as cu111 default libs
GitOrigin-RevId: 721ca73bae
5 years ago
Megvii Engine Team
649e4dd750
test(cuda): fix test for cu111
GitOrigin-RevId: 04fe5eb23f
5 years ago
Megvii Engine Team
05692d058f
refactor(functional): move matmul dot svd to math pkg
GitOrigin-RevId: 15eb08bacb
5 years ago
Megvii Engine Team
02df634da2
fix(mge/quantization): set ``q_dict`` as an instance property
GitOrigin-RevId: 2f32008aad
5 years ago
Megvii Engine Team
ccbc676158
Revert "fix(whl): append buildid to wheel version"
This reverts commit 00dabfa147 .
GitOrigin-RevId: 87e11e21be
5 years ago
Megvii Engine Team
0e3a6329ff
build(cuda): support cu111 build
GitOrigin-RevId: b3067ba4d5
5 years ago
Megvii Engine Team
cd02d7c85e
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
5 years ago
Megvii Engine Team
e82fa4ec23
fix(gopt): using new_inp for build_chain in DelayBroadcast pass
GitOrigin-RevId: efc6377197
5 years ago
Megvii Engine Team
a09fc5f784
fix(mgb/serialization): disable inplace arith graph opt in graph load
GitOrigin-RevId: d63baf8356
5 years ago
Megvii Engine Team
fc8d13cda3
fix(build): fix linkage on ubuntu1604
GitOrigin-RevId: ee2a8360b0
5 years ago
Megvii Engine Team
43b4d4a430
feat(imperative/test): add more correctness tests
GitOrigin-RevId: 2ef6027bb1
5 years ago
Megvii Engine Team
91a3580f75
refactor(mge/cgtools): remove load_and_inference and use GraphInference
GitOrigin-RevId: 0e688ebd59
5 years ago
Megvii Engine Team
4485e780ae
fix(mge/functional): fix linspace device and open other trace tests
GitOrigin-RevId: 4667c4adec
5 years ago
Megvii Engine Team
334eda8717
refactor(mge): test trace inside opr_test
GitOrigin-RevId: 2cf1135c1c
5 years ago
Megvii Engine Team
938152af63
fix(mge/functional): convert input type to float32 for more elemwise op
GitOrigin-RevId: cf3bf8cb80
5 years ago
Megvii Engine Team
194660462c
feat(mge/funcitonal): add cvt_color opr python interface
GitOrigin-RevId: 29e069fb23
5 years ago
Megvii Engine Team
412d1f0cdc
fix(mge/tensor): implement abtract method to fix lint errors
GitOrigin-RevId: d53f2eac6a
5 years ago
Megvii Engine Team
dbb3dd681f
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
5 years ago
Megvii Engine Team
ff05667b48
test(dist): refactor distributed test with fixtures
GitOrigin-RevId: e69acb72da
5 years ago
Megvii Engine Team
9fb8444d24
fix(imperative): catch python exception in c++
GitOrigin-RevId: 16a2abfdad
5 years ago
Megvii Engine Team
e3a3e0cd3c
feat(mge/functional): add warp_affine
GitOrigin-RevId: 2b333ccd12
5 years ago
Megvii Engine Team
8e50a6daa7
feat(mge/utils): add GraphInference in cgtools
GitOrigin-RevId: 72f2201169
5 years ago
Megvii Engine Team
adc49de803
feat(mge/imperative): expose c++ tensor reference count
GitOrigin-RevId: 1940881adc
5 years ago
Megvii Engine Team
fe1680b378
fix(imperative/ops): improve infer_output_attrs for broadcast
GitOrigin-RevId: 6b7ed55769
5 years ago
Megvii Engine Team
27a9326c66
fix(mge/module): fix frozen batch norm
GitOrigin-RevId: 143d468a37
5 years ago
Megvii Engine Team
c3ba02801a
chore(mge): correct license headers
GitOrigin-RevId: 2dd4438f90
5 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
5 years ago
Megvii Engine Team
9a09f761bb
fix(whl): append buildid to wheel version
GitOrigin-RevId: 00dabfa147
5 years ago
Megvii Engine Team
af5ab4d976
fix(imperative/tensor): fix Py_DECREF on uninitialized pointer
GitOrigin-RevId: 889255537e
5 years ago
Megvii Engine Team
10d9d77e41
refactor(mge/distributed): remove get_free_ports func
GitOrigin-RevId: b85c4885be
5 years ago
Megvii Engine Team
0adf49b137
fix(mge/distributed): fix deadlock by mixing thread and fork
GitOrigin-RevId: c138cb9c28
5 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
5 years ago
Megvii Engine Team
4aaae9950b
feat(mge/functional): add python wrapper to resize opr
GitOrigin-RevId: b7cc6dd829
5 years ago