p59230764
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
InfiniTensor/test/kernels/cuda
History
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
..
test_cuda_G2BMM.cc ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
test_cuda_GBMM.cc ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
test_cuda_batch_norm.cc ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
test_cuda_clip.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_concat.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_conv.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_conv_fp16.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_conv_transposed_2d.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_element_wise.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_extend.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_gather.cc impl sqrt on CUDA (#109) 2023-08-18 12:17:47 +08:00
test_cuda_inception.cc fix: remove inline keyword in class; rename getter and setter for inputOf and outputOf 2023-03-25 12:04:24 +08:00
test_cuda_matmul.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_pad.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_pooling.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_reduce_mean.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_reshape.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_resize.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_slice.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_softmax.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_split.cc memory_allocator (#103) 2023-08-13 13:39:35 +08:00
test_cuda_unary.cc impl sqrt on CUDA (#109) 2023-08-18 12:17:47 +08:00
test_perfengine.cc ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00