InfiniTensor/test/operators
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
..
test_conv.cc Fix CMake USE_CUDA (#36) 2022-09-21 12:28:00 +08:00
test_conv_transposed_2d.cc Fix CMake USE_CUDA (#36) 2022-09-21 12:28:00 +08:00
test_element_wise.cc Fix CMake USE_CUDA (#36) 2022-09-21 12:28:00 +08:00
test_matmul.cc Fix CMake USE_CUDA (#36) 2022-09-21 12:28:00 +08:00
test_pooling.cc Fix CMake USE_CUDA (#36) 2022-09-21 12:28:00 +08:00
test_reshape.cc Add: reshape/flatten/identity OP and cuda kernel (#34) 2022-09-21 14:04:30 +08:00