实现CUDA核函数——二维矩阵相乘

初学CUDA，自己练手实现二维矩阵乘法，且并发线程数与矩阵计算数据量自适应。计算结果，与调用numpy、torch库的计算方法进行比较验证通过。

环境：
win10，PyCUDA: (2021, 1)，Python: 3.9.7

说明：调用时传入的grid、block参数，不能超过GPU的限制，所以需要先调用pycuda.driver库进行相关操作获取，网上有很多资料。

代码如下：

# -*- coding: utf-8 -*-
"""
Created on Sat Feb 19 21:33:17 2022

@author: xuning
"""

import pycuda.autoinit
from pycuda.compiler import SourceModule
import pycuda.gpuarray as gpuarray
import numpy as np
import torch
import sys
from IPython.core.interactiveshell import InteractiveShell
InteractiveShell.ast_node_interactivity='all'

print("python ready")
print(f'The version of PyCUDA: {pycuda.VERSION}')
print(f'The version of Python: {sys.version}')

mod = SourceModule(r"""

/* gpuMatMultKernel_ex：GPU下矩阵乘法核函数,扩展版本
*  a:第一个矩阵指针，表示a[M][N]
*  b:第二个矩阵指针，表示b[N][S]
*  result:结果矩阵&