Megvii Engine Team
a78c11099b
fix(imperative): add param(axis) for GetVarShape
GitOrigin-RevId: 0b8f821929
4 years ago
Megvii Engine Team
cde9727a4f
fix(mge/quantization): fix `qat.conv_bn` parameter issue
GitOrigin-RevId: 4aa3836926
4 years ago
Megvii Engine Team
94796060b6
test(mge/quantization): qat module, quantized op and module
GitOrigin-RevId: 17b28060cc
4 years ago
Megvii Engine Team
ab9f44f15c
feat(mge/quantization): add support for easyquant
GitOrigin-RevId: 060d908349
4 years ago
Megvii Engine Team
fc0fcd2f7f
chore(winograd): remove winograd transform code
GitOrigin-RevId: 78c3cfceae
4 years ago
Megvii Engine Team
d1adc9a22f
fix(dnn): fix opencl algo search
GitOrigin-RevId: 25997d0ef1
4 years ago
Megvii Engine Team
368c18607f
fix(mgb/jit): find cuda include path correctly
GitOrigin-RevId: 5ced4206a7
4 years ago
Megvii Engine Team
b04e0466bb
feat(megbrain): add alias name to model serialization
GitOrigin-RevId: d47e650473
4 years ago
Megvii Engine Team
cf53d9e0f8
fix(mgb/tensor): do tensor overlap check only when d2d and h2h
GitOrigin-RevId: 9125936a35
4 years ago
Megvii Engine Team
7e2b2dbffc
fix(dnn/test): delete large size in ARM_COMMON.FP32_GEVM
GitOrigin-RevId: 581ef43816
4 years ago
Megvii Engine Team
69e3e32240
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
4 years ago
Megvii Engine Team
0398a7867f
fix(build/windows/cuda/llvm): fix windows bazel build with cuda
* Adapt to the new version llvm/clang-11
* fix windows bazel build with cuda
* add windows bazel build cuda ci
* opt windows bazel ci scripts
GitOrigin-RevId: 6ea7c66585
4 years ago
Megvii Engine Team
b9c37112a2
refactor(mge/distributed): skip barrier when running with single node
GitOrigin-RevId: 6cd5a2e1e2
4 years ago
Megvii Engine Team
3bf73ff16f
feat(dnn): add cuda preprocess fusion
GitOrigin-RevId: d789c99e59
4 years ago
Megvii Engine Team
86cf7490ec
feat(dnn/aarch64): add quantizeds4 matmul int4x4x16_k8x8x8
GitOrigin-RevId: 7812900244
4 years ago
Megvii Engine Team
bff0fc6172
fix(mge/interpreter): fix outputs check on async level0
GitOrigin-RevId: 46c6a2ce70
4 years ago
Megvii Engine Team
142f31a875
perf(dnn/cuda): change conv_bias heu, prefer dnn chanwise impl, dislike dnn batch gemm conv1x1
GitOrigin-RevId: 323bf6073a
4 years ago
Megvii Engine Team
f214e14695
refactor(mgb/cuda): use single implementation of get_device_prop from utils
GitOrigin-RevId: 5cc95472b9
4 years ago
Megvii Engine Team
54e79dd1d9
perf(mgb/cuda): do not call cudaGetDeviceProperties to avoid io traffic
GitOrigin-RevId: 6aa35928c8
4 years ago
Megvii Engine Team
5f171298aa
feat(mgb/gopt): add AxisAddRemove opr support for cd4 opt pass
GitOrigin-RevId: 85218ee0c4
4 years ago
Megvii Engine Team
93f4977c78
feat(mge/imperative): add thread name
GitOrigin-RevId: 61eece55b8
4 years ago
Megvii Engine Team
98a74e4a7b
refactor(dnn): refactor opr proxy in test
GitOrigin-RevId: a1d8682e6f
4 years ago
Megvii Engine Team
57546b4c3d
test(mge/distributed): fix test skip condition error
GitOrigin-RevId: f540c53302
4 years ago
Megvii Engine Team
90e7cb005c
feat(externcopr/lar): imp lar run extern c opr with dynamic param
GitOrigin-RevId: 81c147aa6c
4 years ago
Megvii Engine Team
dbb64b46d5
feat(debug/android): opt android backtrace
GitOrigin-RevId: d8cef5f7c1
4 years ago
Megvii Engine Team
3e00e3f697
feat(debug/linux): opt linux backtrace
GitOrigin-RevId: bdf554cd3c
4 years ago
Megvii Engine Team
783a612643
feat(debug/macos/windows): imp macos/windows backtrace, fix mem issue
GitOrigin-RevId: 9bed738bb5
4 years ago
Megvii Engine Team
e92670e820
fix(mgb/atlas): when batchsize more than model max batchsize
GitOrigin-RevId: 63fe79eaa9
4 years ago
Megvii Engine Team
147dbf8a0c
fix(test): fix a race condition in TestCudaMemAlloc
GitOrigin-RevId: 99468a7ae3
4 years ago
Megvii Engine Team
7066ad5ba6
feat(dnn): add uint16 support
GitOrigin-RevId: f4c4b1c7b9
4 years ago
Megvii Engine Team
a1877ee0fa
refactor(dnn): refactor algo interface, use algoinfo instead of global algorithm
GitOrigin-RevId: 479718ac75
4 years ago
Megvii Engine Team
cb59c27835
feat(mlir/ir): add more op definitions
GitOrigin-RevId: 1e1285ef41
4 years ago
Megvii Engine Team
9ec8d375f1
feat(externcopr): add config extern c opr dynamic param
GitOrigin-RevId: 0fa3a534af
4 years ago
Megvii Engine Team
ee4ea7fdc8
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
4 years ago
Megvii Engine Team
3ecded74ea
refactor(distributed/server): use port 0 to get available port
GitOrigin-RevId: e367846b92
4 years ago
Megvii Engine Team
88e918e261
feat(mgb/jit): add scf.ForOp in MgbToGpuLoweringPass
GitOrigin-RevId: 3cdae27c37
4 years ago
Megvii Engine Team
7aa54b0ec6
feat(mge): enable memory swap and drop/recomputation
GitOrigin-RevId: c56c87c88b
4 years ago
Megvii Engine Team
6f5d0febf1
perf(dnn/cuda): enhance performance for pooling forward
GitOrigin-RevId: 55fb2a9b25
4 years ago
Megvii Engine Team
0560a218af
chore(dnn/test): refactor megdnn arm_common test
GitOrigin-RevId: 4168910301
4 years ago
Megvii Engine Team
f7731bd437
fix(mgb/jit): fix a pointer bug in mlir executable_cuda
GitOrigin-RevId: 3ec79b7602
4 years ago
Megvii Engine Team
810d8cbaf8
fix(mgb/jit): add cmake target MLIRTosa for latest llvm-project
GitOrigin-RevId: f59f23cb8f
4 years ago
Megvii Engine Team
2ad8c5e1e9
fix(mge/io_remote): fix remote send/recv gradient at trace
GitOrigin-RevId: 7886efd0c1
4 years ago
Megvii Engine Team
f470df4f1a
fix(mgb/opr): fix convbias with no bias when weight preprocess
GitOrigin-RevId: 8837ad539c
4 years ago
Megvii Engine Team
6c4841e807
fix(mge/quantization): `disable_fake_quant` does not work correctly
GitOrigin-RevId: 0c568d3335
4 years ago
Megvii Engine Team
aa953c3bd6
fix(mge/module): fix missing import
GitOrigin-RevId: 7cdf6ac814
4 years ago
Megvii Engine Team
5a01de7851
fix(mge): fix scalar transpose
GitOrigin-RevId: c2b9e025c7
4 years ago
Megvii Engine Team
6b9ac894d3
fix(mgb/topk): fix topk grad
GitOrigin-RevId: a49154ff06
4 years ago
Megvii Engine Team
6856ce9ce2
feat(dnn): support conv bias activation for nchw4 input tensor format and nchw output tensor format
GitOrigin-RevId: 29cd73f87b
4 years ago
Megvii Engine Team
8536864351
chore(version): fix dev version to a large number
GitOrigin-RevId: 9567513c1f
4 years ago
Megvii Engine Team
61c5c9cf74
chore(cmake): normlize some cmake message level
GitOrigin-RevId: 4550cbfaa6
4 years ago