Megvii Engine Team
0c3699ff55
test(imperative/python): fix testcase for magicmind runtime module
GitOrigin-RevId: baf2f72f01
3 years ago
Megvii Engine Team
0d37bfb0cd
test(lite): add ax model test
GitOrigin-RevId: 495cbdd397
3 years ago
Megvii Engine Team
f00741dad5
feat(build): bazel build support flatbuffer serialization
GitOrigin-RevId: 4755c27e06
3 years ago
Megvii Engine Team
012de7695d
feat(mgb/gopt): add profiler cache
In order to improve performance of the profiling procedure. Make layout transform testcase stable. The profiling result in ci environment will be cached in files.
GitOrigin-RevId: ba2743f35f
3 years ago
Megvii Engine Team
bbaf524f78
feat(ci/windows): fix windows ci stability
GitOrigin-RevId: 41366fd50b
3 years ago
Megvii Engine Team
4eda338876
feat(dnn/cuda): generate cutlass kimpls using cmake and bazel
GitOrigin-RevId: da3bcfb85a
4 years ago
Megvii Engine Team
d28eba4ea5
feat(dnn/cuda): add cutlass impls for int4 conv bias
GitOrigin-RevId: 878bb8c955
4 years ago
Megvii Engine Team
12a0e61542
feat(dnn/cuda): add cuda elemwise int4
GitOrigin-RevId: 8a9aaec328
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
5d350fc843
feat(dnn/cuda): add deconv int8 and fix cutlass conv wrapper base on modify cutlass 2.4
GitOrigin-RevId: 49e0565e8a
4 years ago
Megvii Engine Team
9da26407ea
feat(ci): and model to test algo policy compatible
GitOrigin-RevId: e58caf08c8
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
1ee331dd91
fix(git): fix ci resource lfs, caused by f5993
GitOrigin-RevId: 12e443512d
4 years ago
Megvii Engine Team
03c921f7c4
feat(dnn/cuda): add cutlass matmul impls
GitOrigin-RevId: 619c8c299c
4 years ago
Megvii Engine Team
193e77d4f7
misc(megdnn): misc
GitOrigin-RevId: b321f9b983
4 years ago
Megvii Engine Team
43b4d4a430
feat(imperative/test): add more correctness tests
GitOrigin-RevId: 2ef6027bb1
4 years ago
Megvii Engine Team
abe3c165ba
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
4 years ago
Megvii Engine Team
c008cf3757
refactor(mge): delete old dump_with_testcase_mge
GitOrigin-RevId: 3b58b4acd9
4 years ago
Megvii Engine Team
a51d5b4c31
feat(mgb/jit): add mlir backend for cpu and cuda
GitOrigin-RevId: 814fed047e
4 years ago
Megvii Engine Team
69fe5ab3b3
feat(dnn/cuda): add conv2d-sass-kernel
GitOrigin-RevId: f284d5a4ce
5 years ago
Megvii Engine Team
f91881ffdc
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
5 years ago