wangwei
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
1805 Commits
28 Branches
25 MB
548 Download
Tree: 5431929ec7
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '5431929ec7'
${ noResults }
Commit Graph

338 Commits (5431929ec79d0b7ba7550ca3e521051d8ed679b0)

Author SHA1 Message Date
  Megvii Engine Team 5431929ec7 feat(functional): let advance indexing support empty tensor and add more tests 4 years ago
  Megvii Engine Team a430c912fa feat(mgb/opr): let CondTake support empty input 4 years ago
  Megvii Engine Team 432fdb7e6a feat(mgb/opr): let SetSubtensor support empty IO 4 years ago
  Megvii Engine Team e954b8f998 feat(mgb/opr): let Subtensor support empty IO 4 years ago
  Megvii Engine Team 03e80759b6 feat(mge/distributed): add hybird parallel Opr 4 years ago
  Megvii Engine Team e5217db9f9 feat(traced_module): move traced_module out of the experimental folder 4 years ago
  Megvii Engine Team 6d1a4f20e7 feat(traced_module): support tracing submodules in list/dict 4 years ago
  Megvii Engine Team a3f9073c2c feat(traced_module): update graph transform and add _module_name 4 years ago
  Megvii Engine Team c7e730bc12 feat(traced_module): add some functions of graph modification 4 years ago
  Megvii Engine Team f88bd3ae33 refactor(traced_module): let TracedModule own argdef_graph_map 4 years ago
  Megvii Engine Team b1c46ba46c feat(traced_module): add some functions of graph modification 4 years ago
  Megvii Engine Team 442b4f6c26 test(traced_module): add some testcases for traced module 4 years ago
  Megvii Engine Team 9279104b11 feat(mge): add opdef serialization and apply_module_trace 4 years ago
  Megvii Engine Team d86ed426ee fix(dtr): simulate the system stack to avoid stack overflow during recomputing 4 years ago
  Megvii Engine Team a7a3bf2d6c test(subgraph): simple test for subgraph 4 years ago
  Megvii Engine Team c54a731b9b fix(utils/network): fix replace oprs 4 years ago
  Megvii Engine Team ebb7f5cac2 fix(imperative/python/test): add device judgment for the test case of conv-transpose2d-quantized 4 years ago
  Megvii Engine Team 2950dd8d69 test(imperative): test autodiff.Function with non tensor arguments 4 years ago
  Megvii Engine Team b74afde89e feat(mgb/opr): let reduce support empty IO 4 years ago
  Megvii Engine Team 3eb0505f9b feat(imperative): add support for quantized conv transpose2d 4 years ago
  Megvii Engine Team a83098890b test(mge/utils): cover all test data 4 years ago
  Megvii Engine Team 0b4a767965 feat(mge/distributed): enable uint8 for collective communication 4 years ago
  Megvii Engine Team 3f3a256e0f fix(mge/functional): fix conv* dtype promotion 4 years ago
  Megvii Engine Team f76a2cc2c6 feat(mge/opr): add silu and gelu 4 years ago
  Megvii Engine Team f141159088 refactor(mge): loose the error bound of fastrun 4 years ago
  Megvii Engine Team 1f0436967c refactor(mge/distributed): using nccl as default in distributed training 4 years ago
  Megvii Engine Team 207a346351 chore(mge): run get_device_count("gpu") in subprocess 5 years ago
  Megvii Engine Team e6caa9ff89 feat(opr): add bn backward for inference mode 4 years ago
  Xinda Huang c90fa087ea test(mge): delete test_external.py 4 years ago
  Megvii Engine Team b2944559a8 fix(imperative/module): remove ``__getattribute__`` method in module 4 years ago
  Megvii Engine Team 070c811732 fix(imperative): remove convert_inputs 4 years ago
  Megvii Engine Team 1040b77843 fix(mge/functional): fix F.topk(kth_only=True) 4 years ago
  Megvii Engine Team 7c9569e4e5 fix(mge/random): fix random seed 4 years ago
  Megvii Engine Team d7b6bfd56c test(mge/fakequant): use fixed input for lsq test to temperarily avoid precision error 4 years ago
  Megvii Engine Team 5cef74a77e feat(mge/amp): add GradScaler support 4 years ago
  Megvii Engine Team 1bf18252c4 feat(mge/amp): add mix precision autocast support 4 years ago
  Megvii Engine Team dedecf6922 fix(imperative/utils): fix logical error of replace var 4 years ago
  Megvii Engine Team 43098fb8f1 feat(mge): add SlidingWindowTranspose opr 4 years ago
  Megvii Engine Team df79334cae feat(mge/distributed): add user_pop function to save device memory 4 years ago
  Megvii Engine Team 7225b0f09f fix(mge/utils): use static infer manager to get value of network.varnode 4 years ago
  Megvii Engine Team 2d42455fa8 fix(mge/utils): fix toposort to get definition order 4 years ago
  Megvii Engine Team 427113088b fix(module/normalization): fix bug of LayerNorm and support input of any shape 4 years ago
  Megvii Engine Team b078dda90b feat(mge/random): add some random op and remove random/distrbution.py 4 years ago
  Megvii Engine Team f30c0e06a6 feat(mgb/opr): add lsq opr 4 years ago
  Megvii Engine Team d6db4fea93 fix(mge/module): set no_cache=true when loading state dict 4 years ago
  Megvii Engine Team fea1bba23e fix(mge/tools): fix module status error 4 years ago
  Megvii Engine Team 6cd01d5a74 feat(imperative/functional): let elemwise support empty IO & add some tests 4 years ago
  Megvii Engine Team 0245594156 test(autograd): test jvp emulated by 2nd grad 4 years ago
  Megvii Engine Team 8480302da8 fix(autograd): make higher order grad experimental 4 years ago
  Megvii Engine Team 72531f2b4a test(autograd): add more tests for higher order grad 4 years ago