Megvii Engine Team
4348960c40
fix(mge/gopt): fix fp16 compute mode
GitOrigin-RevId: 350625d1aa
5 years ago
Megvii Engine Team
9f4060b050
fix(mgb/gopt): add ShuffleShuffleRemovePass assert
GitOrigin-RevId: 69e07c3197
5 years ago
Megvii Engine Team
9b0978597c
chore(core): remove index_sequence polyfill for pre-C++14
GitOrigin-RevId: 8010000bc2
5 years ago
Megvii Engine Team
074645d82e
feat(core): add array_skip
GitOrigin-RevId: ae0d12fbb8
5 years ago
Megvii Engine Team
2faab90bd0
feat(core): add a naive if_constexpr for C++14
GitOrigin-RevId: bd5811ebde
5 years ago
Megvii Engine Team
8f42fe0d92
feat(core): import a std::apply polyfill
GitOrigin-RevId: da95c85b57
5 years ago
Megvii Engine Team
be5dde9da8
fix(mgb/opr-mm): fix group manager sync error
GitOrigin-RevId: 344aae534c
5 years ago
Megvii Engine Team
2dc9190055
fix(fastrun/persistent_cache): fix fastrun crash
GitOrigin-RevId: b3f7bdf7dc
5 years ago
Megvii Engine Team
c1fb3c8c1a
fix(mgb/gopt): fix ConvertF32ToF16Pass endpoints
GitOrigin-RevId: 850eaa0906
5 years ago
Megvii Engine Team
618faf6497
feat(mgb/profiler): dump static infer info
GitOrigin-RevId: bb9150eb83
5 years ago
Megvii Engine Team
5c224c71f0
refactor(mge/imperative): drop ComputingGraphImpl in proxy graph
GitOrigin-RevId: a6e6f2e76b
5 years ago
Megvii Engine Team
5e7d2a91c2
refactor(mgb): add TensorND::proxy_to_default_cpu
GitOrigin-RevId: 3ab8525f1c
5 years ago
Megvii Engine Team
9e9e8ca0fe
fix(mgb): fix some compile errors
GitOrigin-RevId: ec38f8e15a
5 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
5 years ago
Megvii Engine Team
c76789a5c9
feat(mge/imperative): add io remote wrapper
GitOrigin-RevId: 6dae8287d3
5 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
5 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
5 years ago
Megvii Engine Team
8070f40aa1
fix(mgb/gopt): fix gopt nchwxx convert elemwise and reshape
GitOrigin-RevId: 982dee36e1
5 years ago
Megvii Engine Team
b38e8225be
refactor(mgb/opr-mm): update megray communicator init interface and fix ci
GitOrigin-RevId: 55c59879f2
5 years ago
Megvii Engine Team
5e912eddbd
fix(mgb/opr-mm): fix grad func of reduce and gather
GitOrigin-RevId: 4687faef99
5 years ago
Megvii Engine Team
e3e981ccf0
test(mge/collective_comm): fix collective_comm test and add data parallel test
GitOrigin-RevId: 9209e77973
5 years ago
Megvii Engine Team
d3d9018f8d
feat(imperative/opr-mm): add broadcast
GitOrigin-RevId: 83640255c7
5 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