Megvii Engine Team
c51a687cef
chore(mge): update copyright years
GitOrigin-RevId: 46104ac891
4 years ago
Megvii Engine Team
7c09e41f18
refactor(mgb): add circular dependency check
GitOrigin-RevId: 01fdb8684b
4 years ago
Megvii Engine Team
af42ce7e69
fix(megdnn): some fixes of execution policy
GitOrigin-RevId: 920f39bcb6
4 years ago
Megvii Engine Team
7afa422df4
refactor(megdnn): refactor sub opr setter
GitOrigin-RevId: 475afb9c10
4 years ago
Megvii Engine Team
821656aa4b
refactor(megdnn): refactor brute force algo in batched matmul
GitOrigin-RevId: 5c143ab3ac
4 years ago
Megvii Engine Team
08ff62deb6
refactor(megdnn): refactor batched matmul algo in conv bias
GitOrigin-RevId: 64fda611ff
4 years ago
Megvii Engine Team
8773926ef8
refactor(megdnn): refactor matmul algo in conv bias
GitOrigin-RevId: 932f7d6f81
4 years ago
Megvii Engine Team
e4b71bdf64
refactor(megdnn): remove unnessary 1x1 algo
GitOrigin-RevId: 809f9b2bba
4 years ago
Megvii Engine Team
44c8d2d16f
refactor(megdnn): refactor matmul algo in deformable conv
GitOrigin-RevId: 05291baf98
4 years ago
Megvii Engine Team
b04ad06f84
refactor(megdnn): refactor matmul algo in conv backward filter
GitOrigin-RevId: bb65777583
4 years ago
Megvii Engine Team
25089e520e
refactor(megdnn): refactor matmul algo in conv backward data
GitOrigin-RevId: 8de601df6e
4 years ago
Megvii Engine Team
3620a940b5
fix(megdnn): fix test with workspace_limits to zero
GitOrigin-RevId: c4ec323361
4 years ago
Megvii Engine Team
0d720653ac
refactor(megdnn): add default algo for convolution forward
GitOrigin-RevId: a12a7d399a
4 years ago
Megvii Engine Team
659217acd2
refactor(megdnn): refactor bfloat16 convbias to recursive inteface
GitOrigin-RevId: 378194fb7f
4 years ago
Megvii Engine Team
4a1d52c9c6
refactor(megdnn): refactor bfloat16 matmul to recursive inteface
GitOrigin-RevId: 641c508aec
4 years ago
Megvii Engine Team
b8febaf91f
refactor(megdnn): refactor bfloat16 convolutionbackwardfilter to recursive inteface
GitOrigin-RevId: 37c08a5b8b
4 years ago
Megvii Engine Team
f14e0c17e7
feat(mgb): add recursive for fastrun and megdnn test
GitOrigin-RevId: 743846f645
4 years ago
Megvii Engine Team
85fa988348
refactor(dnn): add get_algorithm_from_desc interface
GitOrigin-RevId: 6d211ca167
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
2b8150ab52
fix(dnn): fix bazel build issue for cambricon platform
GitOrigin-RevId: a79e1b6f6a
4 years ago
Megvii Engine Team
3b28af958a
fix(sdk): add disable-optimize-for-inference option
GitOrigin-RevId: 7e6ea13be7
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
0e8b81c20e
fix(dnn/opencl): fix elemwise negative stride support
GitOrigin-RevId: 506d7e6104
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
329306b031
fix(cmake/cuda): fix build at cuda `copy` env caused by b278a69e1
`copy` env vs `normal` env:
case a:`normal` env install CUDNN/TRT by package tool, eg, apt/yum, which
will put include file(cudnn.h/NvInfer.h) to /usr/include etc.
case b: self copy env after export CPATH with cudnn/trt include dir
case c: just copy libs, not export CPATH, only export
TRT_ROOT_DIR/CUDNN_ROOT_DIR
at case a and case b, BUILD system can find cudnn/trt headers, but
at case c, can not find. `FIX` it!
what`s more, recommend install CUDA env by case a and case c, NOT
recommend install by case b, which caused by CUDA SDK CPATH not
friendly to split c env, for example Android AOSP build env
GitOrigin-RevId: cb92123f9e
4 years ago
Megvii Engine Team
8e50a6daa7
feat(mge/utils): add GraphInference in cgtools
GitOrigin-RevId: 72f2201169
4 years ago
Megvii Engine Team
97207d00b0
fix(mgb/opr): fix bazel compile error by comma in mgb_throw_if's format string
GitOrigin-RevId: 72d19aecd1
4 years ago
Megvii Engine Team
5a44255324
fix(mgb/plugin): fix opr footprint for conv with NCHW32_NCHW4 format
GitOrigin-RevId: 9881b7971c
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
b1806b8432
docs(mgb/opr): improve batch_norm err msg in training mode
GitOrigin-RevId: 4c43bc8b0b
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