Megvii Engine Team
f87bba6806
feat(mgb/jit): add scalar support for mlir
GitOrigin-RevId: 27b1649c04
5 years ago
Megvii Engine Team
11b121a7d2
fix(mgb/jit): link libdevice.bc when generate nvvm ir
GitOrigin-RevId: 49289d65c4
5 years ago
Megvii Engine Team
aa7f28b835
fix(mgb/jit): fix gpu kernel args order
GitOrigin-RevId: a7f0e56747
5 years ago
Megvii Engine Team
9767ca8f19
feat(mgb/jit): refactor code and add more elemwise mode
GitOrigin-RevId: eb6bcadf54
5 years ago
Megvii Engine Team
f9109bed8b
feat(mge): add mlir cmake
GitOrigin-RevId: e86db34b18
5 years ago
Megvii Engine Team
a51d5b4c31
feat(mgb/jit): add mlir backend for cpu and cuda
GitOrigin-RevId: 814fed047e
5 years ago
Megvii Engine Team
e258812f12
feat(dnn): add bool dtype
GitOrigin-RevId: 98c8a092b4
5 years ago
Megvii Engine Team
672d4ad0e0
fix(jit): more testcases on the grad of JITExecutor
GitOrigin-RevId: c3bb405979
5 years ago
Megvii Engine Team
bc95e873ef
fix(jit): fix jit grad
a) fix shape mismatch when take grad of JITExecutor including Dimshuffle
b) avoid redundant computation in the grad of JITExecutor
c) not pass unused vars as inputs to the grad of JITExecutor to save device memory
d) traverse internal graph only once in JITExecutor ctor instead of traverse
whole graph in each call of setup_args()
e) expand the gradient graph into the origin graph if all inputs are const
GitOrigin-RevId: ba6a2b29e9
5 years ago
Megvii Engine Team
09d2b7c3fe
fix(core): make the semantics of instance id clear and correct
GitOrigin-RevId: 2232195c50
5 years ago
luzzyzhang
16f052e916
fix(megdnn): change ver 60 to use cuda capability 50
fix(megdnn): if compute capability less than 60 skip fp16 test
5 years ago
Megvii Engine Team
32cd9ca0bd
fix(mgb): remove unnecessary null pointer checks
GitOrigin-RevId: 85aa0a9bdb
5 years ago
Megvii Engine Team
5fe7cd1457
fix(mge/jit): fix typo InternalGraphGenerator
GitOrigin-RevId: 1ab14acbd2
5 years ago
Megvii Engine Team
f91881ffdc
MegEngine: Initial commit of MegEngine.
GitOrigin-RevId: f0c8338beb
6 years ago