Megvii Engine Team
|
76fa71573b
|
feat(dnn/cuda): add cutlass nchw4 convolution
GitOrigin-RevId: 93c9b212f4
|
4 years ago |
Megvii Engine Team
|
a558d4a253
|
fix(mgb/atlas): remove unnessary setdevice
GitOrigin-RevId: 7492de1f01
|
4 years ago |
Megvii Engine Team
|
1fe8a21299
|
fix(mge): fix sublinear memory in jit.trace
GitOrigin-RevId: 190a330a8c
|
4 years ago |
Megvii Engine Team
|
651920c7ad
|
fix(dnn): fix nchw88 winograd weight preprocess
GitOrigin-RevId: 03c0c52aba
|
4 years ago |
Megvii Engine Team
|
d06f248d90
|
fix(whl/mgb/imperative): fix symbols conflict runtime crash
GitOrigin-RevId: d60c4052c1
|
4 years ago |
Megvii Engine Team
|
0e82b959a1
|
feat(mge/imperative): add sublinear options
GitOrigin-RevId: f0e917f716
|
4 years ago |
Megvii Engine Team
|
50d5421aa7
|
fix(mgb/zmq): fix unused-result when compiling with c++17
GitOrigin-RevId: cbdffd39c6
|
4 years ago |
Megvii Engine Team
|
16324e3076
|
feat(dnn/cuda): add remap backward
GitOrigin-RevId: 1b1bcf5db3
|
5 years ago |
Megvii Engine Team
|
968f74ce88
|
chore(mgb): add no_force_inplace option to ComputingGraph
GitOrigin-RevId: 350c90fb86
|
4 years ago |
Megvii Engine Team
|
6e882c1a86
|
feat(whl/imperative): compat for build python whl imperative and legacy runtime
GitOrigin-RevId: 7f6629ae1f
|
4 years ago |
Megvii Engine Team
|
7f857bd471
|
feat(mgb/rocm): add cmake for rocm and fix compile errors and bn
GitOrigin-RevId: c73ed4adc3
|
4 years ago |
Megvii Engine Team
|
9510136223
|
fix(mgb/rocm): remove begin-internal of rocm
GitOrigin-RevId: 1523833fcb
|
4 years ago |
Megvii Engine Team
|
6b380e8965
|
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
|
4 years ago |
Megvii Engine Team
|
2dbe8194ad
|
fix(mge/opr): fix reduction static infer value
GitOrigin-RevId: 6623d4719a
|
4 years ago |
Megvii Engine Team
|
c20d4cc6dc
|
feat(dnn): fix opt pass nchw44 can not dump resnet
GitOrigin-RevId: 28e5c37f53
|
4 years ago |
Megvii Engine Team
|
00ef677249
|
fix(mgb): remove internal for cambricon and atlas
GitOrigin-RevId: 861e349eb4
|
4 years ago |
Megvii Engine Team
|
bca00f2e22
|
fix(dnn): midout at where neccessary in megdnn
GitOrigin-RevId: 191334bd96
|
4 years ago |
Megvii Engine Team
|
a1e6720756
|
feat(dnn): enable bool comparison
GitOrigin-RevId: 735693b81e
|
4 years ago |
Megvii Engine Team
|
4a178a8dba
|
feat(windows/cuda/cmake): support cmake cuda build on windows
GitOrigin-RevId: 4d9832e559
|
4 years ago |
Megvii Engine Team
|
1915593b6c
|
fix(version_depend): add a fake version on dev
make dev useable on all megskull version
GitOrigin-RevId: 8c4739d470
|
4 years ago |
Megvii Engine Team
|
56381f808b
|
fix(dnn/arm): use vcvtq_f32_s32 for all arm code
GitOrigin-RevId: 27effe7d24
|
4 years ago |
Megvii Engine Team
|
1173205726
|
fix(gopt): nchw_nchwxx useable and opt pass use nchw_nchwxx_valid
GitOrigin-RevId: 60942aca5b
|
4 years ago |
Megvii Engine Team
|
eb18eba87d
|
fix(gopt): fix nchw44 nchw44_dot gopt test
GitOrigin-RevId: 06b38dcd30
|
4 years ago |
Megvii Engine Team
|
eab7ab0530
|
fix(gopt): gen nchw_nchw44 when kernel is optimized
GitOrigin-RevId: 89083be200
|
4 years ago |
Megvii Engine Team
|
777f3ea970
|
refactor(gopt): format code
GitOrigin-RevId: 9d5c87000f
|
4 years ago |
Megvii Engine Team
|
30ce3c60bd
|
Revert "fix(mgb/opr): change EQ opr's backward_graph to nullptr instead of InvalidGrad"
This reverts commit 6355337641 .
GitOrigin-RevId: 6897f6d45d
|
4 years ago |
Megvii Engine Team
|
14e71b551b
|
feat(imperative): add helper for dnn opr caller
GitOrigin-RevId: 828cd69803
|
4 years ago |
Megvii Engine Team
|
230ab45a1e
|
fix(mgb/naive): fix naive convolution no dispatch kernel in handle
GitOrigin-RevId: 4038fe23a4
|
4 years ago |
Megvii Engine Team
|
1bce857cb8
|
fix(mgb/opr-mm): use comp_node of config as default in CollectiveComm
GitOrigin-RevId: 6b43c9fc93
|
4 years ago |
Megvii Engine Team
|
27205461ae
|
feat(mgb/opr-mm): add register info cache for multi-machine oprs
GitOrigin-RevId: d5ae3c5a7c
|
4 years ago |
Megvii Engine Team
|
96ec586d28
|
fix(dnn): fix bool cvt
GitOrigin-RevId: 2f883dcbe0
|
4 years ago |
Megvii Engine Team
|
f829f836b9
|
test(mgb/index): add empty index desc tests
GitOrigin-RevId: 1a71ad3ede
|
4 years ago |
Megvii Engine Team
|
e73f2799d0
|
fix(mgb/index): enable index desc empty
GitOrigin-RevId: 4f0ab7c6e7
|
4 years ago |
Megvii Engine Team
|
ff60fdb82d
|
feat(dnn): add bool type cvt on gpu
GitOrigin-RevId: ab0fecf368
|
4 years ago |
Megvii Engine Team
|
e8571cca51
|
fix(mgb/cuda): fix cuda host alloc set device
GitOrigin-RevId: f4756e8981
|
4 years ago |
Megvii Engine Team
|
f7b5eced23
|
refactor(mgb/opr-mm): set False as default value of local_grad
GitOrigin-RevId: 2f9603b087
|
4 years ago |
Megvii Engine Team
|
c7b6ef35c1
|
feat(dnn/cuda): add warp perspective backward mat idx
GitOrigin-RevId: b4b494bb69
|
5 years ago |
Megvii Engine Team
|
09b5f3d434
|
fix(mgb/core): fix multi thread pool deactive and multi thread conflict
GitOrigin-RevId: 36787a08a5
|
4 years ago |
Megvii Engine Team
|
ef239f835f
|
feat(windows/python_whl): make windows HAPPY for build megbrain python package
GitOrigin-RevId: 92b2c07bf9
|
4 years ago |
Megvii Engine Team
|
e258812f12
|
feat(dnn): add bool dtype
GitOrigin-RevId: 98c8a092b4
|
4 years ago |
Megvii Engine Team
|
734c498d27
|
perf(mgb/core): improve DevMemAlloc when it has single stream
GitOrigin-RevId: 61874faa6d
|
4 years ago |
Megvii Engine Team
|
39bd66fc63
|
fix(mgb): fix TensorRT missing cudaSetDevice
GitOrigin-RevId: 40eb119e48
|
4 years ago |
Megvii Engine Team
|
ab9dfbcefc
|
test(mgb): fix tensorrt tests missing cudaSetDevice
GitOrigin-RevId: faeb6ae070
|
4 years ago |
Megvii Engine Team
|
b43fb1a97c
|
perf(mgb): add CUDA host memory allocator
test(mgb): add SimpleCachingAlloc test
GitOrigin-RevId: 17f381e4ac
|
4 years ago |
Megvii Engine Team
|
2afceb4187
|
fix(mgb/atlas): use dyn output alloc if enable dynamic batchsize
GitOrigin-RevId: 45a6c6ad51
|
4 years ago |
Megvii Engine Team
|
6bcc6faec8
|
feat(mge/imperative/opr): modify batch_norm to support frozen BN
fix(mge/imperative): cmake uses MGE_BUILD_IMPERATIVE_RT flag
GitOrigin-RevId: 8ea21af9da
|
4 years ago |
Megvii Engine Team
|
54d18115b6
|
fix(imperative): fix grad of BatchNorm
GitOrigin-RevId: 1e8d8afaf2
|
4 years ago |
Megvii Engine Team
|
80c4705317
|
perf(mgb): use midout in megbrain to reduce binary size
GitOrigin-RevId: ddc8af79af
|
4 years ago |
Megvii Engine Team
|
4348960c40
|
fix(mge/gopt): fix fp16 compute mode
GitOrigin-RevId: 350625d1aa
|
4 years ago |
Megvii Engine Team
|
9f4060b050
|
fix(mgb/gopt): add ShuffleShuffleRemovePass assert
GitOrigin-RevId: 69e07c3197
|
4 years ago |