Megvii Engine Team
32b31fd578
fix(mgb): change the check method of cuda sm code
GitOrigin-RevId: 23dbc9b574
3 years ago
Megvii Engine Team
5f86368219
Revert "feat(dnn): add elemwise modes"
This reverts commit cb713ddb24 .
GitOrigin-RevId: 02adf025e6
3 years ago
Megvii Engine Team
d2a1905ad5
Revert "feat(mgb): add cumprod opr"
This reverts commit 3436c3bdaa .
GitOrigin-RevId: 95ab3d1aa7
3 years ago
Megvii Engine Team
49e14f87b5
feat(mgb): add cumprod opr
GitOrigin-RevId: 3436c3bdaa
3 years ago
Megvii Engine Team
87aedc2991
feat(dnn): add elemwise modes
GitOrigin-RevId: cb713ddb24
3 years ago
Megvii Engine Team
c49d3070ba
refactor(imperative/ops): extends DnnOprCaller with template
GitOrigin-RevId: 402cba209a
3 years ago
Megvii Engine Team
3fbceb3a38
fix(mgb/version): fix nvinfer.h not found
GitOrigin-RevId: 981dd9a7d2
3 years ago
Megvii Engine Team
d60d028a40
feat(mge/device): enable to get cuda/cudnn/tensorrt version
GitOrigin-RevId: 5864c61d10
3 years ago
Megvii Engine Team
6dfd5a4cd4
fix(win7): workaround for hang when progress exit on win7+32bit
GitOrigin-RevId: b49b0b230e
3 years ago
Megvii Engine Team
51397bfc08
feat(mgb): supports value infer and empty input tensor in ElemwiseMultiType
GitOrigin-RevId: 05577a8bc8
3 years ago
Megvii Engine Team
247e2f59a4
feat(mgb/dnn): add modes that the output type is bool in elemwise
GitOrigin-RevId: fd0134fca2
3 years ago
Megvii Engine Team
cfed86f9af
feat(persistentcache): change file persistent cache with append model
GitOrigin-RevId: 7a427bdab4
3 years ago
Megvii Engine Team
d19fc2c1da
fix(imperative): add alloc TensorPtr in imperative
GitOrigin-RevId: 1b438fc436
3 years ago
Megvii Engine Team
ac51f7807d
feat(mge/distributed): add support for batch send recv op
GitOrigin-RevId: eb3d712704
3 years ago
Megvii Engine Team
35cf0422f0
fix(ci): relax async timeout
GitOrigin-RevId: 551de481f0
3 years ago
Megvii Engine Team
5a35513856
fix(mgb): fix profile skip condition
GitOrigin-RevId: f196eabc98
3 years ago
Megvii Engine Team
5bdc430ec1
fix(mgb/fastrun): fix megbrain fastrun memory overflow bug
GitOrigin-RevId: f56aa5a505
3 years ago
Megvii Engine Team
a4f019e79d
fix(mgb/atlas): fix issue with profiling on Atlas
GitOrigin-RevId: fbae81940d
3 years ago
Megvii Engine Team
cfc41648a4
fix(mge): fix grad of maximum(x, x)
GitOrigin-RevId: e0e2efb71b
3 years ago
Megvii Engine Team
ed92b9c125
fix(cmake): fix compilation options for benchmark in cmake
GitOrigin-RevId: d763fd3262
3 years ago
Megvii Engine Team
f0088335bb
feat(mgb): upgrade flatbuffer
GitOrigin-RevId: 7b1a04934e
3 years ago
Megvii Engine Team
4d22e85b46
feat(ci): add completeness compatibility check
GitOrigin-RevId: a7f75c4c5e
3 years ago
Megvii Engine Team
a32b727720
fix(build): upgrade bazel riscv toolchains
GitOrigin-RevId: 8ac61cc4b6
3 years ago
Megvii Engine Team
1192a9a61b
fix(imperative): fix adaptive pool2d
GitOrigin-RevId: 624b47a51a
3 years ago
Megvii Engine Team
24c5c19bf0
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
3 years ago
Megvii Engine Team
f96429c031
feat(imperative): support empty tensor in roi_align
GitOrigin-RevId: aeb2770401
3 years ago
Megvii Engine Team
64a8aaaf42
fix(build): remove ununsed functions when cuda disabled
GitOrigin-RevId: 03f50683f6
3 years ago
Megvii Engine Team
972a8d5489
feat(ci): add model compatibility check in ci
GitOrigin-RevId: d17c21ab4f
3 years ago
Megvii Engine Team
e694301721
fix(ops/jit): skip lookup include path when nvcc executable not found
GitOrigin-RevId: f5e7dce1c5
3 years ago
Megvii Engine Team
c2deef1a97
feat(mge): aad atlas710 support
GitOrigin-RevId: 6458c5c23c
3 years ago
Megvii Engine Team
b36b5bd8cb
refactor(mgb): check input when profiling
GitOrigin-RevId: 1d722dd741
3 years ago
Megvii Engine Team
6c9b3a58e3
refactor(dnn): remove algorithm cache queries
GitOrigin-RevId: b7a1dc62d8
3 years ago
Megvii Engine Team
8563f51404
fix(imperative): fix buildin reduce keepdim
GitOrigin-RevId: 38d90ab38a
3 years ago
Megvii Engine Team
eef0308b29
feat(serialization): add no_change_graph and version param whem dump model
GitOrigin-RevId: 65064452c9
3 years ago
Megvii Engine Team
4ab5f970e9
fix(build): fix ci error
GitOrigin-RevId: 9cbf64dda2
3 years ago
Megvii Engine Team
b9a6932341
feat(imperative): channel default model format to fbs v2
GitOrigin-RevId: 6066516c31
3 years ago
Megvii Engine Team
283a2f77e7
feat(serialization): support test for new serialization format
GitOrigin-RevId: 9a8047ec0b
3 years ago
Megvii Engine Team
50faabf614
feat(serialization): support the registry for new serialization format
GitOrigin-RevId: 8eacd5e77c
3 years ago
Megvii Engine Team
a694fb3304
feat(serialization): implement the new serialization format
GitOrigin-RevId: 00f87f7ccd
3 years ago
Megvii Engine Team
ca4a5da013
feat(serialization): add new serialization format define
GitOrigin-RevId: 177bfdd56b
3 years ago
Megvii Engine Team
3bd40887b6
feat(mgb/opr): add NHWC support for AdaptivePooling
GitOrigin-RevId: b23e37ac23
4 years ago
Megvii Engine Team
533fb5bf49
feat(imperative): support formatted tensor and add special op rules
GitOrigin-RevId: 77ff909f23
4 years ago
Megvii Engine Team
98b5ee78c1
feat(mge/dnn): add lamb optimizer
GitOrigin-RevId: 5a27157456
3 years ago
Megvii Engine Team
02bfb8f8b9
feat(lite): add and fix some feature for load and run fitting mode
GitOrigin-RevId: bbddc9bb79
3 years ago
Megvii Engine Team
80e1f38bea
fix(gtest): fix ci error report stack-use-after-scope
how to reproduce the problem:
1: build with asan(revert this MR)
2: then taskset process to one cpu:
taskset 01 ./megbrain_test --gtest_filter=TestAsyncQueue.SynchronizerWaiterStarving
GitOrigin-RevId: eb6f7aa4d8
3 years ago
Megvii Engine Team
c2e9860feb
chore(license): remove all license in file header
GitOrigin-RevId: a0e31247a6
3 years ago
Megvii Engine Team
38b492727e
fix(opr): fix no update ptr in reduce operator when input change
GitOrigin-RevId: a443a79ac0
3 years ago
Megvii Engine Team
4cce2480d5
fix(dnn/opencl): fix some bug for dnn opencl conv bias and relayout format
GitOrigin-RevId: b5bb07d90d
3 years ago
Megvii Engine Team
1783b8977a
feat(profiler): integrate cupti backend
GitOrigin-RevId: dec8be1908
3 years ago
Megvii Engine Team
bde2efa3b5
feat(lite/load_and_run): support put and get model redis cache
GitOrigin-RevId: 55c82e28c1
3 years ago