CUDA之矩阵乘法——非方阵计算

最新推荐文章于 2025-09-26 19:55:59 发布

原创

最新推荐文章于 2025-09-26 19:55:59 发布 · 2.4k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#cuda

该博客详细介绍了如何使用CUDA进行非方阵矩阵乘法的计算，涉及A矩阵的M * N维度和B矩阵的N * M维度，通过CUDA代码实现高效运算。

说明

A矩阵为M * N，B矩阵为N * M

代码

#include "device_functions.h"
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include <stdio.h>
#include <iostream>

typedef struct {
    int width;
    int height;
    int stride;
    float* elements;
} Matrix;

#define BLOCK_SIZE  2
#define N           4
#define M           8

__device__ float GetElement(const Matrix A, int row, int col) {
    return A.elements[row * A.stride + col];
}

__device__ void SetElement(Matrix A, int row, int col, float value) {
    A.elements[row * A.stride + col] = value;
}

__device__ Matrix