96 Commits (66509a54b3528767eb9c0059af420a53940f20dd)

Author SHA1 Message Date
  Megvii Engine Team 66509a54b3 test(mgb): enable gmock 4 years ago
  Megvii Engine Team 91efd67dbf refactor(mge/jit): change dump options, add test 4 years ago
  Megvii Engine Team 099ffeac12 chore(mge): typo and reword 4 years ago
  Megvii Engine Team 39d4328309 fix(mge/functional): functional api fixes 4 years ago
  Megvii Engine Team e507228e74 feat(mge/examples): add distributed training examples using launcher 4 years ago
  Megvii Engine Team 523ce65e0d fix(mge/imperative): fix cgtools related tests 4 years ago
  Megvii Engine Team dd39265e95 fix(mgb/dtype): enable TypeCvt for bool when trace(symbolic=True) 4 years ago
  Megvii Engine Team 5c37a64e3f feat(sdk/load_and_run): add dump with testcase for mge support imperative 4 years ago
  Megvii Engine Team ac11c38a97 feat(mge/imperative): add graph load and cgtools for imperative 4 years ago
  Megvii Engine Team 76f367962f fix(mge/trace): fix op order in symbolic 4 years ago
  Megvii Engine Team 10e942d952 refactor(mge): polish api 4 years ago
  Megvii Engine Team 9389a805d8 fix(mge/oprmm): fix grad for collective comm 4 years ago
  Megvii Engine Team 6f581906a1 refactor(mge/profiler): refactor profiler 4 years ago
  Megvii Engine Team 5474b000a7 fix(mge/functional): fix convert_inputs before apply 4 years ago
  Megvii Engine Team 478f2c51dc feat(mge/jit): add trace/dump options 4 years ago
  Megvii Engine Team 44d0b5daf5 feat(imperative): enable to() to copy to device 4 years ago
  Megvii Engine Team d49a5c1be2 fix(mge/imperative): remove codes newly introduced in numpy 1.19 4 years ago
  Megvii Engine Team 69a7c55f55 test(mge/function): fix test for new optimizer api 4 years ago
  Megvii Engine Team a66d4b8bb8 fix(mge/parampacksplit): fix param pack split mem forward 4 years ago
  Megvii Engine Team bd3b9cb6d4 fix(mge/oprmm): fix grad for collective comm 4 years ago
  Megvii Engine Team 8d02d10483 refactor(mge/distributed): change bcast_params_ to bcast_list_ 4 years ago
  Megvii Engine Team c7acba41fc refactor(mge/optimizer): refine gradmanager api, record = __enter__ 4 years ago
  Megvii Engine Team 8c482b6709 fix(mge/grad): make register_after_backward_callback private 4 years ago
  Megvii Engine Team 66b6daf777 test(mge/optimizer): fix test for new optimizer api 4 years ago
  Megvii Engine Team e9104ef157 fix(mge/parampack): fix copy stream, import cycle 4 years ago
  Megvii Engine Team e283663a02 fix(mge/imperative): update tests to new optimizer api 4 years ago
  Megvii Engine Team b5016b9d29 feat(mge/parampack): add parampack in allreduce callback 4 years ago
  Megvii Engine Team 5ae89c799b refactor(mgb/grad): place grad at param.grad 4 years ago
  Megvii Engine Team 9faa32fcb7 fix(mge/imperative): fix grad callback 4 years ago
  Megvii Engine Team 6d4fd9384d fix(mge/optimizer): remove distributed optimizer 4 years ago
  Megvii Engine Team e50fa074cb fix(mge/imperative): remove backward from optimizer 4 years ago
  Megvii Engine Team 6070266766 refactor(mge/grad_manager): refactor gradmanager, add allreduce callback 4 years ago
  Megvii Engine Team 3f2eac2fe1 fix(mge/imperative): move functional/distributed.py to distributed/functional.py 4 years ago
  Megvii Engine Team b3889938dc feat(mge/examples): add trace & dump example of cifar10 quantization 4 years ago
  Megvii Engine Team 67859f04e1 fix(imperative): add __array__ and __array_wrap__ for tensorwrapper 4 years ago
  Megvii Engine Team b1ab3646f5 feat(imperative): add tensor sanity check 4 years ago
  Megvii Engine Team 5aa19f3d4a test(mge/imperative): add more testcases for function 4 years ago
  Megvii Engine Team 2a300415bc fix(mge): remove a duplicated __init__ call in jit.trace 4 years ago
  Megvii Engine Team 517c1ec007 fix(mge): disable dedup for io callback oprs 4 years ago
  Megvii Engine Team 939bf3b28d fix(mge/imperative): impl hashable for nms and cond_take 4 years ago
  Megvii Engine Team 8dc23e0fdf fix(mge/functional): fix indexing_one_hot and remote_recv 4 years ago
  Megvii Engine Team 3bbfef3009 fix(mge/imperative): impl hashable for coll-comm 4 years ago
  Megvii Engine Team 14ceda0dd3 ci(mge): fix Windows CI 4 years ago
  Megvii Engine Team 1fe8a21299 fix(mge): fix sublinear memory in jit.trace 4 years ago
  Megvii Engine Team 2df1ab9633 refactor(mge/jit): skip seed when checking equal rng op 4 years ago
  Megvii Engine Team 2beb65b19d fix(mge/module): tensor shape will not work when constructing numpy array 4 years ago
  Megvii Engine Team fc3eca845b feat(mge/imperative/jit): add dump input shape and xornet example 4 years ago
  Megvii Engine Team 1569cab0bc feat(mge/imperative): add shufflenet example 4 years ago
  Megvii Engine Team 087ceb52ef feat(mge/imperative): add more optimizer trace tests 4 years ago
  Megvii Engine Team 38a5c1c970 fix(mge): fix refcnt in TensorShape caster 4 years ago