CUDA C++ Template de Paramètre de Noyau

Tous explicite de la spécialisation des déclarations doit être visible au moment de l'instanciation du modèle. Votre spécialisation explicite déclaration n'est visible que dans les grains.cu de l'unité de traduction, mais pas dans la main.cu.

Le code suivant est en effet de fonctionner correctement (à l'exception de l'ajout d'un __global__ qualifier lors de l'instanciation explicite de l'instruction).

#include<cuda.h>
#include<cuda_runtime.h>
#include<stdio.h>
#include<conio.h>

template<bool approx>
__global__ void kernel()
{
    if(approx)
    {
        printf("True branch\n");
    }
    else
    {
        printf("False branch\n");
    }
}

template __global__ void kernel<false>();

int main(void) {
    kernel<false><<<1,1>>>();
    getch();
    return 0;
}

MODIFIER

En C++, basé sur un modèle de fonctions ne sont pas compilés jusqu'à ce qu'une instanciation explicite de la fonction est rencontré. De ce point de vue, CUDA, qui supporte entièrement les modèles, se comporte exactement de la même manière qu'en C++.

Pour prendre un exemple concret, lorsque le compilateur trouve quelque chose comme

template<class T>
__global__ void kernel(...params...)
{
    ...
    T a;
    ...
}

il vérifie juste la syntaxe de la fonction, mais ne produit pas de code objet. Ainsi, si vous compilez un fichier avec un simple basé sur un modèle de fonction comme ci-dessus, vous aurez un "vide", fichier de l'objet. C'est raisonnable, étant donné que le compilateur ne sais pas quel type d'assigner à a.

Le compilateur génère un code objet uniquement lorsqu'il rencontre une instanciation explicite de la fonction de modèle. C'est à ce moment-là, combien de compilation basé sur un modèle de fonctions de travail et ce comportement introduit une restriction de fichiers multiples projets: la mise en œuvre (définition) basé sur un modèle de fonction doit être dans le même fichier que sa déclaration. Donc, vous ne pouvez pas séparer l'interface contenues dans kernels.cuh dans un fichier d'en-tête séparé de kernels.cu, qui est la principale raison pour laquelle la première version de votre code ne compile pas. En conséquence, vous devez inclure à la fois l'interface et la mise en œuvre dans un fichier qui utilise les modèles, à savoir, vous devez inclure dans main.cu à la fois, kernels.cuh et kernels.cu.

Depuis aucun code n'est généré sans une instanciation explicite, les compilateurs tolérer l'inclusion plus d'une fois le même fichier de modèle avec les deux déclarations et les définitions dans un projet sans générer d'erreurs de liaison.

Il y a plusieurs tutoriels sur l'utilisation de templates en C++. Un Idiot Guide pour les Modèles C++ - Partie 1, en dehors de l'irritant titre, vous donneront des instructions étape par étape introduction sur le sujet.

OriginalL'auteur JackOLantern