Megvii Engine Team
|
a8b146d74b
|
Merge pull request #397 from kagome112:master
GitOrigin-RevId: e40b12f76e
|
2 years ago |
Megvii Engine Team
|
5655636c1c
|
chore(mge/metric): funtional.topk_acc (add nn) use functional.metric
GitOrigin-RevId: 77fd432cb2
|
2 years ago |
Megvii Engine Team
|
d5688c3d38
|
chore(mge/functional): deprecate debug_param and prefer config
GitOrigin-RevId: 0d7126641b
|
2 years ago |
Megvii Engine Team
|
9bb2e395ce
|
chore(mge/tensorcache): rename the module making it not public
GitOrigin-RevId: 559d205a8c
|
2 years ago |
Megvii Engine Team
|
6e56d1bd57
|
chore(mge/deprecated): add __deprecated__ attr for condition case
GitOrigin-RevId: 6d8e7398e4
|
2 years ago |
Megvii Engine Team
|
98460f583f
|
chore(mge/misc): rename APIs that should not be public
GitOrigin-RevId: b72517bfe9
|
2 years ago |
Megvii Engine Team
|
217999b1fa
|
feat(arm): add winograd F43 NCHW44 algo and winograd F43 44 algo
GitOrigin-RevId: a981b2f61b
|
2 years ago |
Megvii Engine Team
|
f0f6f5fe5c
|
build(mge/whl): package libnvrtc-builtins.so.11.x
GitOrigin-RevId: 607af27256
|
2 years ago |
Megvii Engine Team
|
fb9d73b279
|
fix(src/tensorrt): trt7 manage all workspace
GitOrigin-RevId: b78c80c8f1
|
2 years ago |
Megvii Engine Team
|
dee0228982
|
fix(serialization): when the dump fbsv2 model is used, the middle_tensor becomes optional
GitOrigin-RevId: 3d0bbfd441
|
2 years ago |
Megvii Engine Team
|
b2959589fc
|
docs(mge): update AvgPool2d example
GitOrigin-RevId: 2d5e515055
|
2 years ago |
Megvii Engine Team
|
ada54ba5a0
|
docs(mge): update AvgPool2d & functional.nn.avg_pool2d example
GitOrigin-RevId: e2859c7f99
|
2 years ago |
Megvii Engine Team
|
d8ea01680c
|
docs(mge): update AvgPool2d & functional.nn.max_pool2d docstring
GitOrigin-RevId: d69c5b024f
|
2 years ago |
Megvii Engine Team
|
01353b2d90
|
Merge pull request #442 from Seeker98:avgpool_doc
GitOrigin-RevId: d048955398
|
2 years ago |
Megvii Engine Team
|
1529bce525
|
perf(opencl): add opencl weight transpose kernel
GitOrigin-RevId: 1102ecf4bb
|
2 years ago |
Megvii Engine Team
|
76cd4a5207
|
feat(mgb): add cu114 wheel
GitOrigin-RevId: 013c1cfa50
|
2 years ago |
Megvii Engine Team
|
25f97b76ea
|
feat(imperative): load_nerwork_and_run enable weight preprocess
GitOrigin-RevId: 0642b237a7
|
2 years ago |
Megvii Engine Team
|
5ee0094322
|
fix(dnn/cuda): fix ptx mma algo compute bugs
GitOrigin-RevId: 19628d0c94
|
2 years ago |
Megvii Engine Team
|
d4bf57d6ac
|
fix(lite): fix lar multithread options invalid
GitOrigin-RevId: 55f83036b1
|
2 years ago |
Megvii Engine Team
|
1404437a90
|
fix(mgb): fix the compatibility issue of cuda stub with older version drivers
GitOrigin-RevId: 628afbf3cf
|
2 years ago |
Megvii Engine Team
|
a6a2646c10
|
feat(arm): add AlgoFP32Winograd F43, and add filter size into name of winograd-related algorithms
GitOrigin-RevId: 909503a90d
|
2 years ago |
Megvii Engine Team
|
b8821edb3d
|
perf(dnn/aarch64): optimize aarch64 sigmoid with asm
GitOrigin-RevId: 7d54d67669
|
2 years ago |
Megvii Engine Team
|
2b99bfec4e
|
feat(arm): supports weight pre-processing for winograd benchmark tests
GitOrigin-RevId: 1797f3b91c
|
2 years ago |
Megvii Engine Team
|
ab34bac4fc
|
fix(opencl/extern_c_opr): fix cl_mem UAF issue 2/2
GitOrigin-RevId: ef79af3c6a
|
2 years ago |
Megvii Engine Team
|
421bcfd3d8
|
style(mgb/tools): add format for tools, dnn and ci
GitOrigin-RevId: 5684e5ea43
|
3 years ago |
Megvii Engine Team
|
99309fa3d6
|
feat(mge/functional): add param output_padding for deconv ops
GitOrigin-RevId: 8a69608953
|
2 years ago |
Megvii Engine Team
|
116781ba9c
|
fix(mgb): fix megtee build errors
GitOrigin-RevId: b351dd3994
|
2 years ago |
Megvii Engine Team
|
54b5db1729
|
feat(x86/rvv): add AGENT_NCHW_NCHW44 algo
GitOrigin-RevId: 8cf6c3fac0
|
2 years ago |
Megvii Engine Team
|
eaa180181a
|
feat(x86/rvv): opt gi intrinsic helper
for rvv, detail: https://github.com/riscv-collab/riscv-gnu-toolchain/issues/1106
GitOrigin-RevId: 27615584c0
|
2 years ago |
Megvii Engine Team
|
399db31aab
|
fix(dnn): fix build
GitOrigin-RevId: d91077248a
|
2 years ago |
Megvii Engine Team
|
b8c7557b86
|
fix(mm): fix mm error when use sync
GitOrigin-RevId: 63387bda04
|
2 years ago |
Megvii Engine Team
|
73ad06baba
|
fix(mgb): fix fbsv2 model format no dump tensor format
GitOrigin-RevId: 29f785b701
|
2 years ago |
Megvii Engine Team
|
399200b3a7
|
perf(serialization): optimized the memory usage when load new format model
GitOrigin-RevId: 2b7313ebe3
|
2 years ago |
Megvii Engine Team
|
f31e52d521
|
feat(mgb): warpperspective support multi src input
GitOrigin-RevId: 0887656864
|
2 years ago |
Megvii Engine Team
|
669816e291
|
feat(dnn): warpperspective support multi src input
GitOrigin-RevId: 8a4789852e
|
2 years ago |
Megvii Engine Team
|
33b27be8fc
|
fix(mgb/build): fix trt8 build error
GitOrigin-RevId: 14acc064e0
|
2 years ago |
Megvii Engine Team
|
bccda5c427
|
fix(mgb/imperative): fix repeat bug in trace mode
GitOrigin-RevId: 9547fc6102
|
2 years ago |
Megvii Engine Team
|
fca6c76a0e
|
fix(lite): fix input invalid bug in lar for fitting mode
GitOrigin-RevId: 45d81c9a96
|
2 years ago |
Megvii Engine Team
|
1b94380794
|
fix(dnn): fix reduce sum/mean error when b is large
GitOrigin-RevId: d1bae619b1
|
2 years ago |
Megvii Engine Team
|
c7a9909839
|
feat(cuda): add int4 ptx 256x64 mma kernel
GitOrigin-RevId: 8f7475b0f6
|
2 years ago |
Megvii Engine Team
|
cf3ca1e9a2
|
feat(cuda): add int4 ptx 128x256 mma kernel
GitOrigin-RevId: 1ae7c9f034
|
2 years ago |
Megvii Engine Team
|
1f8e930e28
|
feat(cuda): add int4 ptx 128x128 mma kernel
GitOrigin-RevId: 5a8b9c3f8e
|
2 years ago |
Megvii Engine Team
|
1a2ed8c47b
|
feat(cuda): add convbias ptx algo testcase
GitOrigin-RevId: 9ad6d4561f
|
2 years ago |
Megvii Engine Team
|
64551105f9
|
feat(cuda): add convbias ptx algo
GitOrigin-RevId: 08e9f66641
|
2 years ago |
Megvii Engine Team
|
8395a459b5
|
fix(dnn/fallback): fix naive shift multidefination error and optimize GiCvtFromInt32V4ToUint8
GitOrigin-RevId: 6660c35214
|
2 years ago |
Megvii Engine Team
|
cc21855074
|
feat(lite): load_and_run support optimize for inference
GitOrigin-RevId: d9abb8de9e
|
2 years ago |
Megvii Engine Team
|
9bbe550032
|
fix(opencl/extern_c_opr): fix cl_mem UAF issue when
run model OpenCL + ExternCOprRunner, for example
graph: part_a(OpenCL) --> part_b(ExternCOprRunner) --> part_c(OpenCL)
GitOrigin-RevId: f754b559a2
|
2 years ago |
Megvii Engine Team
|
23a3d13350
|
fix(dnn/softmax): create redcue and elemwise opr when get workspace size
GitOrigin-RevId: 476a39bdd3
|
2 years ago |
Megvii Engine Team
|
2797fcfad0
|
fix(mge/device): add missed API to __all__ scope
GitOrigin-RevId: c3ce90990f
|
2 years ago |
Megvii Engine Team
|
d7c546c90a
|
fix(mge/interpreter): regenerates tensor when its dev value is needed
GitOrigin-RevId: ed26d52ee4
|
2 years ago |