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
Megvii Engine Team
df976782fa
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
4 years ago
Megvii Engine Team
3bda334798
fix(dnn/fallback): fix segmentfault caused by im2col/conv1x1 using
fallback naive matmul.
GitOrigin-RevId: 03ef904b11
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
87ff58f7fc
fix(megdnn): add algo for matmul/batchedmatrixmul of naive and opencl
GitOrigin-RevId: 2409b6ba16
4 years ago
Megvii Engine Team
169fa53d54
fix(mgb): fix execution_policy set of matmul
GitOrigin-RevId: 90f539b0be
4 years ago
Megvii Engine Team
abe3c165ba
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
4 years ago
Megvii Engine Team
1b1ad56a82
fix(mgb/gopt): fix warp fusion opt pass
GitOrigin-RevId: a40bbcd719
4 years ago
Megvii Engine Team
99b176236b
fix(mgb/opr): fix Reduce static value inference
GitOrigin-RevId: 5e5c56064c
4 years ago
Megvii Engine Team
a3caa5d3b7
fix(mgb(dnn)): fix convbias cudnnConvBiasActivation
GitOrigin-RevId: c0e44feffb
4 years ago