160 Commits (0bed6c0fd9f27a87194c16d3b612ec117db2983c)

Author SHA1 Message Date
  Megvii Engine Team fd6f8e58b0 feat(mgb/dtype): add dtype qint1 3 years ago
  Megvii Engine Team f7a5fe17d3 feat(mge/config): add a config option for memory forwarding 3 years ago
  Megvii Engine Team 2a900a69cb perf(imperative): improve reduce op performance 3 years ago
  Megvii Engine Team 273c0e8745 fix(autodiff): fix some bugs in relation to 2nd order grad 3 years ago
  Megvii Engine Team 2f3bc2db9d perf(mge/utils): move astensor1d into C++ 3 years ago
  Megvii Engine Team fa62f6c06e perf(mge/utils): move convert_input into C++ 3 years ago
  Megvii Engine Team d98be08030 perf(mge): move Const into C++ 3 years ago
  Megvii Engine Team 1709b3940b perf(mge/functional): speed up Broadcast and Reshape 3 years ago
  Megvii Engine Team 0f736a0ab4 perf(mge/functional): speed up Dimshuffle 3 years ago
  Megvii Engine Team 3e5e08b0b4 perf(mge/functional): speed up RemoveAxis 3 years ago
  Megvii Engine Team 3c3fc6f33c refactor(imperative): move python code of elemwise/reduce/conv2d/bn to c++ 3 years ago
  Megvii Engine Team 87f00232f2 fix(mge/gm): fix missing dtype checking while attach tensors 3 years ago
  Megvii Engine Team ca00177719 perf(dispatch): speed up dispatch system 3 years ago
  Megvii Engine Team 59cbf9583d fix(subgraph): use CompiledOp in cpu to avoid workspace error 3 years ago
  Megvii Engine Team b6ce02a152 fix(subgraph): fallback back to cg if jit unsupported 3 years ago
  Megvii Engine Team 21f5a7fcc0 fix(subgraph): fix device recognition and scalar propagate 3 years ago
  Megvii Engine Team 2775f4580c feat(subgraph): subgraph builder supports jit and custom grad 3 years ago
  Megvii Engine Team ca2deebc0f fix(imperative/tensor): make @ operator has the same functionality as matmul functional 3 years ago
  Megvii Engine Team e860a08386 refactor(mge/indexing): move indexing into c++ 3 years ago
  Megvii Engine Team e6706be23a refactor(imperative): remove infer_output_mem_desc 3 years ago
  Megvii Engine Team c34a75d0f4 fix(trace): assume result is not scalar when shape is valid 3 years ago
  Megvii Engine Team de8ffe0c12 refactor(imperative): unify interpreter option setting 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 a72e0cb568 feat(imperative,src): add jit builder for custom op 3 years ago
  Megvii Engine Team 7fa70df9f2 style(imperative): remove absolute imports in megengine 3 years ago
  Megvii Engine Team b75d10094d fix(mge/elwise): removed back to fp32 mode 3 years ago
  Megvii Engine Team 9fd2e66350 feat(mge/elwise): removed back to fp32 mode 3 years ago
  Megvii Engine Team 8a222d2c9c fix(mge): replace functional paramter 3 years ago
  Megvii Engine Team 27aa648ba7 fix(mge): control compute_mode by context 3 years ago
  Megvii Engine Team a404c508e9 feat(mge): support dump with specific format 3 years ago
  Megvii Engine Team cebda6ff8d feat(mge/imperative): add ctc loss 3 years ago
  Megvii Engine Team cdb692d2fa refactor(imperative): add TODO tag for some functions 3 years ago
  Megvii Engine Team 90dd07161c refactor(imperative): modify the python interface of custom op 3 years ago
  Megvii Engine Team cbf024bf20 feat(imperative): add adaptor between custom op and imperative runtime 3 years ago
  Megvii Engine Team 27f2ecc436 feat(imperative/opr): add extern c opr and support dump 3 years ago
  Megvii Engine Team c5f8b58385 docs(docstring): transfer to google style 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 a7a3bf2d6c test(subgraph): simple test for subgraph 3 years ago
  Megvii Engine Team d063d5774f perf(functional): use fma to reduce elemwise but disable subgraph compilation 3 years ago
  Megvii Engine Team 2a063f8e87 fix(subgraph): fix scope mismatch of subgraph content 3 years ago
  Megvii Engine Team 8c47c1f149 perf(syncbn): reimplement with subgraph 3 years ago
  Megvii Engine Team 18274e023c fix(imperative): fix error message when applying custom function with non-tensor arguments 3 years ago
  Megvii Engine Team bfc4e7a966 docs(mge): fix amp docstring problems 3 years ago
  Megvii Engine Team 070c811732 fix(imperative): remove convert_inputs 3 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 ffe2bb2eb2 fix(mge): fix some errors caused by unknown shape when using symbolic trace or building graph 4 years ago
  Megvii Engine Team 2cd9823210 fix(mgb/tensorrt): fix trt runtime, padding channel to a multiple of 4 when using kCHW4 IOFormat 3 years ago
  Megvii Engine Team 522e556b83 feat(autodiff): support higher order grad 4 years ago