GPU-JS报错：kernel参数过多，矩阵相乘脚本运行异常求助

核心原因 #

这个错误通常是内核函数声明的参数与实际传入的参数不匹配，或输入数据格式不符合GPU.js的解析要求，导致内核误判参数数量。以下是具体排查和修复步骤：

1. 补全未定义的输入矩阵 #

你的代码中直接调用multiplyMatrix(a, b)，但未定义a和b变量。GPU.js需要明确的二维矩阵输入，否则会触发参数解析错误。添加矩阵初始化代码：

// 创建512x512的随机输入矩阵
const a = Array.from({ length: 512 }, () => Array.from({ length: 512 }, () => Math.random()));
const b = Array.from({ length: 512 }, () => Array.from({ length: 512 }, () => Math.random()));

2. 核对GPU.js版本兼容性 #

若你使用的是较新版本的GPU.js，旧示例写法可能存在解析偏差。可以显式声明参数类型帮助内核正确识别：

const multiplyMatrix = gpu.createKernel(function(a, b) {
  let sum = 0;
  for (let i = 0; i < 512; i++) {
    sum += a[this.thread.y][i] * b[i][this.thread.x];
  }
  return sum;
})
.setOutput([512, 512])
.setArgumentsType(['Array(Array(Number))', 'Array(Array(Number))']);

3. 确保参数数量严格匹配 #

检查内核函数声明的参数数量（此处为2个：a、b）与调用时传入的参数数量完全一致，避免多传或少传。

4. 重新安装依赖排除损坏 #

若依赖安装不完整，可能导致内核解析异常，执行以下命令重装：

npm uninstall gpu.js
npm install gpu.js

完整可运行代码 #

const { GPU } = require('gpu.js');
const gpu = new GPU();

// 初始化输入矩阵
const a = Array.from({ length: 512 }, () => Array.from({ length: 512 }, () => Math.random()));
const b = Array.from({ length: 512 }, () => Array.from({ length: 512 }, () => Math.random()));

const multiplyMatrix = gpu.createKernel(function(a, b) {
  let sum = 0;
  for (let i = 0; i < 512; i++) {
    sum += a[this.thread.y][i] * b[i][this.thread.x];
  }
  return sum;
}).setOutput([512, 512]);

const c = multiplyMatrix(a, b);
console.log('结果矩阵维度:', c.shape);

内容的提问来源于stack exchange，提问作者Lunatyk