From 5c369c6e26da9d6a26a99980803c683cacea7d6a Mon Sep 17 00:00:00 2001 From: lucasdelimanogueira Date: Tue, 30 Apr 2024 01:38:08 -0300 Subject: [PATCH] matmul cuda --- csrc/cuda.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/csrc/cuda.h b/csrc/cuda.h index 1e83602..567b36b 100644 --- a/csrc/cuda.h +++ b/csrc/cuda.h @@ -13,10 +13,10 @@ __global__ void elementwise_mul_tensor_cuda_kernel(float* data1, float* data2, float* result_data, int size); __host__ void elementwise_mul_tensor_cuda(Tensor* tensor1, Tensor* tensor2, float* result_data); + __global__ void matmul_tensor_cuda_kernel(float* data1, float* data2, float* result_data, int rows1, int cols1, int cols2); + __host__ void matmul_tensor_cuda(Tensor* tensor1, Tensor* tensor2, float* result_data); + __global__ void pow_tensor_cuda_kernel(float* data, float power, float* result_data, int size); __host__ void pow_tensor_cuda(Tensor* tensor, float power, float* result_data); - __global__ void matmul_tensor_cuda_kernel(float* data1, float* data2, float* result_data, int rows1, int cols1, int cols2) - __host__ void matmul_tensor_cuda(Tensor* tensor1, Tensor* tensor2, float* result_data); - #endif /* CUDA_KERNEL_H_ */