RuntimeError: CUDA error: __global__ function call is not configured

表明在 CUDA 设备上调用的核函数 没有正确配置线程块和网格维度

一般体现在:

直接调用 kernel 函数,而不是通过 launch 函数 指定 kernel 函数调用

解决方法(示例):

cpp 复制代码
// kernel function
__global__ void Idtest_kernel(float *a, int N) {
    int idx = blockIdx.x * blockDim.x + threadIdx.x;
    if(idx < N){
      a[idx] = idx;
    }
}

// launch kernel function
void launch_Idtest_kernel(torch::Tensor a) {
  int N = a.numel();
  int block_size = 256;
  int grid_size = (N + block_size - 1) / block_size;
  Idtest_kernel<<<grid_size, block_size>>>(
    reinterpret_cast<float *>(a.data_ptr()),
    N);
}
相关推荐
sunfove几秒前
空间几何的基石:直角、柱、球坐标系的原理与转换详解
人工智能·python·机器学习
<-->1 分钟前
pytorch vs ray
人工智能·pytorch·python
知乎的哥廷根数学学派3 分钟前
基于多尺度特征提取和注意力自适应动态路由胶囊网络的工业轴承故障诊断算法(Pytorch)
开发语言·网络·人工智能·pytorch·python·算法·机器学习
Yuer20255 分钟前
低熵回答倾向:语言模型中的一种系统稳定态
人工智能·机器学习·语言模型·ai安全·edca os
yuzhiboyouye10 分钟前
c/p比结合VIX值,最早的信号
人工智能
Byron Loong12 分钟前
【机器视觉】GTX5050到GTX5090算力比较
人工智能
郝学胜-神的一滴13 分钟前
《机器学习》经典教材全景解读:周志华教授匠心之作的技术深探
数据结构·人工智能·python·程序人生·机器学习·sklearn
知乎的哥廷根数学学派13 分钟前
基于物理约束与多源知识融合的浅基础极限承载力智能预测与工程决策优化(以模拟信号为例,Pytorch)
人工智能·pytorch·python·深度学习·神经网络·机器学习
yubo050914 分钟前
【无标题】
人工智能·深度学习
AI有元力14 分钟前
GEO优化全链路解密:从策略到服务,系统性赢得AI生态
人工智能