InfiniTensor/test
wendy12022 3c6e208f42
ADD:concat/split operator and cuda kernels (#29)
* ADD:concat/split operator and cuda kernels

refector

minor change comment

ADD:concat/split operator and cuda kernels

merge split_kernel and concat_kernel to split_concat_kernel.

Revert "fix"

This reverts commit 459926be09a838658ec55f1e0a72b3cf17037d5c.

fix

ADD:concat/split operator and cuda kernels

change whole tensor name to composed tensor

fix some

remove unused header.

rebase

add CudaKernel

add test for split.

ADD split operator and cuda kernel.

modify test.

ADD:concat operator and cuda kernel.

ADD:concat/split operator and cuda kernels

fix some

remove unused header.

rebase

add CudaKernel

ADD:concat/split operator and cuda kernels

add test for split.

ADD split operator and cuda kernel.

modify test.

ADD:concat operator and cuda kernel.

* remove extra comment; typo fix.

Co-authored-by: Haojie Wang <haojie0429@gmail.com>
2022-09-29 11:01:30 +08:00
..
core Tensor serialization (#25) 2022-09-13 11:27:41 +08:00
kernels ADD:concat/split operator and cuda kernels (#29) 2022-09-29 11:01:30 +08:00
nnet Add TVM codegen for MemboundOp (#35) 2022-09-22 18:06:45 +08:00
operators ADD:concat/split operator and cuda kernels (#29) 2022-09-29 11:01:30 +08:00
script cuDNN conv tuning (#16) 2022-08-29 21:37:07 +08:00