Megvii Engine Team
43cb305e53
feat(mge): add imperative pad
GitOrigin-RevId: de79de536f
3 years ago
Megvii Engine Team
567586a037
feat(mgb): fastrun algo profile deduplication
GitOrigin-RevId: 0d1bed781d
3 years ago
Megvii Engine Team
8110bb219a
fix(cmake/env): fix a bug about env set in cmake build mode
GitOrigin-RevId: 492eaf428a
3 years ago
Megvii Engine Team
0a665ea452
feat(mge/device): enable to get cuda compute capability
GitOrigin-RevId: b5d3f2225c
3 years ago
Megvii Engine Team
722aecd437
feat(mgb): support fp16 nhwc backward
GitOrigin-RevId: 954ac6405a
3 years ago
Megvii Engine Team
7a9f2ed95e
perf(functional/split): add python binding for Split opr
GitOrigin-RevId: 3359f5da26
3 years ago
Megvii Engine Team
ab309eb5fc
feat(mgb/opr): let Split support empty IO
GitOrigin-RevId: aad6dc06bf
3 years ago
Megvii Engine Team
a829270489
fix(mge/tracing): replace detach as fast path copy
GitOrigin-RevId: d765725d5a
3 years ago
Megvii Engine Team
8485eff12c
docs(mge/functional): add docs for megengine.functional.sub
GitOrigin-RevId: 40208739bd
3 years ago
Megvii Engine Team
32ad926559
feat(mge): improve presentation of async errors
GitOrigin-RevId: af5426c37d
3 years ago
Megvii Engine Team
e6a8b0256e
fix(mge): ignore errors caused by earlier async errors
GitOrigin-RevId: ce2028d38a
3 years ago
Megvii Engine Team
0708bc780c
fix(dnn/cuda): disallow implicit dtype conversion in cublaslt matmul algos
disable tensor op matmul kernels when input and output tensors are in f32 data type to avoid potential accuracy loss
GitOrigin-RevId: 36859cba5a
3 years ago
Megvii Engine Team
3f01112ac2
build(require): update pytest-sphinx version requirement
GitOrigin-RevId: 6da2fe8f4c
3 years ago
Megvii Engine Team
ee790cb62c
fix(imperative/python): fix warp_perspective doc and arange dtype
GitOrigin-RevId: dc4e021ec0
3 years ago
Megvii Engine Team
c5f8b58385
docs(docstring): transfer to google style
GitOrigin-RevId: a71245c553
3 years ago
Megvii Engine Team
76ce81e828
fix(mge): fix F.nn.dropout train and inference bugs
GitOrigin-RevId: 9d9f246d7b
3 years ago
Megvii Engine Team
5431929ec7
feat(functional): let advance indexing support empty tensor and add more tests
GitOrigin-RevId: 49e1492934
3 years ago
Megvii Engine Team
703b783c24
feat(mgb/opr): let Indexing(Set)MultiAxisVec support empty input
GitOrigin-RevId: f15b1d45a1
3 years ago
Megvii Engine Team
a430c912fa
feat(mgb/opr): let CondTake support empty input
GitOrigin-RevId: dfb401a945
3 years ago
Megvii Engine Team
432fdb7e6a
feat(mgb/opr): let SetSubtensor support empty IO
GitOrigin-RevId: 13909e3b11
3 years ago
Megvii Engine Team
e954b8f998
feat(mgb/opr): let Subtensor support empty IO
GitOrigin-RevId: a768498104
3 years ago
Megvii Engine Team
1e83ab638e
feat(dnn): add channelwise conv for fp16 nchw88
GitOrigin-RevId: 1bb64f82c5
3 years ago
Megvii Engine Team
28c066ee4f
fix(cmake/llvm): fix llvm build if env install llvm-12-dev
now, we do not depends LLVM_ENABLE_BINDINGS and LLVM_INCLUDE_DOCS module
so disable it! if later depends on this module, need upgrade
third_party/llvm-project/ to 12 to compat build env which install
llvm-12-dev
GitOrigin-RevId: 62164c0e7f
3 years ago
Megvii Engine Team
7b855dc64a
fix(dnn/cuda): fix compilation for windows bazel
GitOrigin-RevId: 2023dea19c
3 years ago
Megvii Engine Team
3abe0b2462
fix(mgb): fix rocm pooling
GitOrigin-RevId: 44876d398e
3 years ago
Megvii Engine Team
f9722af3df
fix(src/json): fix parsing error when inf is dumped as a Number
GitOrigin-RevId: 62cfba48c0
3 years ago
Megvii Engine Team
16678bb998
fix(dnn): fix_short_cutlass_name_gemm
GitOrigin-RevId: cc0a2db9da
3 years ago
huangxinda
6f455ae906
feat(mge/third_party): update cutlass version
3 years ago
zhanghaolong
7a0d9626f0
opt(cpuifno): move cpuinfo from pytorch to MegEngine
3 years ago
Megvii Engine Team
4c13bc7e1b
feat(dnn/cuda): add nhwc int8 deconv
GitOrigin-RevId: ad361a0f81
3 years ago
Megvii Engine Team
11f022ff7c
feat(dnn/cuda): add nhwc int8 imma conv and conv fuse typecvt
GitOrigin-RevId: 229e1eb4be
3 years ago
Megvii Engine Team
03e80759b6
feat(mge/distributed): add hybird parallel Opr
GitOrigin-RevId: ff26671746
3 years ago
Megvii Engine Team
61a5df321e
refactor(third-party/cpuinfo): opt cpuinfo repo
GitOrigin-RevId: 1c3a823b64
3 years ago
Megvii Engine Team
a4f0e58180
fix(mgb/extern_c_opr): throw exception when extern c opr loader was created
GitOrigin-RevId: bf48e1f3bb
3 years ago
Megvii Engine Team
e5217db9f9
feat(traced_module): move traced_module out of the experimental folder
GitOrigin-RevId: 36c76b5277
3 years ago
Megvii Engine Team
6d1a4f20e7
feat(traced_module): support tracing submodules in list/dict
GitOrigin-RevId: 4076b47a89
3 years ago
Megvii Engine Team
a3f9073c2c
feat(traced_module): update graph transform and add _module_name
GitOrigin-RevId: ef63ae0fd0
3 years ago
Megvii Engine Team
b3d0affab1
feat(traced_module): support trace custom qat module
GitOrigin-RevId: 49f70a5f46
3 years ago
Megvii Engine Team
15712807b9
feat(traced_module): add name to Node
GitOrigin-RevId: 39c2809067
3 years ago
Megvii Engine Team
e918f0aa75
feat(traced_module): add treedef leaf node check and add some graph api
GitOrigin-RevId: 36c069bfee
3 years ago
Megvii Engine Team
c7e730bc12
feat(traced_module): add some functions of graph modification
GitOrigin-RevId: 09691ebd33
3 years ago
Megvii Engine Team
f88bd3ae33
refactor(traced_module): let TracedModule own argdef_graph_map
GitOrigin-RevId: 80d685b9a3
3 years ago
Megvii Engine Team
b1c46ba46c
feat(traced_module): add some functions of graph modification
GitOrigin-RevId: ac0603057a
3 years ago
Megvii Engine Team
4bb253695b
feat(traced_module): let CallFunction own graph
GitOrigin-RevId: 66cdbca7e5
3 years ago
Megvii Engine Team
9a6a379346
feat(traced_module): add visit method
GitOrigin-RevId: 251ecebf87
4 years ago
Megvii Engine Team
442b4f6c26
test(traced_module): add some testcases for traced module
GitOrigin-RevId: 0d6bb20b2b
4 years ago
Megvii Engine Team
f2691566fd
feat(traced_module): add pytree
GitOrigin-RevId: 6c6e53521c
4 years ago
Megvii Engine Team
bee305beb2
feat(traced_module): add functional trace and CallMethod/Function expr
GitOrigin-RevId: ad2cdc1b61
4 years ago
Megvii Engine Team
763c56f3b9
feat(imperative): add traced module
GitOrigin-RevId: 28c3503f2e
4 years ago
Megvii Engine Team
9279104b11
feat(mge): add opdef serialization and apply_module_trace
GitOrigin-RevId: 5b45bded1d
3 years ago