Megvii Engine Team
fd6f8e58b0
feat(mgb/dtype): add dtype qint1
GitOrigin-RevId: abe9fb68b1
3 years ago
Megvii Engine Team
2a900a69cb
perf(imperative): improve reduce op performance
GitOrigin-RevId: 26d982a7b8
3 years ago
Megvii Engine Team
72a70dd6a7
perf(imperative): specialize convolution implementation
GitOrigin-RevId: 33634c550f
3 years ago
Megvii Engine Team
2b80806f21
perf(imperative/src): improve dot performance
GitOrigin-RevId: 35b5bd164f
3 years ago
Megvii Engine Team
e400b7ffe5
perf(imperative): enable memory forwarding for imperative
GitOrigin-RevId: 7c1993979c
3 years ago
Megvii Engine Team
95ac055538
feat(dnn,mgb,imperative): add diag opr implement
GitOrigin-RevId: 43016ffa2b
3 years ago
Megvii Engine Team
cbbca5fb10
feat(mge): add softmax op use cudnn api
GitOrigin-RevId: 7734ebf8c4
3 years ago
Megvii Engine Team
20b42a8c3b
fix(dnn): add naive lstm kernel
GitOrigin-RevId: f08ef810cf
3 years ago
Megvii Engine Team
2faa6ea5a9
Merge pull request #213 from kxz18:rnn
GitOrigin-RevId: 9e9215c115
3 years ago
Megvii Engine Team
1999307015
feat(mgb/opr): add dropout kernel
GitOrigin-RevId: d248bd2005
3 years ago
Megvii Engine Team
a93741815b
feat(mgb/opr): add layernorm forward and backward kernel
GitOrigin-RevId: 0cd484e753
3 years ago
Megvii Engine Team
93310c0e4b
fix(mgb/gopt): fix cpu global layout transform fastrun error
GitOrigin-RevId: ea254297e5
3 years ago
kxz@thumt102-1
8f48da7ffe
feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm
3 years ago
Megvii Engine Team
2881934cb8
feat(dnn/check_non_finite): addmul scale to check_non_finite opr
GitOrigin-RevId: c35a219e52
3 years ago
Megvii Engine Team
11d75fecb5
feat(dnn/check_non_finite): add batch check_non_finite
GitOrigin-RevId: e108133282
3 years ago
Megvii Engine Team
3d45d35241
feat(mgb/gopt): profiler support checking algo availability
GitOrigin-RevId: 39cad612cc
3 years ago
Megvii Engine Team
3116e128c5
fix(ci/integration_test): fix benchmark torch version
GitOrigin-RevId: bd964ed505
3 years ago
Megvii Engine Team
d90cb7763c
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
3 years ago
Megvii Engine Team
68cdabd288
feat(opr): indexing_multi_axis_vec support nd index
GitOrigin-RevId: 07b1248bdc
3 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
3 years ago
Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
3 years ago
Megvii Engine Team
f5cb21ed3a
fix(mgb/opr): add non finite check
GitOrigin-RevId: a9fcd0a350
3 years ago
Megvii Engine Team
8cb201868e
fix(mgb): fix fastrun cache serialization method
GitOrigin-RevId: cd8bed3cc9
3 years ago
Megvii Engine Team
b3e54eade1
feat(dnn/bn): use new cudnn BN kernel to support NHWC
GitOrigin-RevId: 9d80f2009d
3 years ago
Megvii Engine Team
3977b7aa0b
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
3 years ago
Megvii Engine Team
c33126ab5c
feat(mgb/gopt): add reformat manager
GitOrigin-RevId: b9791b131a
3 years ago
Megvii Engine Team
d69b59035d
feat(dnn): add an get_all_algorithms_safe interface
GitOrigin-RevId: e3734e4531
3 years ago
Megvii Engine Team
694aa1bd92
feat(dnn): add heuristic cache
GitOrigin-RevId: 35e942b5e3
3 years ago
Megvii Engine Team
bc9cfc277a
feat(mgb): add arm resize nchwxx and naive nearest interp
GitOrigin-RevId: d5fbd59a30
3 years ago
Megvii Engine Team
56c1b626bf
refactor(dnn): move arch-dependant code to arch.h
GitOrigin-RevId: 52350144b1
3 years ago
Megvii Engine Team
67575d582c
feat(mge/opr): add interpolate bilinear mode
GitOrigin-RevId: f7023a3fd3
3 years ago
Megvii Engine Team
0558b2123d
feat(mge/opr): add interpolate nearest mode
GitOrigin-RevId: d384b87f50
3 years ago
Megvii Engine Team
55efc8e197
feat(mgb/gopt): add reformat emitter
GitOrigin-RevId: 937b20a57c
4 years ago
Megvii Engine Team
c9d060307f
feat(dnn/common): add named tensor shape
GitOrigin-RevId: 918928b8ba
4 years ago
Megvii Engine Team
d195fdec71
refactor(mgb): refactor has-usable-algo function for global optimizer
GitOrigin-RevId: 6610516650
3 years ago
Megvii Engine Team
eab6afab47
feat(mgb): add padding opr for megbrain
GitOrigin-RevId: 490e0c5d5a
4 years ago
Megvii Engine Team
c88a4e5b32
fix(mgb): fix get env macro
GitOrigin-RevId: 79c602f79c
3 years ago
Megvii Engine Team
1af350c6d2
feat(dnn): add fill kernel
GitOrigin-RevId: d2cee3a7a0
3 years ago
Megvii Engine Team
3eb0505f9b
feat(imperative): add support for quantized conv transpose2d
GitOrigin-RevId: ffd6431299
3 years ago
Megvii Engine Team
43098fb8f1
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
4 years ago
Megvii Engine Team
b078dda90b
feat(mge/random): add some random op and remove random/distrbution.py
GitOrigin-RevId: 4c05ebc266
4 years ago
Megvii Engine Team
f30c0e06a6
feat(mgb/opr): add lsq opr
GitOrigin-RevId: 45494a2b57
4 years ago
Megvii Engine Team
1cfdbc565c
feat(dnn): add deterministic max pooling
GitOrigin-RevId: 9ab4c7a748
4 years ago
Megvii Engine Team
a5060a2bfe
feat(mgb/opr): add check_has_inf kernel and opr
GitOrigin-RevId: 0d042dbfce
4 years ago
Megvii Engine Team
7b2a76d1ee
refactor(mgb): make conv handle noncontiguous tensors
GitOrigin-RevId: 86282709b3
4 years ago
Megvii Engine Team
40085acbae
fix(mgb): remove unnecessary cudnn8 warning
GitOrigin-RevId: 04cf1bfca9
4 years ago
Megvii Engine Team
2eea00097c
feat(mgb): add fast run batch size graph option
GitOrigin-RevId: 94e333ec80
4 years ago
Megvii Engine Team
86b69cacd0
fix(dnn): fixes for int4
GitOrigin-RevId: 845e164fd3
4 years ago
Megvii Engine Team
12a0e61542
feat(dnn/cuda): add cuda elemwise int4
GitOrigin-RevId: 8a9aaec328
4 years ago
Megvii Engine Team
858261af1f
fix(python_module): fix conversion between numpy-ndarray and mgb tensor for qint4 and quint4
GitOrigin-RevId: 7450c4f25e
4 years ago