MegEngine

661 Commits

28 Branches

25 MB

577 Download

Tree: 9767ca8f19

Commit Graph

Author SHA1 Message Date

Author	SHA1	Message	Date
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	f91881ffdc	MegEngine: Initial commit of MegEngine. GitOrigin-RevId: `f0c8338beb`	6 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

f91881ffdc

MegEngine: Initial commit of MegEngine.

GitOrigin-RevId: f0c8338beb

6 years ago

3 Commits (9767ca8f19c654a44f9dbd855d2b18794b35455a)