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
Megvii Engine Team
aa20404027
feat(lite): add lite static all in one
GitOrigin-RevId: 7e6d15d929
3 years ago
Megvii Engine Team
a0231a7920
fix(dnn/cuda): fix algo matmul for conv bwd filter
fix fastrun workspace size not available exception and device OOM error caused by the incorrect workspace size calculation of algo matmul of conv bwd filter
GitOrigin-RevId: de96b4fe11
3 years ago
Megvii Engine Team
f3ed59d336
feat(dnn/opencl): add heuristic rule for elemwise
GitOrigin-RevId: 2bc574b5a7
3 years ago
Megvii Engine Team
29d24dbb80
fix(mge/function): fix interpolate unsupport fp16 error
GitOrigin-RevId: 7fc6271986
3 years ago
Megvii Engine Team
36df3850f3
test(mgb): remove the padding random test case
padding random test case in ci 2080ti will failed in random
GitOrigin-RevId: b83e7b8799
3 years ago
Megvii Engine Team
e21967bb40
feat(mge): add env MGE_FASTRUN_CACHE_DIR
GitOrigin-RevId: 0351ff88d1
3 years ago
Megvii Engine Team
6a1ec8a890
feat(mge): add git commit-id into fastrun cache key
GitOrigin-RevId: fb614b4d9c
3 years ago
Megvii Engine Team
ae87876d34
feat(mge): refactor weightscaler
GitOrigin-RevId: 7f874388f7
3 years ago
Megvii Engine Team
5d9ac970ab
fix(mgb): fix fastrun compnode
GitOrigin-RevId: 8db93facb9
3 years ago
Megvii Engine Team
56c1b626bf
refactor(dnn): move arch-dependant code to arch.h
GitOrigin-RevId: 52350144b1
3 years ago
Megvii Engine Team
67575d582c
feat(mge/opr): add interpolate bilinear mode
GitOrigin-RevId: f7023a3fd3
3 years ago