Megvii Engine Team
dbd9483993
feat(dnn,src,imperative): add groupnorm op
GitOrigin-RevId: de3c3d10e5
3 years ago
Megvii Engine Team
95b2a36f55
feat(core): add zero copy input output from out of megengine
GitOrigin-RevId: 388d9d9551
3 years ago
Megvii Engine Team
563a0d010a
feat(opencl/ion): support ion
GitOrigin-RevId: c9979476de
3 years ago
Megvii Engine Team
6d21982773
feat(opencl/svm): add opencl svm more test
GitOrigin-RevId: 3ebfbed0bd
3 years ago
Megvii Engine Team
4c27522a2d
fix(src/core): fix weight preprocess free combine memory UAF issue
GitOrigin-RevId: b779e05956
3 years ago
Megvii Engine Team
f287501eb2
fix(src/core): fix pushdown alloc_result inconsistent when run for many time
GitOrigin-RevId: 3b087203a1
3 years ago
Megvii Engine Team
9e020d23b4
feat(imperative,mgb): add RegionRestrictedConvolution megbrain and imperative opr
GitOrigin-RevId: c0106ade08
3 years ago
Megvii Engine Team
d3b2b51918
feat(mge/opr): add meshgrid opr
GitOrigin-RevId: 6f703295be
3 years ago
Megvii Engine Team
dc0ab9b64e
feat(lite): replace warp when src is discrete input
GitOrigin-RevId: 2bf7980ac6
3 years ago
Megvii Engine Team
411253f852
feat(mge): implement warp_affine backward
GitOrigin-RevId: 7d7261cf64
3 years ago
Megvii Engine Team
116781ba9c
fix(mgb): fix megtee build errors
GitOrigin-RevId: b351dd3994
3 years ago
Megvii Engine Team
b8c7557b86
fix(mm): fix mm error when use sync
GitOrigin-RevId: 63387bda04
3 years ago
Megvii Engine Team
399200b3a7
perf(serialization): optimized the memory usage when load new format model
GitOrigin-RevId: 2b7313ebe3
3 years ago
Megvii Engine Team
cc21855074
feat(lite): load_and_run support optimize for inference
GitOrigin-RevId: d9abb8de9e
3 years ago
Megvii Engine Team
626222c698
fix(test): fix test for brainpp docker env
GitOrigin-RevId: c4c2cc73d2
3 years ago
Megvii Engine Team
fac67e7c2b
feat(gopt): support nchw44 global pooling with fuse_grain
GitOrigin-RevId: 4c43a149f8
3 years ago
Megvii Engine Team
32b31fd578
fix(mgb): change the check method of cuda sm code
GitOrigin-RevId: 23dbc9b574
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
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
cfed86f9af
feat(persistentcache): change file persistent cache with append model
GitOrigin-RevId: 7a427bdab4
3 years ago
Megvii Engine Team
35cf0422f0
fix(ci): relax async timeout
GitOrigin-RevId: 551de481f0
3 years ago
Megvii Engine Team
a32b727720
fix(build): upgrade bazel riscv toolchains
GitOrigin-RevId: 8ac61cc4b6
3 years ago
Megvii Engine Team
24c5c19bf0
fix(imperative): make functional ops support negative axis
GitOrigin-RevId: f61e01270b
3 years ago
Megvii Engine Team
64a8aaaf42
fix(build): remove ununsed functions when cuda disabled
GitOrigin-RevId: 03f50683f6
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
50faabf614
feat(serialization): support the registry for new serialization format
GitOrigin-RevId: 8eacd5e77c
3 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
bde2efa3b5
feat(lite/load_and_run): support put and get model redis cache
GitOrigin-RevId: 55c82e28c1
3 years ago
Megvii Engine Team
48526abb79
fix(mgb): fix concat cd4 tensor check size invalid
GitOrigin-RevId: 065e0b4be0
3 years ago
Megvii Engine Team
c87d998e59
feat(mgb): add interface to support opencl IO zero copy when inference
GitOrigin-RevId: a1d7021892
3 years ago
Megvii Engine Team
a0e531180d
fix(src/comp_node): fix calling cuda driver api
GitOrigin-RevId: cc33af2ac4
3 years ago
Megvii Engine Team
e2f5156b69
refactor(megbrain): save fastrun result to algorithm cache
GitOrigin-RevId: 45301ebb4d
3 years ago
Megvii Engine Team
7dc347697a
feat(dnn/cuda): add typecvt uint16
GitOrigin-RevId: d1368c414e
3 years ago
Megvii Engine Team
b92866d2c2
fix(build): fix build depends dirty file issue
GitOrigin-RevId: 435d8b5c50
3 years ago
Megvii Engine Team
27d4c4b36c
refactor(stats): use static inline variable declaration
GitOrigin-RevId: 7d86e5f257
3 years ago
Megvii Engine Team
787a22a9d6
perf(tensor): implement __new__ in cpp
GitOrigin-RevId: 4defd249c3
3 years ago
Megvii Engine Team
99df4a7996
fix(dtype): dtype scalar set_retain_dtype supports bool
GitOrigin-RevId: aafd378e1b
3 years ago
Megvii Engine Team
7bf5b0ee1e
test(imperative): check env values after each pytest
GitOrigin-RevId: 826788113a
3 years ago
Megvii Engine Team
409c988163
fix(imperative): add matmul apply_on_varnode
GitOrigin-RevId: 2cf6bf237c
3 years ago
Megvii Engine Team
b9cbc10120
feat(lite): add pack model
GitOrigin-RevId: 1a150f2af3
3 years ago