Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
4 years ago
Megvii Engine Team
3e4e4c4604
feat(mgb/jit): add graph_opt_config and jit_config interfaces
GitOrigin-RevId: 170d9eeab2
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
5 years ago
Megvii Engine Team
8f7f52ae4d
feat(jit): add memfwd in jit executor opr
GitOrigin-RevId: b58860bbe8
5 years ago
Megvii Engine Team
d2910f7ef5
fix(mgb/jit): add bind_shape feature to MLIRCompiler
GitOrigin-RevId: bec6796fbf
5 years ago
Megvii Engine Team
5d0f8da46a
feat(mgb/jit): add Dimshuffle and lowering passes in jit mlir backend
GitOrigin-RevId: ce6f4ea42a
5 years ago
Megvii Engine Team
0007b9e09b
build(third_party): update llvm-project
GitOrigin-RevId: 26bb606606
5 years ago
Megvii Engine Team
404ef808fa
feat(mgb/jit): adapt jit mlir backend to new mgb dialect and add typecvt
GitOrigin-RevId: bd1b80c84f
5 years ago
Megvii Engine Team
9682db9820
feat(mgb): add jit mlir elemwise broadcast
GitOrigin-RevId: 89d5e2f91e
5 years ago
Megvii Engine Team
170897f2e1
feat(mgb/jit): add more elemwise modes for mlir backend
GitOrigin-RevId: 5883c68804
5 years ago
Megvii Engine Team
5cfb46e174
feat(mgb/jit): add llvm-lit mlir pass check
GitOrigin-RevId: acc798a49f
5 years ago
Megvii Engine Team
23437864f9
fix(mgb/jit): mlir doesn't support broadcast
GitOrigin-RevId: 08bfc4c34a
5 years ago
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
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
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
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