Megvii Engine Team
c53cad2049
feat(cmake): format all cmake file
GitOrigin-RevId: 0a4ecab99b
3 years ago
Megvii Engine Team
d90cb7763c
feat(src/core): record support change ptr basic
GitOrigin-RevId: b12f1c4a66
3 years ago
Megvii Engine Team
369c2ccc5a
style(all): reformat c++ code
GitOrigin-RevId: 3ffd1b211f
3 years ago
Megvii Engine Team
f76a2cc2c6
feat(mge/opr): add silu and gelu
GitOrigin-RevId: 75aa42947e
3 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
55974e8cf9
feat(log): opt log
* opt log at release mode
* add MGE_OVERRIDE_LOG_LEVEL for runtime debug
//! env to config LogLevel
//! DEBUG = 0, INFO = 1, WARN = 2, ERROR = 3, NO_LOG = 4
//! for example , export MGE_OVERRIDE_LOG_LEVEL=0, means set LogLevel to DEBUG
GitOrigin-RevId: 16cd674c56
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
8f7f52ae4d
feat(jit): add memfwd in jit executor opr
GitOrigin-RevId: b58860bbe8
4 years ago
Megvii Engine Team
38ea5f1ba0
fix(mgb/jit): lower FuncOp to GPUFuncOp directly
GitOrigin-RevId: d7c9c0f548
4 years ago
Megvii Engine Team
d2910f7ef5
fix(mgb/jit): add bind_shape feature to MLIRCompiler
GitOrigin-RevId: bec6796fbf
4 years ago
Megvii Engine Team
368c18607f
fix(mgb/jit): find cuda include path correctly
GitOrigin-RevId: 5ced4206a7
4 years ago
Megvii Engine Team
cb59c27835
feat(mlir/ir): add more op definitions
GitOrigin-RevId: 1e1285ef41
4 years ago
Megvii Engine Team
88e918e261
feat(mgb/jit): add scf.ForOp in MgbToGpuLoweringPass
GitOrigin-RevId: 3cdae27c37
4 years ago
Megvii Engine Team
f7731bd437
fix(mgb/jit): fix a pointer bug in mlir executable_cuda
GitOrigin-RevId: 3ec79b7602
4 years ago
Megvii Engine Team
5d0f8da46a
feat(mgb/jit): add Dimshuffle and lowering passes in jit mlir backend
GitOrigin-RevId: ce6f4ea42a
4 years ago
Megvii Engine Team
0007b9e09b
build(third_party): update llvm-project
GitOrigin-RevId: 26bb606606
4 years ago
Megvii Engine Team
404ef808fa
feat(mgb/jit): adapt jit mlir backend to new mgb dialect and add typecvt
GitOrigin-RevId: bd1b80c84f
4 years ago
Megvii Engine Team
d793c87cdd
refactor(mlir/dialect): redefine mgb dialect
GitOrigin-RevId: ac06534746
4 years ago
Megvii Engine Team
9682db9820
feat(mgb): add jit mlir elemwise broadcast
GitOrigin-RevId: 89d5e2f91e
4 years ago
Megvii Engine Team
170897f2e1
feat(mgb/jit): add more elemwise modes for mlir backend
GitOrigin-RevId: 5883c68804
4 years ago
Megvii Engine Team
d47cf332cc
build(third_party): update llvm-project and adapt to mlir interface changes
GitOrigin-RevId: dd45984cca
4 years ago
Megvii Engine Team
5cfb46e174
feat(mgb/jit): add llvm-lit mlir pass check
GitOrigin-RevId: acc798a49f
4 years ago
Megvii Engine Team
23437864f9
fix(mgb/jit): mlir doesn't support broadcast
GitOrigin-RevId: 08bfc4c34a
4 years ago
Megvii Engine Team
f87bba6806
feat(mgb/jit): add scalar support for mlir
GitOrigin-RevId: 27b1649c04
4 years ago
Megvii Engine Team
11b121a7d2
fix(mgb/jit): link libdevice.bc when generate nvvm ir
GitOrigin-RevId: 49289d65c4
4 years ago
Megvii Engine Team
aa7f28b835
fix(mgb/jit): fix gpu kernel args order
GitOrigin-RevId: a7f0e56747
4 years ago
Megvii Engine Team
9767ca8f19
feat(mgb/jit): refactor code and add more elemwise mode
GitOrigin-RevId: eb6bcadf54
4 years ago
Megvii Engine Team
f9109bed8b
feat(mge): add mlir cmake
GitOrigin-RevId: e86db34b18
4 years ago
Megvii Engine Team
a51d5b4c31
feat(mgb/jit): add mlir backend for cpu and cuda
GitOrigin-RevId: 814fed047e
4 years ago
Megvii Engine Team
e258812f12
feat(dnn): add bool dtype
GitOrigin-RevId: 98c8a092b4
4 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
20450817da
fix(mge/sdk): fix README.md and add missed example
GitOrigin-RevId: 0836029312
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