Megvii Engine Team
a3ea1f153c
feat(mgb/opr): add fast profile and combined Execution strategy
GitOrigin-RevId: 843dc3a790
4 years ago
Megvii Engine Team
c8697a7005
feat(imperative/src): python wrapper for cambricon and atlas runtime opr
GitOrigin-RevId: bd969d1339
4 years ago
Megvii Engine Team
d970b85deb
feat(functional): add conv3d
GitOrigin-RevId: c06ac77c6c
4 years ago
Megvii Engine Team
ae3123b315
feat(mge): add python graph for mgb graph editing
GitOrigin-RevId: 6a9d5beba2
4 years ago
Megvii Engine Team
c82d88751a
fix(dnn/cuda): add cuda nchw int8 conv impl with nchw4 to fix cu111 compatibility
GitOrigin-RevId: 771968f9ac
4 years ago
Megvii Engine Team
798ae5e58c
fix(imperative/opr): close SCALAR_IDX warning of IndexingMultiAxisVec for proxy_graph
GitOrigin-RevId: 6e0d888f85
4 years ago
Megvii Engine Team
3fab21ed31
fix(lite): fix BUILD and misc
GitOrigin-RevId: 48aa0e0924
4 years ago
Megvii Engine Team
65c3159cf1
feat(lite): add version interface
GitOrigin-RevId: cd09b271a9
4 years ago
Megvii Engine Team
f2b42bf09e
chore(dotprod): add arm dotprod attribute for easy use
GitOrigin-RevId: 78c3e72218
4 years ago
Megvii Engine Team
c33a717314
feat(dnn): repalce is_reproducible with algo attribute in opencl, cpu, rocm and cuda
GitOrigin-RevId: 86dead0a11
4 years ago
Megvii Engine Team
b0944dc72a
fix(mgb): fix tx1 compile
GitOrigin-RevId: 0eeb62ff73
4 years ago
Megvii Engine Team
0035efa636
fix(mge): limit task queue size
GitOrigin-RevId: 9481d38914
4 years ago
Megvii Engine Team
b9bbf802bd
fix(mgb/opr): fix string append error
GitOrigin-RevId: 295a941edd
4 years ago
Megvii Engine Team
47bdb4ef1d
fix(mgb/core): add MGB_HAVE_THREAD macro to thread_local
GitOrigin-RevId: cfcb50b041
4 years ago
Megvii Engine Team
ad87f78a14
chore(imperative): refine tblgen for generating op name
GitOrigin-RevId: f47ceae726
4 years ago
Megvii Engine Team
4b98c72171
fix(mge/sublinear): change no virtual grad warning to log_debug
GitOrigin-RevId: 6b5808f2d4
4 years ago
Megvii Engine Team
51868533c8
fix(mgb/gopt): fix opt pass elementwise operation shape issue at tranform to NCHW4
GitOrigin-RevId: c0c4e3f82e
4 years ago
Megvii Engine Team
0a191f8d48
fix(misc): some fixes to make faster_rcnn run in no-symbolic-trace
GitOrigin-RevId: 0fc40ad9c8
4 years ago
Megvii Engine Team
6fb19b664b
feat(imperative/src): name operators automatically when tracing
GitOrigin-RevId: ff8eb003c5
4 years ago
Megvii Engine Team
09de5a0725
feat(mgb/serialization): be able to serialize operator names
GitOrigin-RevId: d295abb5da
4 years ago
Megvii Engine Team
bb8f292816
refactor(mge/func): change `conv_execution_strategy` to `execution_strategy`
GitOrigin-RevId: 7aef9935ee
4 years ago
Megvii Engine Team
cf27dd642c
fix(cuda): use cudnn8.0.4 as cu111 default libs
GitOrigin-RevId: 721ca73bae
4 years ago
Megvii Engine Team
649e4dd750
test(cuda): fix test for cu111
GitOrigin-RevId: 04fe5eb23f
4 years ago
Megvii Engine Team
2e4b9a42f7
fix(mgb/gopt): fix folding conv dimshuffle opt pass
GitOrigin-RevId: 878b7de9de
4 years ago
Megvii Engine Team
d62dabe5f8
fix(mgb): fix matmul model compat in flatbuffer
GitOrigin-RevId: 2effad8e5f
4 years ago
Megvii Engine Team
b6376c83a0
fix(mgb/comp_node): use extra physical device to decide whether reuse existed CompNodeImpl
GitOrigin-RevId: 5dddc68a84
4 years ago
Megvii Engine Team
0e3a6329ff
build(cuda): support cu111 build
GitOrigin-RevId: b3067ba4d5
4 years ago
Megvii Engine Team
cd02d7c85e
feat(mge/opr): support deformable conv2d/psroi_pooling
GitOrigin-RevId: 501cadda76
4 years ago
Megvii Engine Team
f3378100f9
feat(mgb): enable output dynamic memory alloc
GitOrigin-RevId: c809629034
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
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
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
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
194660462c
feat(mge/funcitonal): add cvt_color opr python interface
GitOrigin-RevId: 29e069fb23
4 years ago
Megvii Engine Team
dbb3dd681f
refactor(profiler): integrate profiler into interpreter
GitOrigin-RevId: ccc984acbd
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
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
b1806b8432
docs(mgb/opr): improve batch_norm err msg in training mode
GitOrigin-RevId: 4c43bc8b0b
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
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
df976782fa
feat(mge/functional): add matinv
GitOrigin-RevId: d4fa8a8277
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