Megvii Engine Team
4f28e14684
fix(dnn): fix compatibility broken of convolution format
GitOrigin-RevId: c17f02d03d
3 years ago
Megvii Engine Team
f642b05e17
test(mge): update traced_module unit test
GitOrigin-RevId: 3948d50d79
3 years ago
Megvii Engine Team
fb20cb36a0
docs(mge/traced_module): update traced_module api doc
GitOrigin-RevId: 19a95d26c7
3 years ago
Megvii Engine Team
c7a8d945c7
fix(mge/traced_module): let graph record total_id
GitOrigin-RevId: f99178f3ac
3 years ago
Megvii Engine Team
8b40f57738
feat(mgb/dnn): add conv1x1 algo for matrix mul
GitOrigin-RevId: 585b2c045a
3 years ago
Megvii Engine Team
4de62ad6cd
docs(mge/device): fix docstring of get_cuda_compute_capability
GitOrigin-RevId: 4ec04e8b9b
3 years ago
Megvii Engine Team
fb49a2834f
refactor(mgb/dnn): refactor enum used in serializing
GitOrigin-RevId: e57af4a59c
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
103d7f33ba
refactor(dnn/rocm): update hip license header
GitOrigin-RevId: 79d684755d
4 years ago
Megvii Engine Team
5aa52d3863
feat(dnn/rocm): add adaptive pooling opr
GitOrigin-RevId: e844b3e770
3 years ago
Megvii Engine Team
83cf4ee64e
refactor(dnn/rocm): remove some useless includes
GitOrigin-RevId: 3d2c315a36
3 years ago
Megvii Engine Team
323a4642e6
feat(dnn/rocm): add topk opr
GitOrigin-RevId: 5ecb079854
3 years ago
Megvii Engine Team
f4784f4af1
feat(dnn/rocm): add argsort opr
GitOrigin-RevId: b4c3eb4707
3 years ago
Megvii Engine Team
6082c353e7
feat(dnn/rocm): support bool in type_cvt and elemwise
GitOrigin-RevId: ad5ec7bc1c
4 years ago
Megvii Engine Team
0ad377c7cf
fix(mgb/gopt): add error message when input dtype is not equal to param dtype in BN2Elemwise pass
GitOrigin-RevId: 3d09a2a12e
3 years ago
Megvii Engine Team
93f70a9539
feat(src): add MGB_CUSTOM_OP macro
GitOrigin-RevId: beddcf9745
3 years ago
Megvii Engine Team
b3e958d0bc
fix(src): fix the warnings and copy.bara.sky in custom op
GitOrigin-RevId: 4ade455897
3 years ago
Megvii Engine Team
cdb692d2fa
refactor(imperative): add TODO tag for some functions
GitOrigin-RevId: e295a1fa55
3 years ago
Megvii Engine Team
90dd07161c
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
3 years ago
Megvii Engine Team
cbf024bf20
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
3 years ago
Megvii Engine Team
39ba30214d
feat(src): add adaptor between custom op and graph runtime
GitOrigin-RevId: 834c0ac72b
3 years ago
Megvii Engine Team
c9134717c6
feat(src/custom): add tensor and tensor attrs in custom namespace
GitOrigin-RevId: 9f69259205
3 years ago
Megvii Engine Team
43fb2c4afb
feat(opr): let roll support empty IO
GitOrigin-RevId: b9a59b623a
3 years ago
Megvii Engine Team
b2827cb1b1
feat(opr): let Dot, MatrixMul and BatchedMatrixMul support empty input
GitOrigin-RevId: 10a3c5b106
3 years ago
Megvii Engine Team
50f738772b
feat(opr): let Copy and Identity support empty IO
GitOrigin-RevId: 4e49d8eae8
3 years ago
Megvii Engine Team
74cbc10d82
feat(opr): let batchnorm support empty IO
GitOrigin-RevId: 219411c80c
3 years ago
Megvii Engine Team
8ae61752a0
feat(opr): let nms support empty IO
GitOrigin-RevId: 4c51b1aedb
3 years ago
Megvii Engine Team
1a1748daf1
feat(opr): let Argsort support empty IO
GitOrigin-RevId: 05fcac6e47
3 years ago
Megvii Engine Team
7234efe1e7
feat(opr): let random op support empty output
GitOrigin-RevId: a2174975aa
3 years ago
Megvii Engine Team
3bc94738ba
fix(profiler): do not assume clock duration and time_point type
GitOrigin-RevId: 2015bc30bd
3 years ago
Megvii Engine Team
b7e6bd7bba
feat(profiler): custom event support device
GitOrigin-RevId: 4709c44b39
3 years ago
Megvii Engine Team
defbc20efb
fix(profiler): fix profiler in dtr
GitOrigin-RevId: 79b5a6b526
3 years ago
Megvii Engine Team
d4c71f92dc
perf(profiler): reduce profiler host overhead
GitOrigin-RevId: 92cea560f5
3 years ago
Megvii Engine Team
8fddd8089d
fix(profiler): respect record_device option
GitOrigin-RevId: 7c9a8cfba7
4 years ago
Megvii Engine Team
59d5976622
fix(profiler): collect records when thread exit
GitOrigin-RevId: fb796dd61c
4 years ago
Megvii Engine Team
a841d8e76f
refactor(profiler): refactor event processing
GitOrigin-RevId: 26f0c5a6e1
4 years ago
Megvii Engine Team
dff7719e87
feat(mge/distributed): add preload host data with op
fix(mge/distributed): change api name with preload
fix(mge/distributed): fix recursive model in preload tensor
fix(mge/distributed): fix recursive when cache contain None
GitOrigin-RevId: 80e2a6dd70
3 years ago
Megvii Engine Team
8b94f49328
fix(dnn/cuda): fix elemwise and relayout int4 bug when last shape is 1
GitOrigin-RevId: e7d64c4987
3 years ago
Megvii Engine Team
694aa1bd92
feat(dnn): add heuristic cache
GitOrigin-RevId: 35e942b5e3
3 years ago
Megvii Engine Team
ca4374b6e2
perf(subgraph): remove unnecessary sync
GitOrigin-RevId: 1546286706
3 years ago
Megvii Engine Team
ad079009e1
fix(interpreter): avoid deadlock in GetValue
GitOrigin-RevId: fd43873167
3 years ago
Megvii Engine Team
4688bbabad
refactor(functional): hide functional.vision and replace with functional.nn
GitOrigin-RevId: b6888c7f65
3 years ago
Megvii Engine Team
8796586b89
refactor(functional): import all from metric in nn
GitOrigin-RevId: 41ab78d78d
3 years ago
Megvii Engine Team
fb15b3012c
docs(readme): fix md relative path
GitOrigin-RevId: 27c830e530
3 years ago
Megvii Engine Team
bc9cfc277a
feat(mgb): add arm resize nchwxx and naive nearest interp
GitOrigin-RevId: d5fbd59a30
3 years ago
zhanghaolong
589b427e15
Update README.md
3 years ago
Megvii Engine Team
d5cff81c1e
fix(mge): limit task queue size
GitOrigin-RevId: a25c3390fc
3 years ago
Megvii Engine Team
4ef911361f
docs(readme): update build md
GitOrigin-RevId: 279a202eb2
3 years ago
Megvii Engine Team
bc9c47e77d
fix(mgb): use link.exe when compile windows
GitOrigin-RevId: 0b9073acce
3 years ago
Megvii Engine Team
668a848600
fix(cmake/windows): fix low probability build failed on Windows
GitOrigin-RevId: 21fe964bbb
3 years ago