Megvii Engine Team
ea4e6ab93a
fix(mgb/opr): fix shape cache of NvOF
GitOrigin-RevId: 456ba478e9
4 years ago
Megvii Engine Team
bdb853ee6f
fix(mgb): fix extra device malloc when load MultipleDeviceTensorWithFormatHolder
GitOrigin-RevId: adf4a7f77a
4 years ago
Megvii Engine Team
95ac055538
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
4 years ago
Megvii Engine Team
cbbca5fb10
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
4 years ago
Megvii Engine Team
20b42a8c3b
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
4 years ago
Megvii Engine Team
2faa6ea5a9
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
4 years ago
Megvii Engine Team
85ea882cb5
fix(mgb/ops): immutable tensor support empty storage
GitOrigin-RevId: 2851498fce
4 years ago
Megvii Engine Team
1999307015
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
4 years ago
Megvii Engine Team
a93741815b
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
4 years ago
Megvii Engine Team
6011f51001
style(all): fix clang-format for MGB_DEFINE inside another macro
GitOrigin-RevId: 8c2b6a2aed
4 years ago
Megvii Engine Team
e715423f20
feat(src/gopt): add optpass on arm for fusing typecvt and elemwise to elemwise multi type
GitOrigin-RevId: e6bcbbf91b
4 years ago
Megvii Engine Team
d9a46ea47b
fix(dnn): correct behaviour of floor div for int tensor
GitOrigin-RevId: 1444f69cce
4 years ago
Megvii Engine Team
0ad5eeaedd
feat(mgb/gopt): global layout transform support opencl
GitOrigin-RevId: 132605c7d9
4 years ago
kxz@thumt102-1
8f48da7ffe
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
4 years ago
Megvii Engine Team
2881934cb8
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
4 years ago
Megvii Engine Team
b8ccc6a211
fix(mgb): fix loss execution policy after opr shallow copy
GitOrigin-RevId: 4738136e4a
4 years ago
Megvii Engine Team
c27f678230
feat(src/opr): add api of training of cpp and related test
GitOrigin-RevId: befb85fd43
4 years ago
Megvii Engine Team
11556cba13
feat(megbrain): add mc20 bazel
GitOrigin-RevId: 96d57656e4
4 years ago
Megvii Engine Team
b8f810ee31
feat(megbrain): add mc20 runtime opr
GitOrigin-RevId: 2ee07b213b
4 years ago
Megvii Engine Team
67e4e83419
fix(lite): fix the force_output_use_user_specified_memory when out var not supported
GitOrigin-RevId: ffaf4c1416
4 years ago
Megvii Engine Team
accb2d8d47
fix(mgb/serialize): fix flatbuffer compatibility issues
GitOrigin-RevId: e4771d6bc4
4 years ago
Megvii Engine Team
11d75fecb5
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
4 years ago
Megvii Engine Team
0be6ca8899
fix(src/core): fix record change ptr bug on comp node copy
GitOrigin-RevId: 0f68966211
4 years ago
Megvii Engine Team
d90cb7763c
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
4 years ago
Megvii Engine Team
68cdabd288
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
4 years ago
Megvii Engine Team
25ec2530ba
feat(whl/api/lar): enable megengine dll on Windows
1: reduce python whl package size
2: unify api link logic on all OS
3: add option: MGE_WINDOWS_BUILD_WITH_STATIC_CRT
--- default OFF
--- if build CRT(vc runtime) with STATIC with megengine.dll
some CRT api will crash, for example, flush, so if you
build with static megengine, and do not want to install CRT
you can set MGE_WINDOWS_BUILD_WITH_STATIC_CRT TRUE
--- how to install CRT:
https://docs.microsoft.com/en-us/cpp/windows/latest-supported-vc-redist?view=msvc-160
install VC_redist.x64.exe
4: rename megengine_export to megengine_shared(only export needed symbols ),
caused by runtime symbols conflict with pytorch
GitOrigin-RevId: 93d8d80f29
4 years ago
Megvii Engine Team
d9a9d9d49e
fix(imperative/fastrun): set workspace limit for imperative rt
GitOrigin-RevId: 474dc691a3
4 years ago
Megvii Engine Team
a09a2b730d
fix(mgb/opr): fix fastrun workspace limit for imperative rt
GitOrigin-RevId: bd69a82d4c
4 years ago
Megvii Engine Team
5207154e5c
Revert "fix(mgb): fix fastrun no_profiling_on_shape_change"
This reverts commit 2735536424 .
GitOrigin-RevId: 6a641808f0
4 years ago
Megvii Engine Team
5885b137fa
feat(dnn/arm): support layout like NHWC channel like broadcast on arm
GitOrigin-RevId: fb4300004c
4 years ago
Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
4 years ago
Megvii Engine Team
bfb30dcb81
chore(format): fix compile bugs after code format
GitOrigin-RevId: 11a4b06f6f
4 years ago
Megvii Engine Team
f5cb21ed3a
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
4 years ago
Megvii Engine Team
8cb201868e
fix(mgb): fix fastrun cache serialization method
GitOrigin-RevId: cd8bed3cc9
4 years ago
Megvii Engine Team
dc28a43409
fix(mgb/bn): fix empty tensor input problem and other minor problems
GitOrigin-RevId: 1708b76cb8
4 years ago
Megvii Engine Team
270f1aa28f
feat(mgb/serialization): add Accessor for OprLoader to fix BN output compatibility
GitOrigin-RevId: 3b95da02c8
4 years ago
Megvii Engine Team
c0ccd0ea7e
feat(mge/bn): add NHWC support for bn
GitOrigin-RevId: 0a5bb6f72d
4 years ago
Megvii Engine Team
b982be567a
feat(mge/imperative): add permutation support for the tensor
GitOrigin-RevId: 7ed0447bfe
4 years ago
Megvii Engine Team
3977b7aa0b
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
4 years ago
Megvii Engine Team
d69b59035d
feat(dnn): add an get_all_algorithms_safe interface
GitOrigin-RevId: e3734e4531
4 years ago
Megvii Engine Team
93f70a9539
feat(src): add MGB_CUSTOM_OP macro
GitOrigin-RevId: beddcf9745
4 years ago
Megvii Engine Team
b3e958d0bc
fix(src): fix the warnings and copy.bara.sky in custom op
GitOrigin-RevId: 4ade455897
4 years ago
Megvii Engine Team
cdb692d2fa
refactor(imperative): add TODO tag for some functions
GitOrigin-RevId: e295a1fa55
4 years ago
Megvii Engine Team
90dd07161c
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
4 years ago
Megvii Engine Team
39ba30214d
feat(src): add adaptor between custom op and graph runtime
GitOrigin-RevId: 834c0ac72b
4 years ago
Megvii Engine Team
43fb2c4afb
feat(opr): let roll support empty IO
GitOrigin-RevId: b9a59b623a
4 years ago
Megvii Engine Team
b2827cb1b1
feat(opr): let Dot, MatrixMul and BatchedMatrixMul support empty input
GitOrigin-RevId: 10a3c5b106
4 years ago
Megvii Engine Team
50f738772b
feat(opr): let Copy and Identity support empty IO
GitOrigin-RevId: 4e49d8eae8
4 years ago
Megvii Engine Team
74cbc10d82
feat(opr): let batchnorm support empty IO
GitOrigin-RevId: 219411c80c
4 years ago
Megvii Engine Team
8ae61752a0
feat(opr): let nms support empty IO
GitOrigin-RevId: 4c51b1aedb
4 years ago