Megvii Engine Team
|
6ac68090c8
|
refactor(mgb/opr): make trt batch flag only depend on inputs dimension
GitOrigin-RevId: f2f1a10762
|
4 years ago |
Megvii Engine Team
|
38fdd762fa
|
build(cuda): link to cuda_stub
GitOrigin-RevId: 073d3313cf
|
4 years ago |
Megvii Engine Team
|
14a089c49d
|
fix(dnn): change ci to cudnn804, reopen testcase
GitOrigin-RevId: 90713a801b
|
4 years ago |
Megvii Engine Team
|
bcbfbbd743
|
fix(mgb): fix TensorRT runtime opr profiling
GitOrigin-RevId: 3545aa53b2
|
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
|
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
|
0e3a6329ff
|
build(cuda): support cu111 build
GitOrigin-RevId: b3067ba4d5
|
4 years ago |
Megvii Engine Team
|
364afec033
|
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
|
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
|
9e9e8ca0fe
|
fix(mgb): fix some compile errors
GitOrigin-RevId: ec38f8e15a
|
4 years ago |
Megvii Engine Team
|
ab3efab066
|
fix(mgb): add compute capability check
GitOrigin-RevId: fba819afb3
|
4 years ago |
Megvii Engine Team
|
cb5c5fd5dc
|
feat(mgb/opt): add nchw->nchw4 for tensorrt replace pass
GitOrigin-RevId: db114549be
|
5 years ago |
Megvii Engine Team
|
765c9d354c
|
fix(mgb/trt): change log level of trt log
GitOrigin-RevId: 5f5495e303
|
5 years ago |
Megvii Engine Team
|
a7e28712b5
|
fix(mgb): print verbose logs as debug log in TensorRT logger
GitOrigin-RevId: e197d565cb
|
5 years ago |
Megvii Engine Team
|
f91881ffdc
|
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
|
5 years ago |