Megvii Engine Team
|
478f2c51dc
|
feat(mge/jit): add trace/dump options
GitOrigin-RevId: 0f43c14599
|
4 years ago |
Megvii Engine Team
|
44d0b5daf5
|
feat(imperative): enable to() to copy to device
GitOrigin-RevId: f9caf17d24
|
4 years ago |
Megvii Engine Team
|
69a7c55f55
|
test(mge/function): fix test for new optimizer api
GitOrigin-RevId: 8ae7720fe6
|
4 years ago |
Megvii Engine Team
|
66b6daf777
|
test(mge/optimizer): fix test for new optimizer api
GitOrigin-RevId: 482ee62652
|
4 years ago |
Megvii Engine Team
|
e283663a02
|
fix(mge/imperative): update tests to new optimizer api
GitOrigin-RevId: 3d06e3db3c
|
4 years ago |
Megvii Engine Team
|
67859f04e1
|
fix(imperative): add __array__ and __array_wrap__ for tensorwrapper
GitOrigin-RevId: 87d4ab6c8e
|
4 years ago |
Megvii Engine Team
|
5aa19f3d4a
|
test(mge/imperative): add more testcases for function
GitOrigin-RevId: e0675994f1
|
4 years ago |
Megvii Engine Team
|
939bf3b28d
|
fix(mge/imperative): impl hashable for nms and cond_take
GitOrigin-RevId: 56918db014
|
4 years ago |
Megvii Engine Team
|
8dc23e0fdf
|
fix(mge/functional): fix indexing_one_hot and remote_recv
GitOrigin-RevId: 00bdfb502b
|
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
|
fc3eca845b
|
feat(mge/imperative/jit): add dump input shape and xornet example
GitOrigin-RevId: 5e2acd4052
|
4 years ago |
Megvii Engine Team
|
087ceb52ef
|
feat(mge/imperative): add more optimizer trace tests
GitOrigin-RevId: 4127de1d22
|
4 years ago |
Megvii Engine Team
|
403a1e7b24
|
feat(mge/imperative): add graph profiler
GitOrigin-RevId: c73563f337
|
4 years ago |
Megvii Engine Team
|
d06f248d90
|
fix(whl/mgb/imperative): fix symbols conflict runtime crash
GitOrigin-RevId: d60c4052c1
|
4 years ago |
Megvii Engine Team
|
1b2194cdce
|
feat(whl/internal): put imperative rt to megbrain
GitOrigin-RevId: f3b6e492d7
|
4 years ago |
Megvii Engine Team
|
a3b2232ba7
|
feat(mge): add trace.dump
GitOrigin-RevId: ea4c9d33c8
|
4 years ago |
Megvii Engine Team
|
0e82b959a1
|
feat(mge/imperative): add sublinear options
GitOrigin-RevId: f0e917f716
|
4 years ago |
Megvii Engine Team
|
afddefb677
|
feat(mge/imperative): add more trace test
GitOrigin-RevId: b02e420a8a
|
4 years ago |
Megvii Engine Team
|
d4bad71190
|
feat(mge): add jit.trace
GitOrigin-RevId: ec647324c0
|
4 years ago |
Megvii Engine Team
|
5474538f39
|
refactor(mge/imperative): fork multipledispatch
GitOrigin-RevId: a7c25a4302
|
4 years ago |
Megvii Engine Team
|
03320a0548
|
fix(functional): change to user-friendly nms api and fix concat
GitOrigin-RevId: 76b4190270
|
4 years ago |
Megvii Engine Team
|
46f03b9f7c
|
perf(mge/imperative): optimize gpu memory with standalone grad_fn
GitOrigin-RevId: 44952b235e
|
4 years ago |
Megvii Engine Team
|
a226f02e7a
|
fix(mge/imperative): fix syncbn in symbolic mode
GitOrigin-RevId: a9794318a7
|
4 years ago |
Megvii Engine Team
|
b9df2db1c5
|
build(third_party): update megray commit id
GitOrigin-RevId: cc753f941b
|
4 years ago |
Megvii Engine Team
|
0067dcf068
|
test(mge/imperative): add module test to imperative
GitOrigin-RevId: e690bc42b0
|
4 years ago |
Megvii Engine Team
|
7fadc16d3c
|
refactor(mge/functional): support tensor shape in interpolate and split
GitOrigin-RevId: 6430b64f01
|
4 years ago |
Megvii Engine Team
|
e1fba6ece7
|
test(mge/distributed): add get_device_count_by_fork to fix distributed test skip
GitOrigin-RevId: 9ffd8a6149
|
4 years ago |
Megvii Engine Team
|
60076f4715
|
test(mge/parampack): add parampack related opr test
GitOrigin-RevId: 550c6b6443
|
4 years ago |
Megvii Engine Team
|
7e8f720953
|
refactor(mge/tensor): tensor reduce supports keepdims
GitOrigin-RevId: 8ed95e0fb8
|
4 years ago |
Megvii Engine Team
|
6e882c1a86
|
feat(whl/imperative): compat for build python whl imperative and legacy runtime
GitOrigin-RevId: 7f6629ae1f
|
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
|
92f7cceb11
|
fix(mge/functional): fix cond_take when index is numpy array
GitOrigin-RevId: 5fb93740f4
|
4 years ago |
Megvii Engine Team
|
f4927db2fb
|
feat(mge/functional): support where func
GitOrigin-RevId: 9df6421ebe
|
4 years ago |
Megvii Engine Team
|
227ec04bd3
|
fix(mge/test): fix pytest failed issue
GitOrigin-RevId: 96df470406
|
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 |