Megvii Engine Team
|
4eda338876
|
feat(dnn/cuda): generate cutlass kimpls using cmake and bazel
GitOrigin-RevId: da3bcfb85a
|
4 years ago |
Megvii Engine Team
|
8d248a6a9a
|
fix(dnn/cuda): fix testcase for fallback nchw qs8 conv
GitOrigin-RevId: 646440db59
|
4 years ago |
Megvii Engine Team
|
894a2407c2
|
feat(dnn/cuda): add relayout format kernel for nchw <-> nhwc
GitOrigin-RevId: e11f3e5408
|
4 years ago |
Megvii Engine Team
|
43c59204df
|
refactor(dnn/cuda): refactor relayout format kernels
GitOrigin-RevId: ab86e66533
|
4 years ago |
Megvii Engine Team
|
f41a808694
|
feat(dnn/cuda): add nhwc int4 conv support
GitOrigin-RevId: 5236b235d0
|
4 years ago |
Megvii Engine Team
|
5a14a89224
|
refactor(dnn/cuda): refactor cutlass kernel generator for gemm and gemv
GitOrigin-RevId: 11d78ab227
|
4 years ago |
Megvii Engine Team
|
b33217d8f0
|
refactor(dnn/cuda): refactor cutlass kernel generator for deconv operation
GitOrigin-RevId: 88e962a912
|
4 years ago |
Megvii Engine Team
|
4abf7bd36f
|
refactor(dnn/cuda): refactor kernel generator for cutlass convolution kernels
GitOrigin-RevId: 7882f9c68c
|
4 years ago |
Megvii Engine Team
|
b4687ce8da
|
feat(dnn/cuda): add convolution with i8 input and u4 output
GitOrigin-RevId: 8be439abf1
|
4 years ago |
Megvii Engine Team
|
00083d13b6
|
fix(dnn/cuda): fix recursive algo search for fallback_nchw_qs8
GitOrigin-RevId: 6be2991224
|
4 years ago |
Megvii Engine Team
|
bba04f02e5
|
feat(mgb/gopt): add fusion support for conv, astype(s4) and reformat
GitOrigin-RevId: 6329ca2c5f
|
4 years ago |
Megvii Engine Team
|
66f70578c2
|
feat(dnn/cuda): add convolution with i8 input and i4 output
GitOrigin-RevId: 10512645d5
|
4 years ago |
Megvii Engine Team
|
6d686ff26f
|
feat(gopt/inference): allow Float32 output dtype in EnableNCHW64Pass
GitOrigin-RevId: 1891efb76f
|
4 years ago |
Megvii Engine Team
|
7d3df995cb
|
feat(gopt/inference): allow Float32 output dtype in EnableNCHW4Pass
GitOrigin-RevId: 81100dbaf7
|
4 years ago |
Megvii Engine Team
|
633016a962
|
fix(dnn/cuda): fix AlgoFallbackNCHWQS8 to support Float32 dst
GitOrigin-RevId: 06f90f5cf3
|
4 years ago |
Megvii Engine Team
|
e6caa9ff89
|
feat(opr): add bn backward for inference mode
GitOrigin-RevId: bb643cb62f
|
4 years ago |
Xinda Huang
|
c90fa087ea
|
test(mge): delete test_external.py
|
3 years ago |
Megvii Engine Team
|
b2944559a8
|
fix(imperative/module): remove ``__getattribute__`` method in module
GitOrigin-RevId: 5ac525f010
|
4 years ago |
Megvii Engine Team
|
77ead9377b
|
fix(src/serialization): fix compatibility error of oss model
GitOrigin-RevId: 43e0fa4fe1
|
3 years ago |
Megvii Engine Team
|
070c811732
|
fix(imperative): remove convert_inputs
GitOrigin-RevId: a3c43db746
|
3 years ago |
Megvii Engine Team
|
f40df60242
|
docs(mge): refactor docs to remove warnings
GitOrigin-RevId: efefc2a4a2
|
3 years ago |
Megvii Engine Team
|
1040b77843
|
fix(mge/functional): fix F.topk(kth_only=True)
GitOrigin-RevId: ddecd1d14b
|
4 years ago |
Megvii Engine Team
|
551cc701c6
|
docs(distributed.functional): add return type for all_reduce_max (jira #MGE-2706)
GitOrigin-RevId: a29f1f1880
|
3 years ago |
Megvii Engine Team
|
72ff7aeccb
|
feat(docs): add docs for megengine.functional.ones_like(jira #MGE-2702)
GitOrigin-RevId: e808fc9e4e
|
3 years ago |
Megvii Engine Team
|
7c9569e4e5
|
fix(mge/random): fix random seed
GitOrigin-RevId: 121f459b1b
|
3 years ago |
Megvii Engine Team
|
07de15713c
|
fix(mgb): remove static mem record from tee
GitOrigin-RevId: ac61b2a5eb
|
4 years ago |
Megvii Engine Team
|
d7b6bfd56c
|
test(mge/fakequant): use fixed input for lsq test to temperarily avoid precision error
GitOrigin-RevId: e91c71874e
|
3 years ago |
Megvii Engine Team
|
5cef74a77e
|
feat(mge/amp): add GradScaler support
GitOrigin-RevId: 0ab4910360
|
4 years ago |
Megvii Engine Team
|
1bf18252c4
|
feat(mge/amp): add mix precision autocast support
GitOrigin-RevId: 6fbffc4845
|
4 years ago |
Megvii Engine Team
|
f12355f727
|
fix(imperative/grad): fix hardcode dtype in subtensor_grad_rule
GitOrigin-RevId: 50da4af26d
|
4 years ago |
Megvii Engine Team
|
4e4497b903
|
refactor(mgb/dnn): x86 pooling rebase algochooser
GitOrigin-RevId: 96cdc57180
|
3 years ago |
Megvii Engine Team
|
a33c3b73bd
|
refactor(mgb/dnn): arm pooling rebase algochooser
GitOrigin-RevId: 21d17e647a
|
3 years ago |
Megvii Engine Team
|
8dea6b3c68
|
build(dnn): compat for more windows env
GitOrigin-RevId: 5ec4be2888
|
3 years ago |
Megvii Engine Team
|
56b94d89a2
|
feat(dtr): add sqrt sampling
GitOrigin-RevId: 8cb2ceb520
|
3 years ago |
Megvii Engine Team
|
8a73193c2d
|
feat(dtr): remove eviction threshold
GitOrigin-RevId: 35c2014bf3
|
3 years ago |
Megvii Engine Team
|
69d1fd0f32
|
refactor(opdef): split apply_on_physical_tensor into infer_output_mem_desc and execute
GitOrigin-RevId: 4d62b7cbbd
|
3 years ago |
Megvii Engine Team
|
75eb04c559
|
feat(mge/experimental): add WeightScaler support
GitOrigin-RevId: a1d5b68498
|
4 years ago |
Megvii Engine Team
|
dedecf6922
|
fix(imperative/utils): fix logical error of replace var
GitOrigin-RevId: 614302552c
|
4 years ago |
Megvii Engine Team
|
ea70d99b4d
|
fix(mge/convbias): make fallback convbias support nhwcd4 layout
GitOrigin-RevId: 1c306f867d
|
4 years ago |
Megvii Engine Team
|
497ef6c337
|
fix(mge/dist): fix gl oom error
GitOrigin-RevId: 4ba3d2cfd7
|
3 years ago |
Megvii Engine Team
|
43098fb8f1
|
feat(mge): add SlidingWindowTranspose opr
BREAKING CHANGE:
GitOrigin-RevId: 54d726d2fe
|
4 years ago |
Megvii Engine Team
|
df79334cae
|
feat(mge/distributed): add user_pop function to save device memory
BREAKING CHANGE:
GitOrigin-RevId: 0a8e406da5
|
3 years ago |
Megvii Engine Team
|
1eaf32cd78
|
fix(mgb): fix typo in message
GitOrigin-RevId: 92d778b5df
|
4 years ago |
Megvii Engine Team
|
7225b0f09f
|
fix(mge/utils): use static infer manager to get value of network.varnode
GitOrigin-RevId: ecc47edab8
|
4 years ago |
Megvii Engine Team
|
ffe2bb2eb2
|
fix(mge): fix some errors caused by unknown shape when using symbolic trace or building graph
GitOrigin-RevId: 70ddc06eee
|
4 years ago |
Megvii Engine Team
|
2d42455fa8
|
fix(mge/utils): fix toposort to get definition order
GitOrigin-RevId: 47a26dd6dd
|
4 years ago |
Megvii Engine Team
|
0c97b2a3ec
|
fix(module): remove assert during forward
GitOrigin-RevId: bb04d861a3
|
3 years ago |
Megvii Engine Team
|
427113088b
|
fix(module/normalization): fix bug of LayerNorm and support input of any shape
GitOrigin-RevId: fd643addb5
|
4 years ago |
Megvii Engine Team
|
a95f6d4f75
|
perf(trace): add fastpath for const value assert
GitOrigin-RevId: 9a966f257f
|
3 years ago |
Megvii Engine Team
|
2cd9823210
|
fix(mgb/tensorrt): fix trt runtime, padding channel to a multiple of 4 when using kCHW4 IOFormat
GitOrigin-RevId: c5f1ed70da
|
3 years ago |