Megvii Engine Team
ca1d74ad3f
fix(mge/utils): fix types import error
GitOrigin-RevId: d2d4bd2725
4 years ago
Megvii Engine Team
abe3c165ba
feat(mge): tensorrt runtime opr
GitOrigin-RevId: 2fdd00adcb
4 years ago
Megvii Engine Team
82847bc70a
docs(mge): add conv2d output shape computation formula
GitOrigin-RevId: 7ee476a80d
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
69e3e32240
feat(imperative): auto generated opdef header and python binding
GitOrigin-RevId: d2f22ad5fe
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
2e87420865
fix(module): fix docs in normalization
GitOrigin-RevId: 326234c35a
4 years ago
Megvii Engine Team
7167fdbd49
feat(mge/module): add normalization module includes group_norm, instance_norm and layer_norm
GitOrigin-RevId: 9f253e3250
4 years ago
Megvii Engine Team
94dba16ff4
perf(mge/imperative): misc optimizations
GitOrigin-RevId: bbe7a10b00
4 years ago
Megvii Engine Team
442058aead
feat(mge/quantize): add batch matmul activation module for inference
GitOrigin-RevId: f6dbd1f4c0
4 years ago
Megvii Engine Team
8e11204ab6
perf(mge/module): optimize conv_bn qat module to improve performance
GitOrigin-RevId: 9415b83d9c
4 years ago
Megvii Engine Team
82b0f67770
fix(mge/core): fix dtype promotion issue for quantized dtype
GitOrigin-RevId: 9d09c8fa6f
4 years ago
Megvii Engine Team
9cb3c07cd4
feat(mge/functional): add elemwise mode support string input
GitOrigin-RevId: 57be5cec7b
4 years ago
Megvii Engine Team
8fad00a158
feat(mge/functional/nn): add conv1d padding
GitOrigin-RevId: 1bbfd36b96
4 years ago
Megvii Engine Team
c89209450d
ci(mge): remove blank line for windows doctest
GitOrigin-RevId: 09cf6518e8
4 years ago
Megvii Engine Team
b309890c66
docs(mge): pytest for sphinx docstring
GitOrigin-RevId: 8bed12562a
4 years ago
Megvii Engine Team
fccb2510c4
perf(mge): add opdef for elemwise and batchnorm
GitOrigin-RevId: d51fad9867
4 years ago
Megvii Engine Team
6701346398
docs(mge): fix some docstring format problem
GitOrigin-RevId: cbc5ab04b3
4 years ago
Megvii Engine Team
20e93630f2
feat(mge): rename batch_norm2d -> batch_norm
GitOrigin-RevId: 253e8564ea
4 years ago
Megvii Engine Team
7589bce454
feat(mge): do not export F.nn.sync_batch_norm
GitOrigin-RevId: aa05e3aac4
4 years ago
Megvii Engine Team
d4e2d27213
feat(mge): do not export F.nn.conv_bias_activation
GitOrigin-RevId: e536f87fad
4 years ago
Megvii Engine Team
cf0507e1ba
feat(mge): remove F.identity
GitOrigin-RevId: 858be627ac
4 years ago
Megvii Engine Team
164d9cdf1f
feat(mge): do not export F.nn.linear
GitOrigin-RevId: 589964729d
4 years ago
Megvii Engine Team
fd802e978f
feat(mge): do not export F.nn.embedding
GitOrigin-RevId: eb90506535
4 years ago
Megvii Engine Team
205a39b07d
feat(mge): remove add_update
GitOrigin-RevId: 3068593eeb
4 years ago
Megvii Engine Team
67d33303ad
feat(mge): remove fast_tanh
GitOrigin-RevId: 7ad5f7ecce
4 years ago
Megvii Engine Team
78d2189441
feat(mge): rename comparison operators to their full name
GitOrigin-RevId: e503b038a1
4 years ago
Megvii Engine Team
2bd84d6789
feat(mge): add adaptive pooling python wrapper
GitOrigin-RevId: 789f1511ec
4 years ago
Megvii Engine Team
0e3037105c
feat(mge/module): add group_local_conv
GitOrigin-RevId: 0503377bc5
4 years ago
Megvii Engine Team
026af62042
docs(mge): docs typo fix
GitOrigin-RevId: 851f6de02f
4 years ago
Megvii Engine Team
477820fe5d
refactor(mge): apply python-module changes
GitOrigin-RevId: 2c769864c3
4 years ago
Megvii Engine Team
66f2dbd753
feat(mge/imperative): add module __repr__ for imperative
GitOrigin-RevId: ac13cc4659
4 years ago
Megvii Engine Team
4ab259f558
refactor(mge): add some deprecation and migration guides
GitOrigin-RevId: 4793adf02b
4 years ago
Megvii Engine Team
715009b564
refactor(mge/api): remove external, dropout fix
GitOrigin-RevId: 5e6ff1a372
4 years ago
Megvii Engine Team
a8c75ee530
refactor(mge/functional): make docs better
GitOrigin-RevId: acb287d48b
4 years ago
Megvii Engine Team
6c1dbd40d9
docs(mge): fix doctest
GitOrigin-RevId: 131fed8733
4 years ago
Megvii Engine Team
39d4328309
fix(mge/functional): functional api fixes
GitOrigin-RevId: fa206c4ff6
4 years ago
Megvii Engine Team
10e942d952
refactor(mge): polish api
- refactor(mge): add support for optimizer.step().clear_grad() idiom
- refactor(mge): rename some methods of GradManager
- refactor(mge): remove tensor_nn and TensorDict
- refactor(mge): remove Buffer
- refactor(mge): remove requires_grad flag
- refactor(mge): add a default grad=None attribute to Tensor
- refactor(mge): deprecation for 1.0
GitOrigin-RevId: 3b723d9387
4 years ago
Megvii Engine Team
2beb65b19d
fix(mge/module): tensor shape will not work when constructing numpy array
GitOrigin-RevId: 5a0d705970
4 years ago
Megvii Engine Team
0067dcf068
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
4 years ago
Megvii Engine Team
40d18c8920
fix(mge/imperative): fix tests when shape is tensor
GitOrigin-RevId: fd0095c1ec
4 years ago
Megvii Engine Team
6b380e8965
feat(mge/imperative): run oss test and restore cmake list build items
GitOrigin-RevId: 11411b6964
4 years ago