Megvii Engine Team
5e7d2a91c2
refactor(mgb): add TensorND::proxy_to_default_cpu
GitOrigin-RevId: 3ab8525f1c
4 years ago
Megvii Engine Team
9e9e8ca0fe
fix(mgb): fix some compile errors
GitOrigin-RevId: ec38f8e15a
4 years ago
Megvii Engine Team
d782edf80f
refactor(mgb): decouple node insertion from ComputingGraphImpl
GitOrigin-RevId: 59b45fcb17
5 years ago
Megvii Engine Team
d42cf4cd65
refactor(mgb): replace static_cast<ComputingGraphImpl*> with a checked version
GitOrigin-RevId: d05b114668
5 years ago
Megvii Engine Team
55844d3e24
fix(core): fix CPU compnode get_ram_status_byte free_memory error
GitOrigin-RevId: a6d4351840
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
4cf193ab97
fix(mgb/opr-mm): fix device id in TestOprIORemote.IdentityWithGopt
GitOrigin-RevId: 0184036b8b
4 years ago
Megvii Engine Team
c76789a5c9
feat(mge/imperative): add io remote wrapper
GitOrigin-RevId: 6dae8287d3
4 years ago
Megvii Engine Team
9006d76bea
fix(serialization): do not gate enum entries by macro
GitOrigin-RevId: aa27ecbf21
5 years ago
Megvii Engine Team
ad4928c995
fix(serialization): remove #include flatbuffers/flatbuffers.h in public header
Removes the need of Flatbuffers for (most) internal users.
GitOrigin-RevId: 6585e9c848
5 years ago
Megvii Engine Team
177c268898
feat(osx/python_whl): fix python pytest failedon osx
GitOrigin-RevId: 97d2b496bb
4 years ago
Megvii Engine Team
4d541b72ce
feat(osx/python_whl): make macos HAPPY for build megbrain python package
GitOrigin-RevId: 7ffd9b9541
5 years ago
Megvii Engine Team
ab3efab066
fix(mgb): add compute capability check
GitOrigin-RevId: fba819afb3
4 years ago
Megvii Engine Team
8070f40aa1
fix(mgb/gopt): fix gopt nchwxx convert elemwise and reshape
GitOrigin-RevId: 982dee36e1
4 years ago
Megvii Engine Team
b38e8225be
refactor(mgb/opr-mm): update megray communicator init interface and fix ci
GitOrigin-RevId: 55c59879f2
4 years ago
Megvii Engine Team
5e912eddbd
fix(mgb/opr-mm): fix grad func of reduce and gather
GitOrigin-RevId: 4687faef99
4 years ago
Megvii Engine Team
e3e981ccf0
test(mge/collective_comm): fix collective_comm test and add data parallel test
GitOrigin-RevId: 9209e77973
4 years ago
Megvii Engine Team
d3d9018f8d
feat(imperative/opr-mm): add broadcast
GitOrigin-RevId: 83640255c7
4 years ago
Megvii Engine Team
3bd8ef3589
feat(mgb/compnode): add atlas compnode
GitOrigin-RevId: 19f3c33003
5 years ago
Megvii Engine Team
7c2782366b
feat(core): redact exception messages if logging is disabled
GitOrigin-RevId: 1d6bda2380
5 years ago
Megvii Engine Team
6137adc05a
feat(core): move more exceptions to base exception
mgb::cg::GraphError
mgb::NumRangeChecker::Error
mgb::VarSanityCheck::Error
GitOrigin-RevId: 4d79e3629c
5 years ago
Megvii Engine Team
49f02e71fc
chore(format): format megbrain/exception.h
GitOrigin-RevId: 744ed23b2d
5 years ago
Megvii Engine Team
9667d93cfe
feat(core): internal changes to exception types
See comments.
GitOrigin-RevId: ecd3c22faa
5 years ago
Megvii Engine Team
cf3a55ce17
fix(mgb/opr-mm): remove PeerDesc from RemoteSend and RemoteRecv
GitOrigin-RevId: b7a7bbd0da
5 years ago
Megvii Engine Team
d53dab2fb2
fix(build): export target megengine
GitOrigin-RevId: 6daf6b2648
5 years ago
Megvii Engine Team
a808f807c6
feat(externcopr/nvidaopticalflow): fix output and fix cu context issue on
diff thread, nvof loader outptr share with mgb ptr, upgrade lastest nvof sdk
GitOrigin-RevId: 6e003c6428
5 years ago
Megvii Engine Team
486cbdea8b
fix(mgb/opt): nchw to nchw4 pass suppport ic less than 4
GitOrigin-RevId: a3c205f38f
5 years ago
Megvii Engine Team
cb5c5fd5dc
feat(mgb/opt): add nchw->nchw4 for tensorrt replace pass
GitOrigin-RevId: db114549be
5 years ago
Megvii Engine Team
2e70cf1dd0
feat(mgb/opt): add nchw->nchw4 in tensorcore pass
GitOrigin-RevId: 755f8dfefe
5 years ago
Megvii Engine Team
1e8337f14a
fix(mgb/gopt): remove redundant reshape in nchw->nchw4 pass
GitOrigin-RevId: 0f5c7c3e48
5 years ago
Megvii Engine Team
946a340c3d
feat(ci/midout): opt midout and add midout ci
GitOrigin-RevId: 1e5fe75255
5 years ago
Megvii Engine Team
90aa65d98d
fix(mgb/serialization): fix multiple graph load error
GitOrigin-RevId: 89414b014b
5 years ago
Megvii Engine Team
6972fc7d75
feat(opr/standalone): import NMSKeep from MegSkull contrib
1. clang-format.
2. Use name "NMSKeepMGB" during serialization to avoid conflict with
MegSkull contrib one.
GitOrigin-RevId: 627c60e482
5 years ago
Megvii Engine Team
6d367454cf
feat(mge/opr-mm): add param local_grad for collective_comm opr
GitOrigin-RevId: cc120cfb55
5 years ago
Megvii Engine Team
0ccb965c8e
fix(mgb/gopt): fix convert format nchw->nchw4 pass
GitOrigin-RevId: 1813753b14
5 years ago
Megvii Engine Team
7ae05ac886
feat(imperative): merge common c++ code to megbrain
GitOrigin-RevId: d093778e10
5 years ago
Megvii Engine Team
5d7fdd4762
fix(brain): fix winodws build
GitOrigin-RevId: 74eaea594e
5 years ago
Megvii Engine Team
318f6d75fa
fix(mge): fix some warnings
GitOrigin-RevId: 38b285f991
5 years ago
Megvii Engine Team
1a70842eac
fix(serialization): fix model compatibility
GitOrigin-RevId: 1310cdc182
5 years ago
Megvii Engine Team
328fb36f17
feat(mgb/opr-mm): add Scatter, Gather, AllToAll oprs
GitOrigin-RevId: f75169ecd6
5 years ago
Megvii Engine Team
786afef461
feat(build): install CMake config module and pkg-config descriptor
Also, upgrade to CMake 3.13.
The commit also contains significant refactors, as otherwise it is not
possible to properly export target `megengine` to
MegEngine-targets.cmake:
1. Optionally use system provided Flatbuffers.
2. Optionally use system provided MKL-DNN (Tested with Debian).
3. Refactor megbrain and megdnn targets into object libraries.
4. Set different path in BUILD_INTERFACE and INSTALL_INTERFACE of
various target_include_directories.
5. Specify PUBLIC/PRIVATE on various target_link_libraries.
GitOrigin-RevId: df118a879e
5 years ago
Megvii Engine Team
4d35397bdf
fix(dnn/fallback): fix conv1x1/im2col usable and fuse-conv-bias get fp32xfp32-->qint8 error
GitOrigin-RevId: 5a3bfedd8a
5 years ago
Megvii Engine Team
12dc36a6ab
feat(mgb/gopt): add interface to reproducible
GitOrigin-RevId: f341bea40b
5 years ago
Megvii Engine Team
09d2b7c3fe
fix(core): make the semantics of instance id clear and correct
GitOrigin-RevId: 2232195c50
5 years ago
Megvii Engine Team
273f891b55
fix(mgb/gopt): fix run-time winograd-transform and nchwxx error
GitOrigin-RevId: aca796f17d
5 years ago
Megvii Engine Team
02abc36ea6
fix(mbg/arm_common): fix nchw44-dot misc issue
GitOrigin-RevId: f870ad964c
5 years ago
Megvii Engine Team
9ed3882a94
fix(opr/dnn): fix winograd fast run mismatch
GitOrigin-RevId: d308085b9f
5 years ago
Megvii Engine Team
18be23f328
fix(mbg/gopt): fix nchwxx gopt with no fuse conv_bias and winograd
fast-run
GitOrigin-RevId: 49ccbdf2d4
5 years ago
Megvii Engine Team
f4ead78852
feat(mgb): static allocation with given padding
GitOrigin-RevId: fdf2de8ad6
5 years ago