Megvii Engine Team
e2b79ea00e
feat(mgb): reduce the number of trtruntimeopr create contexts
GitOrigin-RevId: 14e5d1769e
3 years ago
Megvii Engine Team
f4f20046c4
fix(mgb): fix tensorrt runtimeopr get output var shape bug
GitOrigin-RevId: b830706a89
3 years ago
Megvii Engine Team
a404cd7d06
fix(mgb/src): add tensorRT version check
GitOrigin-RevId: 7abfd30cab
3 years ago
Megvii Engine Team
390d2bb545
feat(mgb): tensorrt runtime opr support mutiple profiles
GitOrigin-RevId: 1157d34e4d
3 years ago
Megvii Engine Team
1708ab2ec6
feat(mgb): add tensorrt runtime dynamic batch testcase
GitOrigin-RevId: 36372437ff
3 years ago
Megvii Engine Team
87c845fd61
feat(mgb): tensorrt runtime opr support dynamic batch trt model
GitOrigin-RevId: 7461de704e
3 years ago
Megvii Engine Team
605836a70f
fix(mgb): fix trt5.1 test can not create FlattenConcatPlugin
GitOrigin-RevId: 1b9a80b04e
3 years ago
Megvii Engine Team
84baf3df1b
feat(mgb): add tensorrt plugin support
GitOrigin-RevId: 5428b4f665
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
2cd9823210
fix(mgb/tensorrt): fix trt runtime, padding channel to a multiple of 4 when using kCHW4 IOFormat
GitOrigin-RevId: c5f1ed70da
4 years ago
Megvii Engine Team
6de3e4baa3
refactor(mgb/opr): make trt batch flag only depend on inputs dimension
GitOrigin-RevId: f2f1a10762
4 years ago
Megvii Engine Team
c9348b162f
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