Question

我有一个.cu文件，当它自己编译时，右键单击并选择编译，它编译得很好，但是当我有另一个头文件，一个c ++头文件，调用这个.cu文件时，构建失败。已编辑.cu文件属性以使用CUDA编译器构建。我得到的错误是＆＃39; blockIdx＆＃39;：未声明的标识符＆blockDim＆＃39;：未声明的标识符等等。基本上我希望用c ++编译器编译cuda代码的错误。那么可以在c ++头文件中包含.cu cuda代码吗？

这是.cu文件：

Matrix.cu

#include <cuda.h>
#include <cuda_runtime.h>
#include <cuda_device_runtime_api.h>

#define BLOCKSIZE 32

using namespace std;



template<typename T> class Matrix
{
public:
typedef T value_type;
~Matrix();
Matrix();
Matrix(int rows, int columns);
int height;
int width;
int stride;
size_t size;

void CreateIdentity(Matrix<T>&I);
private:
vector<T> elements;
T* firstElement;
};


template<typename T>
Matrix<T>::~Matrix()
{
}

template<typename T>
Matrix<T>::Matrix()
{
}

template<typename T>
Matrix<T>::Matrix(int rows, int columns)
{
height = rows;
width = columns;
stride = columns; //in row major order this is equal to the # of columns
elements.resize(rows*columns);
firstElement = elements.data();
size = height*width*sizeof(T);
}



__global__ void IdentityMatrixKernel(float* identity, int size)
{
int index_x = blockIdx.x * blockDim.x + threadIdx.x;
int index_y = blockIdx.y * blockDim.y + threadIdx.y;

// map the two 2D indices to a single linear, 1D index
int grid_width = gridDim.x * blockDim.x;
int index = index_y * grid_width + index_x;

// map the two 2D block indices to a single linear, 1D block index
//int result = blockIdx.y * gridDim.x + blockIdx.x;

// write out the result

if (index % (size+1))
{
    identity[index] = 0;
}

else
{
    identity[index] = 1;
}
}


template<typename T>
void Matrix<T>::CreateIdentity(Matrix<T>&I)
{
float* d_I;
int size1 = I.height;
int size2 = I.height*I.width*sizeof(float);

cudaMalloc(&d_I,size2);

dim3 block_size;
block_size.x = BLOCKSIZE;
block_size.y = BLOCKSIZE;

dim3 grid_size;
grid_size.x = size1/ block_size.x + 1;
grid_size.y = size1/ block_size.y + 1;

IdentityMatrixKernel<<<block_size,grid_size>>>(d_I,size1);

cudaMemcpy(I.GetPointer(),d_I,size2,cudaMemcpyDeviceToHost);

cudaFree(d_I);
}

这是#include＆＃34; Matrix.cu＆＃34;

的头文件

element.h展开

#pragma once
#include "Matrix.cu"
#include <vector>

using namespace std;

class Element
{
public:
Element(void);
~Element(void);
Element(int iD, float k, vector<int> nodes);
Element(int iD, vector<int> nodes, int pId);

void SetElementType(DOF type);
DOF GetElementType();
int GetNodeId(int index);
int GetNodesPerElement();
int GetPartId();

void CalculateShapeFunctions(Matrix<int> spaceCoordinates);
void CalculateSShapeDerivative(Matrix<int> spaceCoordinates);
void CalculateTShapeDerivative(Matrix<int> spaceCoordinates);
Matrix<float> GetShapeFunctions();
float GetSShapeDerivative(int row, int column);
float GetTShapeDerivative(int row, int column);
void SetStrainDisplacement(Matrix<float> B);
Matrix<float> GetStrainDisplacement();

private:

int elementId;
float stiffness;
vector<int> nodeIds;
DOF elementType;
int partId;
Matrix<float> shapeFunctions;
Matrix<float> sShapeDerivative;
Matrix<float> tShapeDerivative;
Matrix<float> strainDisplacement;
};

编辑：

因此我被指示尝试将实现cuda的模板类成员函数分离为.cu文件，同时保持模板类定义和任何模板成员函数不在原始头文件中使用cuda。这似乎在正确的道路上，c ++编译器编译.h文件，而cuda编译器执行.cu，但我无法摆脱链接错误。我知道我需要在.cu文件中为我需要的类型显式实例化我的模板类以避免链接错误，但我似乎仍然得到它们。

我在.cu文件的末尾实例化了我的模板类，如下所示：

template class Matrix<float>;
template class Matrix<int>;
template class Matrix<string>;

我现在使用cuda将模板成员函数链接错误。

Answer 1

答案：.cu文件不能用作#include＆＃34; file.cu＆＃34;像头文件一样，因为它们将使用C ++编译器而不是cuda进行编译。解决方案是将实现cuda的任何内容移动到一个单独的.cu文件中，同时仍然在模板类定义中保留模板函数的定义，并添加一个#include＆＃34; file.h＆＃34;在file.cu中。要解决移动到.cu文件的模板函数声明的任何链接错误，模板类的显式实例化被添加到头文件的底部。由于在使用cuda的模板函数中只使用了浮点类型，因此只添加了float类型的实例化：模板类Matrix。上面的解决方案编译完美。

你能在c ++头文件中包含.cu扩展头吗？

1 个答案: