InfiniTensor/include
wendy12022 9032cbb973
Add: reshape/flatten/identity OP and cuda kernel (#34)
* ADD:reshape/flatten/identity operators and cuda kernel.

fix: use cudaMemcpyAsync

clang format.

ADD flatten/identity operator.

add test for reshape.

ADD: reshape operator and cuda kernel.

* Fix: seperate CUDA tests & remove old header

Co-authored-by: Liyan Zheng <liyan-zheng@outlook.com>
2022-09-21 14:04:30 +08:00
..
core Add: reshape/flatten/identity OP and cuda kernel (#34) 2022-09-21 14:04:30 +08:00
cuda Fix CMake USE_CUDA (#36) 2022-09-21 12:28:00 +08:00
nnet Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
operators Add: reshape/flatten/identity OP and cuda kernel (#34) 2022-09-21 14:04:30 +08:00
utils Tensor serialization (#25) 2022-09-13 11:27:41 +08:00
test.h Extended DataType class and Runtime interaction (#9) 2022-08-23 16:55:59 +08:00