chundoong-lab-ta/SamsungDS23/cuda-mpi-example/matmul.h

6 lines
159 B
C
Raw Normal View History

2023-08-02 02:28:13 +09:00
#pragma once
void matmul_initialize(int M, int N, int K);
void matmul(const float *A, const float *B, float *C, int M, int N, int K);
void matmul_finalize();