wangwei
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
2645 Commits
28 Branches
25 MB
1915 Download
Tree: 2886245bb1
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '2886245bb1'
${ noResults }
Commit Graph

270 Commits (2886245bb194a6047d42bc07c816dbbfa5d351e9)

Author SHA1 Message Date
  Megvii Engine Team b55942a94d feat(dnn/naive/norm,-dnn/cuda/norm,-dnn/test/norm): add norm dnn opr, 4 years ago
  Megvii Engine Team bbafe69974 feat(dnn): add elemwise COND_LT_MOV 4 years ago
  Megvii Engine Team a0a5fcf182 feat(dnn): support tf32 4 years ago
  Megvii Engine Team 81065cf00e build(mgb/cutlass): merge partial headers 4 years ago
  Megvii Engine Team 6f0b582064 chore(imperative/amp): adapt dev 4 years ago
  Megvii Engine Team fc0f454685 fix(dnn/check_non_finite): adjust some details of CheckNonFinite 4 years ago
  Megvii Engine Team 98b5ee78c1 feat(mge/dnn): add lamb optimizer 4 years ago
  Megvii Engine Team c2e9860feb chore(license): remove all license in file header 4 years ago
  Megvii Engine Team e2f5156b69 refactor(megbrain): save fastrun result to algorithm cache 4 years ago
  Megvii Engine Team d968942fe3 perf(cuda): speedup direct large kernel conv 4 years ago
  Megvii Engine Team 7dc347697a feat(dnn/cuda): add typecvt uint16 4 years ago
  Megvii Engine Team 73112558d0 feat(mge/dnn): support checknonfinite for fp16 4 years ago
  Megvii Engine Team 14e9ad625d fix(megdnn): emit define-but-not-referenced and extra-;-ignored warning on cuda9.0~cuda9.1 4 years ago
  Megvii Engine Team 4c0bff1dba refactor(megdnn): refactor TEGRA_X1/X2 macro 4 years ago
  Megvii Engine Team b6ad457269 feat(cuda): support int1 simplewq conv 4 years ago
  Megvii Engine Team fd6f8e58b0 feat(mgb/dtype): add dtype qint1 4 years ago
  Megvii Engine Team 3228fb75a5 fix(cuda): conv algo heuristic choose 4 years ago
  Megvii Engine Team 8c415f4ed7 feat(dnn): cuda nhwc nearest resize support not 1 or 3 channel 4 years ago
  Megvii Engine Team ac26bdcef5 fix(cuda): fix direct conv speed and memory problem 4 years ago
  Megvii Engine Team f7994683bd feat(cuda): add large kernel direct conv to heuristic algo chooser 4 years ago
  Megvii Engine Team 6dc0c0b9cc fix(dnn): fix the sync problem in some kernels 4 years ago
  Megvii Engine Team 04193e3bd1 feat(dnn): add nearest mode for remap and resize 4 years ago
  Megvii Engine Team f78b60ec10 feat(bazel): make bazel gensass depend on cuda toolchain version automatically 4 years ago
  Megvii Engine Team f48227c07d feat(mgb): show more details for cuda driver api call 4 years ago
  Megvii Engine Team d8bb3ff5b4 fix(cuda): fix fp16 tensorcore gemm split k workspace 4 years ago
  Megvii Engine Team d7b0994a3e feat(cuda): add fp16 compute 16 kernel 4 years ago
  Megvii Engine Team 8a2e92bd6c refactor(cuda): depthwish large kernel 4 years ago
  Megvii Engine Team 6b8a69d5b6 feat(cuda): float16 depthwise large kernel conv compute fp32 4 years ago
  Megvii Engine Team bc385b5374 feat(cuda): support float16 depthwise large kernel conv 4 years ago
  Megvii Engine Team 7d2063e35a perf(cuda): speedup conv backward data with small feature map and large filter size 4 years ago
  Megvii Engine Team 72403e8929 perf(cuda): speedup chanwise conv with small feature map and large filter size 4 years ago
  Megvii Engine Team ab6d12caff feat(mge): add conv padding mode 4 years ago
  Megvii Engine Team 47fe766310 feat(dnn/cuda): add implicit bmm kernels for large kernel depthwise convolution backward filter opr 4 years ago
  Megvii Engine Team dcc9693582 feat(dnn/cuda): add heuristic rule for implicit batched gemm large kernel dwconv2d kernels 4 years ago
  Megvii Engine Team 6cefabe734 fix(dnn/cuda): fix ci 4 years ago
  Megvii Engine Team 888f4e46ae feat(dnn/cuda): add implicit bmm large kernel dwconv2d dgrad kernels 4 years ago
  Megvii Engine Team 08d8635ff5 feat(dnn/cuda): add implicit bmm large kernel dwconv2d fprop impl 4 years ago
  Megvii Engine Team 95ac055538 feat(dnn,mgb,imperative): add diag opr implement 4 years ago
  Megvii Engine Team cbbca5fb10 feat(mge): add softmax op use cudnn api 4 years ago
  Megvii Engine Team 20b42a8c3b fix(dnn): add naive lstm kernel 4 years ago
  Megvii Engine Team 2faa6ea5a9 Merge pull request #213 from kxz18:rnn 4 years ago
  Megvii Engine Team 1999307015 feat(mgb/opr): add dropout kernel 4 years ago
  Megvii Engine Team a93741815b feat(mgb/opr): add layernorm forward and backward kernel 4 years ago
  Megvii Engine Team a404cd7d06 fix(mgb/src): add tensorRT version check 4 years ago
  kxz@thumt102-1 8f48da7ffe feat(mgb/opr): add cell level rnn/lstm and sequence level rnn/lstm 4 years ago
  Megvii Engine Team 2881934cb8 feat(dnn/check_non_finite): addmul scale to check_non_finite opr 4 years ago
  Megvii Engine Team 0d16952470 fix(mgb/cuda): fix conv error when the input tensor is too large 4 years ago
  Megvii Engine Team 2696e4efaa feat(dnn): add float16 for remap backward 4 years ago
  Megvii Engine Team 1f0cc891b0 feat(dnn): enable eye to support bool 4 years ago
  Megvii Engine Team 11d75fecb5 feat(dnn/check_non_finite): add batch check_non_finite 4 years ago