2887 Commits (master)
 

Author SHA1 Message Date
  Megvii Engine Team 07bdb3bf1e feat(imperative): add swapaxes 3 years ago
  huangxinda a0862865fe feat(mge/third_party): update cutlass version 3 years ago
  Megvii Engine Team 972a8d5489 feat(ci): add model compatibility check in ci 3 years ago
  Megvii Engine Team e694301721 fix(ops/jit): skip lookup include path when nvcc executable not found 3 years ago
  Megvii Engine Team 02c1a0c3de fix(imperative/format): warn once when parameter without format been attached 3 years ago
  fanhqme2 6ef1e12cfd
fix flops count bug for ConvTranspose2d 3 years ago
  Megvii Engine Team c2deef1a97 feat(mge): aad atlas710 support 3 years ago
  Megvii Engine Team df5ebd3da7 fix(imperative/ops): fix the vmemory problem in 1.9 3 years ago
  Megvii Engine Team 4e66e0eb1f feat(megdnn/softmax): add softmax operator in OpenCL 3 years ago
  Megvii Engine Team b36b5bd8cb refactor(mgb): check input when profiling 3 years ago
  Megvii Engine Team 6c9b3a58e3 refactor(dnn): remove algorithm cache queries 3 years ago
  Megvii Engine Team 8563f51404 fix(imperative): fix buildin reduce keepdim 3 years ago
  Megvii Engine Team 96d90be1c6 feat(dnn): fallback support int4 relayout 3 years ago
  Megvii Engine Team eef0308b29 feat(serialization): add no_change_graph and version param whem dump model 3 years ago
  Megvii Engine Team 4ab5f970e9 fix(build): fix ci error 3 years ago
  Megvii Engine Team b9a6932341 feat(imperative): channel default model format to fbs v2 3 years ago
  Megvii Engine Team 283a2f77e7 feat(serialization): support test for new serialization format 3 years ago
  Megvii Engine Team 50faabf614 feat(serialization): support the registry for new serialization format 3 years ago
  Megvii Engine Team a694fb3304 feat(serialization): implement the new serialization format 3 years ago
  Megvii Engine Team ca4a5da013 feat(serialization): add new serialization format define 3 years ago
  Megvii Engine Team ba32360af4 feat(lite): add set opencl buffer kernel cache lite api 3 years ago
  Megvii Engine Team 711b5bf502 fix(dnn/arm_common): fix some load beyond memory 3 years ago
  Megvii Engine Team 3ebb8db01a feat(third_party/cutlass): update to version 2.8 3 years ago
  Megvii Engine Team da91e650a5 refactor(ops/layer_norm): speed up the host speed of layer_norm 3 years ago
  Megvii Engine Team 67cfce9f2f fix(imperative/amp): add is_scalar check in elemwise and concat 3 years ago
  Megvii Engine Team d313f92610 fix(imperative/amp): fix format transformation for symbol trans 3 years ago
  Megvii Engine Team 261a5bce23 feat(imperative/amp): add dimshuffle in set_format for nhwc 3 years ago
  Megvii Engine Team c9e56f4987 feat(imperative/amp): add dimshuffle before creating nhwc tensor 3 years ago
  Megvii Engine Team d57a071271 feat(imperative/amp): add fallback for op not supported for nhwc tensor 3 years ago
  Megvii Engine Team 38a9aa9faf feat(imperative/amp): add auto dimshuffle for elemwise and concat 3 years ago
  Megvii Engine Team cd26376549 style(imperative/amp): reformat code 3 years ago
  Megvii Engine Team 3892aa0b6e fix(imperative/amp): fix bn params for nhwc amp 3 years ago
  Megvii Engine Team 6f0b582064 chore(imperative/amp): adapt dev 3 years ago
  Megvii Engine Team ee984e8608 fix(imperative/amp): fix distributed backward callback for nhwc amp 3 years ago
  Megvii Engine Team 15c6da6218 feat(imperative/amp): add nhwc support for adaptive pooling 3 years ago
  Megvii Engine Team c28a875fac fix(imperative/amp): adapt new transformation 3 years ago
  Megvii Engine Team fd41302cc1 feat(imperative/amp): add set_format 3 years ago
  Megvii Engine Team fc633ce4ff fix(imperative/amp): fix custom grad in Subgraph 3 years ago
  Megvii Engine Team 673b295d75 feat(imperative/amp): remove conv_format and bn param_dim configs 3 years ago
  Megvii Engine Team 7e9aa742e6 feat(imperative/amp): enable auto_convert_format by default 3 years ago
  Megvii Engine Team fc0f454685 fix(dnn/check_non_finite): adjust some details of CheckNonFinite 3 years ago
  Megvii Engine Team 3bd40887b6 feat(mgb/opr): add NHWC support for AdaptivePooling 3 years ago
  Megvii Engine Team e393d1cf65 feat(mge/amp): add convert_format module for NHWC training 3 years ago
  Megvii Engine Team 533fb5bf49 feat(imperative): support formatted tensor and add special op rules 3 years ago
  Megvii Engine Team 4aa79c453b perf(mge): override grad of matmul 3 years ago
  Megvii Engine Team 98b5ee78c1 feat(mge/dnn): add lamb optimizer 3 years ago
  Megvii Engine Team a926878c01 feat(imperative): remove symbolvar of imperative 3 years ago
  Megvii Engine Team 14813d13c0 fix(whl): fix whl broken: patchelf on big (> 4G) file will 3 years ago
  Megvii Engine Team 9e0583e13a feat(dnn/arm_common): add arm_common chanwise dot 11x11 3 years ago
  Megvii Engine Team 115bcbce2b feat(lite): add fitting mode for load and run 3 years ago