MegEngine

417 Commits

28 Branches

25 MB

4628 Download

Tree: bc95e873ef

Commit Graph

Author SHA1 Message Date

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

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

5 years ago

3 Commits (bc95e873ef8c3fc6a0846c9f40d3ed2a0c01aeb7)