InfiniTensor/include/cuda
Liyan Zheng 4211fd1f32 Fix: matmul transpose in convNHWC2gemm rule 2023-04-23 22:54:50 +08:00
..
cuda_any.h add AnyOp and cuda kernel 2023-04-23 00:16:03 +08:00
cuda_clip.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
cuda_common.h NNET supports TVM backend and kernels (#78) 2023-04-18 00:26:36 +08:00
cuda_conv2dreduce.h Add: hash match for membound kernels 2023-04-20 17:16:01 +08:00
cuda_element_wise.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
cuda_kernel_wihtout_config.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
cuda_pad_slice.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
cuda_runtime.h Fix: matmul transpose in convNHWC2gemm rule 2023-04-23 22:54:50 +08:00
cuda_split_concat.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
cuda_transpose.h add AnyOp and cuda kernel 2023-04-23 00:16:03 +08:00
cuda_unary.h Add activation operators and kernels 2022-09-16 13:58:57 +08:00
cuda_utility.h Simplify tensor transfer between CPU and CUDA (#10) 2022-08-25 11:29:16 +08:00
gather.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
gbmm_g2bmm.cuh Fix: gbmm kernel 2023-04-22 21:14:52 +08:00
gbmm_g2bmm.h Fix CMake USE_CUDA (#36) 2022-09-21 12:28:00 +08:00
operator_timer.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
resize.cuh ADD: reconfig ResizeObj, support "tf_crop_and_resize " and cubic coeff kernel. (#59) 2022-12-24 04:02:21 +08:00
softmax.h Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00