Megvii Engine Team
74cbc10d82
feat(opr): let batchnorm support empty IO
GitOrigin-RevId: 219411c80c
3 years ago
Megvii Engine Team
a7a3bf2d6c
test(subgraph): simple test for subgraph
GitOrigin-RevId: 3d6ecd5db7
3 years ago
Megvii Engine Team
207a346351
chore(mge): run get_device_count("gpu") in subprocess
GitOrigin-RevId: 0f0dc001cf
4 years ago
Megvii Engine Team
ff05667b48
test(dist): refactor distributed test with fixtures
GitOrigin-RevId: e69acb72da
4 years ago
Megvii Engine Team
364afec033
chore(mge): update copyright years
GitOrigin-RevId: 3c0690bcc1
4 years ago
Megvii Engine Team
9d928e7f83
refactor(mge/distributed): sync interpreter for distribtued launcher
GitOrigin-RevId: 8a88c272a1
4 years ago
Megvii Engine Team
ee4ea7fdc8
test(distributed/test): make distributed test more stronger
GitOrigin-RevId: 085fd1dcfd
4 years ago
Megvii Engine Team
33e8879af4
feat(mge/quantization): support distributed qat
GitOrigin-RevId: c915c843b8
4 years ago
Megvii Engine Team
dfc3935d06
fix(pytest/windows/impertive): fix windows wrong test skip
GitOrigin-RevId: ca0f04cfe7
4 years ago
Megvii Engine Team
9748aebeea
refactor(mge): tensor_shape -> symbolic_shape
GitOrigin-RevId: 366dc048bf
4 years ago
Megvii Engine Team
2efba9a3a3
fix(mgb/test): use both rtol and atol for stable test result
GitOrigin-RevId: 82a1453e4a
4 years ago
Megvii Engine Team
6ce796aec4
test(mge): replace `assertTensorClose` with `np.testing.assert_allclose`
GitOrigin-RevId: 2bf8c42cfd
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
a226f02e7a
fix(mge/imperative): fix syncbn in symbolic mode
GitOrigin-RevId: a9794318a7
4 years ago
Megvii Engine Team
0067dcf068
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
4 years ago