149 Commits (0bed6c0fd9f27a87194c16d3b612ec117db2983c)

Author SHA1 Message Date
  Megvii Engine Team d984be591d fix(imperative): restrict value converts to symbolvar 3 years ago
  Megvii Engine Team 94960ecf42 fix(imperative): restrict using convert_inputs in py_apply 3 years ago
  Megvii Engine Team 9779bc7f6d fix(imperative): allow rng op infer shape fallible 3 years ago
  Megvii Engine Team 1709b3940b perf(mge/functional): speed up Broadcast and Reshape 3 years ago
  Megvii Engine Team 27346b0b65 test(opr): add scalar check for opr_test 3 years ago
  Megvii Engine Team df3474ca1d perf(functional): rewrite serval elemwise ops with jit subgraph 3 years ago
  Megvii Engine Team d919aaebc7 test(imperative): reopen special interpolate test and sync when test rng 3 years ago
  Megvii Engine Team 5e17b3e4c6 Merge pull request #426 from Qsingle:fix-pixel_suffle 3 years ago
  Megvii Engine Team 95ac055538 feat(dnn,mgb,imperative): add diag opr implement 3 years ago
  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
  Qsingle 14499e83d9 fix the type error of pixel shuffle 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