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

108 Commits (3eb0505f9b9db06160fc1f4d2d8724856da9d46b)

Author SHA1 Message Date
  Megvii Engine Team 3eb0505f9b feat(imperative): add support for quantized conv transpose2d 3 years ago
  Megvii Engine Team a83098890b test(mge/utils): cover all test data 3 years ago
  Megvii Engine Team 0b4a767965 feat(mge/distributed): enable uint8 for collective communication 3 years ago
  Megvii Engine Team f76a2cc2c6 feat(mge/opr): add silu and gelu 3 years ago
  Megvii Engine Team 207a346351 chore(mge): run get_device_count("gpu") in subprocess 4 years ago
  Megvii Engine Team 070c811732 fix(imperative): remove convert_inputs 3 years ago
  Megvii Engine Team 1040b77843 fix(mge/functional): fix F.topk(kth_only=True) 4 years ago
  Megvii Engine Team 1bf18252c4 feat(mge/amp): add mix precision autocast support 4 years ago
  Megvii Engine Team 43098fb8f1 feat(mge): add SlidingWindowTranspose opr 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 6cd01d5a74 feat(imperative/functional): let elemwise support empty IO & add some tests 4 years ago
  Megvii Engine Team 5198b7834c fix(mge/functional): fix expand_dims for scalar 4 years ago
  Megvii Engine Team 933dd9a497 feat(mge/distributed): add cuda env check before forked thread 4 years ago
  Megvii Engine Team a5060a2bfe feat(mgb/opr): add check_has_inf kernel and opr 4 years ago
  Megvii Engine Team c64b1c94cd feat(imperative/functional): add roll in functional 4 years ago
  Megvii Engine Team d2673c5abf fix(ci/windows): add windows cuda test 4 years ago
  Megvii Engine Team 1525a02530 feat(mge/module): add python wrapper for unfold 4 years ago
  Megvii Engine Team 13b15fb08c feat(megbrain): add correlation opr 4 years ago
  Megvii Engine Team 226f24fa8e test(mge/distributed): add gather scatter reduce broadcast grad test 4 years ago
  Megvii Engine Team b1bf193e4d feat(functional/loss): add reduction choices to loss functions 4 years ago
  Megvii Engine Team 717b88e684 fix(mge/elemwise): fix problem that elemwise.mode is not comparable with string mode 4 years ago
  Megvii Engine Team 984d85ca26 feat(mge/functional): argmin and argmax support negtive axis 4 years ago
  Megvii Engine Team 6972896933 fix(mge/utils): fix network multiple outputs issue 4 years ago
  Megvii Engine Team 4d9073fcba test(dist): remove redundant assert for reduce_sum and gather 4 years ago
  Megvii Engine Team dee5a10af9 feat(distributed): auto detect device and backend when init group 4 years ago
  Megvii Engine Team 933188aa93 feat(functional/nn): support F.warp_perspective with `mat_idx` 4 years ago
  Megvii Engine Team 65af9cfcdc refactor(mge): use lower case for default string parameters in functional and module 4 years ago
  Megvii Engine Team 9dbe71dd1a feat(mge/utils): add array method for varnode 4 years ago
  Megvii Engine Team 988f897ee2 test(mge): make test stable 4 years ago
  Megvii Engine Team fa1ca0ea6c fix(imperative/opr): fix apply_on_var_node for broadcast 4 years ago
  Megvii Engine Team d970b85deb feat(functional): add conv3d 4 years ago
  Megvii Engine Team 46cad4d3c8 feat(functional/ops): add _assert_equal 4 years ago
  Megvii Engine Team dcfb6a537e refactor(mge/functional): move functional api 4 years ago
  Megvii Engine Team fe9c6e26a8 feat(imperative/opr): deprecate resize op and make it as a special case of interpolate 4 years ago
  Megvii Engine Team fa4bf16800 feat(mge/functional): add repeat and tile opr 4 years ago
  Megvii Engine Team 0c37a588ca fix(mge/functional): fix F.ones when input is a tensor of scalar type 4 years ago
  Megvii Engine Team 1d7dd00144 feat(mge/quantization): add QParams and QuantDtypeMeta for quantization data structure 4 years ago
  Megvii Engine Team d0f70a445d fix(imperative/utils): module parameters' name do not have scope after dumping 4 years ago
  Megvii Engine Team cbff4d7c1a fix(imperative/ops): fix infer_output_attrs_fallible for reshape 4 years ago
  Megvii Engine Team 02df634da2 fix(mge/quantization): set ``q_dict`` as an instance property 4 years ago
  Megvii Engine Team cd02d7c85e feat(mge/opr): support deformable conv2d/psroi_pooling 4 years ago
  Megvii Engine Team 4485e780ae fix(mge/functional): fix linspace device and open other trace tests 4 years ago
  Megvii Engine Team 334eda8717 refactor(mge): test trace inside opr_test 4 years ago
  Megvii Engine Team 938152af63 fix(mge/functional): convert input type to float32 for more elemwise op 4 years ago
  Megvii Engine Team 194660462c feat(mge/funcitonal): add cvt_color opr python interface 4 years ago
  Megvii Engine Team ff05667b48 test(dist): refactor distributed test with fixtures 4 years ago
  Megvii Engine Team e3a3e0cd3c feat(mge/functional): add warp_affine 4 years ago
  Megvii Engine Team 364afec033 chore(mge): update copyright years 4 years ago
  Megvii Engine Team 4aaae9950b feat(mge/functional): add python wrapper to resize opr 4 years ago
  Megvii Engine Team df976782fa feat(mge/functional): add matinv 4 years ago