Code

Cumsum op implementation#7050

Code

Merged

oneflow-ci-bot merged 46 commits intomasterOneflow-Inc/oneflow:masterfrom wyushun_dev_cumsumOneflow-Inc/oneflow:wyushun_dev_cumsumCopy head branch name to clipboardJan 1, 2022

+679-75

Commits

Commits on Jan 1, 2022

• add cumsum op's forward definition

  wyushun

  committed
  856f1a5

• add cumsum forward test case

  wyushun

  committed
  3621aa3

• cumsum ver3

  wyushun

  committed
  cbaf9a0

• remove calculating time

  wyushun

  committed
  4ddce0f

• add cumsum forward gpu implementation

  wyushun

  committed
  b6c9cb5

• fix gpu forward error

  wyushun

  committed
  9cabf38

• change var name

  wyushun

  committed
  141ef48

• remove annotation

  wyushun

  committed
  8954f02

• add cumsum cpu forward multi-thread support

  wyushun

  committed
  bfae4c4

• add multi-thread annotation

  wyushun

  committed
  213657f

• add cumsum grad definition

  wyushun

  committed
  9805fdb

• update

  wyushun

  committed
  04d9f62

• add cumsum cpu backward

  wyushun

  committed
  8832bce

• add cumsum cpu backward functor

  wyushun

  committed
  f21703c

• add cumsum autograd

  wyushun

  committed
  9ee1a6b

• update

  wyushun

  committed
  eaade4a

• remove user interface

  wyushun

  committed
  5e3e7ea

• use random method to test cumsum forward

  wyushun

  committed
  89911d7

• add cumsum gpu backward

  wyushun

  committed
  4ea9ebd

• add cumsum gpu test

  wyushun

  committed
  3d72ce2

• fix gpu backward bug

  wyushun

  committed
  9129a96

• add a 3d cuda kernel try

  wyushun

  committed
  7b23a4f

• Revert "add cumsum gpu test"

  Show description for fc7dfd7
  wyushun

  committed
  fc7dfd7

• Revert "Revert "add cumsum gpu test""

  Show description for dc70fdf
  wyushun

  committed
  dc70fdf

• change nele to ele_cnt

  wyushun

  committed
  4763e5a

• add test_cumsum.py in oneflow/test/modules

  wyushun

  committed
  0e0c718

• change original test_cumsum to autotest version

  wyushun

  committed
  caf8671

• optimize cumsum for special up_space and down_space

  wyushun

  committed
  88ac0e0

• add two special cu func

  wyushun

  committed
  4b61610

• add cumsum doc

  wyushun

  committed
  7174053

• update doc

  wyushun

  committed
  382917b

• update doc

  wyushun

  committed
  8943d3a

• update code according to bbuf's review

  wyushun

  committed
  d25f519

• ditto

  wyushun

  committed
  6a22d7a

• change pin/pout to in_ptr/out_ptr

  wyushun

  committed
  a3ce1e6

• remove multi-thread func

  wyushun

  committed
  0f11699

• update doc

  wyushun

  committed
  51398d1

• use tensor processor

  wyushun

  committed
  dbace89

• update by review

  wyushun

  committed
  bac94d2

• update by review

  wyushun

  committed
  c955b8a

• update

  wyushun

  committed
  50d53d2

• update

  wyushun

  committed
  81a4b8d

• auto format by CI

  

  oneflow-ci-bot

  authored and

  wyushun

  committed
  5dc75c6

• auto format by CI

  

  oneflow-ci-bot

  authored and

  wyushun

  committed
  b85be6f

• update doc

  wyushun

  committed
  c7dba96

• update

  wyushun

  committed
  7a8f774