elementwise op#238

Merged

zjing14 merged 22 commits into

May 19, 2022

Commits on May 16, 2022

Add elementwise operation kernel and example
rocking
committed

Commits on May 17, 2022

Add comment
rocking
committed
Add template argument of dim . Prepare to support multiple dimension
rocking
committed
Rename example
rocking
committed
Support 1 dimension
rocking
committed
Add static assert
rocking
committed
Add comment
rocking
committed
Extract pad
rocking
committed
Remove redundant argument
rocking
committed
Support any dimension for elementwise operation
rocking
committed
Remove line
rocking
committed

Commits on May 18, 2022

Let it be the multiple number of CU
rocking
committed
Move thread per block to the parameter of constructor
rocking
committed
rename threadPerBlock with blockSize
rocking
committed
Support double
rocking
committed
rename kernel function name
rocking
committed
remove redundant include header
rocking
committed
Refine type
rocking
committed
Need to the final dimension
rocking
committed
Refine variable name
rocking
committed
Refine type
rocking
committed
Use index_t instead of int in API
rocking
committed