MegEngine

25 MB

Tree: 22853fa20c

Author	SHA1	Message	Date
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	6 years ago
Megvii Engine Team	32cd9ca0bd	fix(mgb): remove unnecessary null pointer checks GitOrigin-RevId: `85aa0a9bdb`	6 years ago
Megvii Engine Team	20450817da	fix(mge/sdk): fix README.md and add missed example GitOrigin-RevId: `0836029312`	6 years ago
Megvii Engine Team	5fe7cd1457	fix(mge/jit): fix typo InternalGraphGenerator GitOrigin-RevId: `1ab14acbd2`	6 years ago
Megvii Engine Team	f91881ffdc	MegEngine: Initial commit of MegEngine. GitOrigin-RevId: `f0c8338beb`	6 years ago