MegEngine

25 MB

Tree: 9767ca8f19

Author	SHA1	Message	Date
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

11 Commits (9767ca8f19c654a44f9dbd855d2b18794b35455a)