139 Commits (68cde8734e735361d40a0a6cf48b920c0cb8ff45)

Author SHA1 Message Date
  Megvii Engine Team 68cde8734e fix(mge/imperative): support broadcast with None 3 years ago
  Megvii Engine Team 0bdd0b1467 refactor(dispatch): switch to new dispatch system 3 years ago
  Megvii Engine Team 1999307015 feat(mgb/opr): add dropout kernel 3 years ago
  Megvii Engine Team a93741815b feat(mgb/opr): add layernorm forward and backward kernel 3 years ago
  Megvii Engine Team d9a46ea47b fix(dnn): correct behaviour of floor div for int tensor 3 years ago
  Megvii Engine Team 2881934cb8 feat(dnn/check_non_finite): addmul scale to check_non_finite opr 3 years ago
  Megvii Engine Team 1f0cc891b0 feat(dnn): enable eye to support bool 3 years ago
  Megvii Engine Team 11d75fecb5 feat(dnn/check_non_finite): add batch check_non_finite 3 years ago
  Megvii Engine Team 89186edc5d fix(dnn): correct reduce/argmxx/fakequant calculation with nan 3 years ago
  Megvii Engine Team 5c94db16f2 feat(mge/functional): add groups support for conv_transpose2d & 3d 3 years ago
  Megvii Engine Team 7591718d20 feat(mge): add functional test 3 years ago
  Megvii Engine Team 846cc13ca8 fix(mge/functional): fix default dtype of F.full 3 years ago
  Megvii Engine Team 177dec94c5 feat(mgb/opr): add bgr2gray mode for cvtcolor opr 3 years ago
  Megvii Engine Team cebda6ff8d feat(mge/imperative): add ctc loss 3 years ago
  Megvii Engine Team f5cb21ed3a fix(mgb/opr): add non finite check 3 years ago
  Megvii Engine Team dc28a43409 fix(mgb/bn): fix empty tensor input problem and other minor problems 3 years ago
  Megvii Engine Team c0ccd0ea7e feat(mge/bn): add NHWC support for bn 3 years ago
  chenjiahui 6b863cc53a feat(imperative): add pixel_shuffle opr 3 years ago
  Megvii Engine Team 97b1b7774d feat(mgb): add megbrain layer norm opr with subgraph 3 years ago
  Megvii Engine Team 43fb2c4afb feat(opr): let roll support empty IO 3 years ago
  Megvii Engine Team b2827cb1b1 feat(opr): let Dot, MatrixMul and BatchedMatrixMul support empty input 3 years ago
  Megvii Engine Team 50f738772b feat(opr): let Copy and Identity support empty IO 3 years ago
  Megvii Engine Team 8ae61752a0 feat(opr): let nms support empty IO 3 years ago
  Megvii Engine Team 1a1748daf1 feat(opr): let Argsort support empty IO 3 years ago
  Megvii Engine Team 43cb305e53 feat(mge): add imperative pad 3 years ago
  Megvii Engine Team ab309eb5fc feat(mgb/opr): let Split support empty IO 3 years ago
  Megvii Engine Team 76ce81e828 fix(mge): fix F.nn.dropout train and inference bugs 3 years ago
  Megvii Engine Team 5431929ec7 feat(functional): let advance indexing support empty tensor and add more tests 3 years ago
  Megvii Engine Team a430c912fa feat(mgb/opr): let CondTake support empty input 3 years ago
  Megvii Engine Team 03e80759b6 feat(mge/distributed): add hybird parallel Opr 3 years ago
  Megvii Engine Team b74afde89e feat(mgb/opr): let reduce support empty IO 3 years ago
  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