chenjiahui
6b863cc53a
feat(imperative): add pixel_shuffle opr
Closes #211
GitOrigin-RevId: 1846714791
3 years ago
Megvii Engine Team
5e34504377
fix(cmake/bazel/clang): remove finite-math-only opt from Ofast,
for keep same build between gcc and clang
more detail:
```
printf("%d\n", std::isnan(std::numeric_limits<float>::quiet_NaN()));
printf("%d\n", std::isnan(std::nan("1")));
```
linux-clang and android-NDK clang have diff build logic with gcc and
macos/windows clang/clangcl
* clang++-12 -Ofast 1.cc
output is:
0
0
* clang++-12 -Ofast 1.cc -fno-finite-math-only
output is:
1
1
* g++ -Ofast 1.cc
output is:
1
1
GitOrigin-RevId: f0622e2ca0
3 years ago
Megvii Engine Team
cccddbc09b
docs(lite): opt pylite md
GitOrigin-RevId: df332ad8b7
3 years ago
Megvii Engine Team
b982be567a
feat(mge/imperative): add permutation support for the tensor
GitOrigin-RevId: 7ed0447bfe
3 years ago
Megvii Engine Team
3977b7aa0b
feat(mgb/shuffle): add shuffle opr
GitOrigin-RevId: 80490a6f84
3 years ago
Megvii Engine Team
17371e79b9
fix(dnn/reduce): fix reduce_mean o16c32 is incorrect for large tensor
GitOrigin-RevId: ebf03d814a
3 years ago
Megvii Engine Team
884a07ffd2
fix(test/random): set a random seed for random unit test
GitOrigin-RevId: ad4b01eac7
3 years ago
Megvii Engine Team
d7cc4628f6
perf(gopt): opt concat for OpenCL
GitOrigin-RevId: 9bb226d4b1
3 years ago
Megvii Engine Team
3f0bb47a43
ci(imperative): add test requires
GitOrigin-RevId: d7a945caf3
3 years ago
Megvii Engine Team
34be1d6e57
feat(mgb): dump static memory info by tensorboard
GitOrigin-RevId: 2020ad81f9
3 years ago
Megvii Engine Team
11a383bc79
feat(mgb): add tensorboard tool c++ layer interface and python parsing
GitOrigin-RevId: ae98bc887f
3 years ago
Megvii Engine Team
8084e4e2ee
feat(mgb): add tensorboard tool python layer interface
GitOrigin-RevId: 065bc4d153
3 years ago
Megvii Engine Team
97b1b7774d
feat(mgb): add megbrain layer norm opr with subgraph
GitOrigin-RevId: 9b7fa821f8
3 years ago
Megvii Engine Team
eca6e1d931
fix(ci): fixes for ci
GitOrigin-RevId: b0a432bd2e
3 years ago
Megvii Engine Team
19d7412a0a
refactor(mgb/gopt): reorganize code of global layout transform
GitOrigin-RevId: 4973820e02
3 years ago
Megvii Engine Team
8ef12bdf22
feat(mgb/gopt): add user inferface for global layout transform
GitOrigin-RevId: b71d6c60ef
3 years ago
Megvii Engine Team
a3cd3fc74f
test(mgb/gopt): add testcase for global layout transform
GitOrigin-RevId: f9669e1ba0
3 years ago
Megvii Engine Team
af576e9ae0
fix(mgb/gopt): fix auto padding for nhwc layout
GitOrigin-RevId: 038e372cbe
3 years ago
Megvii Engine Team
af828ca9ec
feat(mgb/gopt): fix folding conv dimshuffle pass
GitOrigin-RevId: 756878c173
3 years ago
Megvii Engine Team
c67c4b7ddc
feat(mgb/gopt): add layout transform pass
GitOrigin-RevId: 9ed5f5782e
3 years ago
Megvii Engine Team
2ec7c16725
feat(mgb/gopt): profiler support opr filter and var node filter
GitOrigin-RevId: 5f8d86687f
3 years ago
Megvii Engine Team
50ea5ae8e2
feat(mgb/gopt): add dynamic programming solver
GitOrigin-RevId: 595392ec89
3 years ago
Megvii Engine Team
c14e5719f8
feat(mgb/gopt): add profile impl for global layout transform pass
GitOrigin-RevId: 8ef62baf79
3 years ago
Megvii Engine Team
9c0a17d07b
feat(mgb/gopt): add auto aligned reformat impls
GitOrigin-RevId: fd0814fdb3
3 years ago
Megvii Engine Team
2ed76b1660
feat(mgb/gopt): add graph dumper for graph partition
GitOrigin-RevId: 6dbcb67009
3 years ago
Megvii Engine Team
76b2840858
feat(mgb/gopt): add subgraph extractor
GitOrigin-RevId: 56fd701c2c
3 years ago
Megvii Engine Team
8a3eb05a1b
refactor(mgb/gopt): refactor tensor reformat opt pass
GitOrigin-RevId: a1b1e89b76
3 years ago
Megvii Engine Team
c33126ab5c
feat(mgb/gopt): add reformat manager
GitOrigin-RevId: b9791b131a
3 years ago
Megvii Engine Team
65d554ed8e
refactor(dtr): improve dtr in computing graph
GitOrigin-RevId: d599a08e52
3 years ago
Megvii Engine Team
4f28e14684
fix(dnn): fix compatibility broken of convolution format
GitOrigin-RevId: c17f02d03d
3 years ago
Megvii Engine Team
f642b05e17
test(mge): update traced_module unit test
GitOrigin-RevId: 3948d50d79
3 years ago
Megvii Engine Team
fb20cb36a0
docs(mge/traced_module): update traced_module api doc
GitOrigin-RevId: 19a95d26c7
3 years ago
Megvii Engine Team
c7a8d945c7
fix(mge/traced_module): let graph record total_id
GitOrigin-RevId: f99178f3ac
3 years ago
Megvii Engine Team
8b40f57738
feat(mgb/dnn): add conv1x1 algo for matrix mul
GitOrigin-RevId: 585b2c045a
3 years ago
Megvii Engine Team
4de62ad6cd
docs(mge/device): fix docstring of get_cuda_compute_capability
GitOrigin-RevId: 4ec04e8b9b
3 years ago
Megvii Engine Team
fb49a2834f
refactor(mgb/dnn): refactor enum used in serializing
GitOrigin-RevId: e57af4a59c
3 years ago
Megvii Engine Team
d69b59035d
feat(dnn): add an get_all_algorithms_safe interface
GitOrigin-RevId: e3734e4531
3 years ago
Megvii Engine Team
103d7f33ba
refactor(dnn/rocm): update hip license header
GitOrigin-RevId: 79d684755d
4 years ago
Megvii Engine Team
5aa52d3863
feat(dnn/rocm): add adaptive pooling opr
GitOrigin-RevId: e844b3e770
3 years ago
Megvii Engine Team
83cf4ee64e
refactor(dnn/rocm): remove some useless includes
GitOrigin-RevId: 3d2c315a36
3 years ago
Megvii Engine Team
323a4642e6
feat(dnn/rocm): add topk opr
GitOrigin-RevId: 5ecb079854
3 years ago
Megvii Engine Team
f4784f4af1
feat(dnn/rocm): add argsort opr
GitOrigin-RevId: b4c3eb4707
3 years ago
Megvii Engine Team
6082c353e7
feat(dnn/rocm): support bool in type_cvt and elemwise
GitOrigin-RevId: ad5ec7bc1c
4 years ago
Megvii Engine Team
0ad377c7cf
fix(mgb/gopt): add error message when input dtype is not equal to param dtype in BN2Elemwise pass
GitOrigin-RevId: 3d09a2a12e
3 years ago
Megvii Engine Team
93f70a9539
feat(src): add MGB_CUSTOM_OP macro
GitOrigin-RevId: beddcf9745
3 years ago
Megvii Engine Team
b3e958d0bc
fix(src): fix the warnings and copy.bara.sky in custom op
GitOrigin-RevId: 4ade455897
3 years ago
Megvii Engine Team
cdb692d2fa
refactor(imperative): add TODO tag for some functions
GitOrigin-RevId: e295a1fa55
3 years ago
Megvii Engine Team
90dd07161c
refactor(imperative): modify the python interface of custom op
GitOrigin-RevId: e82e5de480
3 years ago
Megvii Engine Team
cbf024bf20
feat(imperative): add adaptor between custom op and imperative runtime
GitOrigin-RevId: d7877f2e32
3 years ago
Megvii Engine Team
39ba30214d
feat(src): add adaptor between custom op and graph runtime
GitOrigin-RevId: 834c0ac72b
3 years ago