InfiniTensor/test/kernels
constroy Li 48847958d0
impl sqrt on CUDA (#109)
* impl sqrt on CUDA
fix parser of Gather and ReduceMean

* fix test_gather

* fix test_cuda_gather

* impl sqrt cpu and add sqrt to test_cuda_unary

* cuda_unary supports arbitary shapes

* fix SplitOp with dim=-1

* fix SplitOp with dim=-1
2023-08-18 12:17:47 +08:00
..
bang memory_allocator (#103) 2023-08-13 13:39:35 +08:00
cuda impl sqrt on CUDA (#109) 2023-08-18 12:17:47 +08:00
intelcpu memory_allocator (#103) 2023-08-13 13:39:35 +08:00